devplacepy/soft_delete.py at 9a8046ab2ad62fe787b86607b09d5aaff4cc103b

 # retoor <retoor@molodetz.nl>
 from .core import _drop_index, _in_clause, _index, _now_iso, db
 from .pagination import build_pagination
 SOFT_DELETE_TABLES = [
     "posts",
     "comments",
     "gists",
     "projects",
     "news",
     "news_images",
     "project_files",
     "attachments",
     "votes",
     "reactions",
     "bookmarks",
     "follows",
     "poll_votes",
     "polls",
     "poll_options",
     "sessions",
     "instances",
     "instance_schedules",
     "backup_schedules",
     "devii_conversations",
     "devii_tasks",
     "devii_lessons",
     "devii_virtual_tools",
     "user_customizations",
     "project_forks",
     "issue_tickets",
     "issue_comment_authors",
     "notification_preferences",
     "deepsearch_sessions",
     "deepsearch_messages",
     "isslop_analyses",
     "devrant_tokens",
     "access_tokens",
     "email_accounts",
     "user_relations",
     "seo_metadata",
 ]
 def ensure_soft_delete_columns(table, *, db_handle=None):
     handle = db_handle or db
     if table not in handle.tables:
         return
     target = handle[table]
     if not target.has_column("deleted_at"):
         target.create_column_by_example("deleted_at", "")
     if not target.has_column("deleted_by"):
         target.create_column_by_example("deleted_by", "")
     _drop_index(handle, f"idx_{table}_deleted")
     _index(
         handle,
         table,
         f"idx_{table}_trash",
         ["deleted_at"],
         where="deleted_at IS NOT NULL",
     )
 def soft_delete(table_name, deleted_by, *, stamp=None, **criteria):
     if table_name not in db.tables:
         return 0
     table = db[table_name]
     if not table.has_column("deleted_at"):
         return 0
     rows = list(table.find(deleted_at=None, **criteria))
     if not rows:
         return 0
     stamp = stamp or _now_iso()
     for row in rows:
         table.update(
             {"id": row["id"], "deleted_at": stamp, "deleted_by": deleted_by}, ["id"]
         )
     return len(rows)
 def soft_delete_in(table_name, column, uids, deleted_by, *, stamp=None, **extra):
     uids = [uid for uid in (uids or []) if uid]
     if not uids or table_name not in db.tables:
         return 0
     if not db[table_name].has_column("deleted_at"):
         return 0
     placeholders, params = _in_clause(uids)
     params["dat"] = stamp or _now_iso()
     params["dby"] = deleted_by
     extra_sql = ""
     for index, (key, value) in enumerate(extra.items()):
         params[f"x{index}"] = value
         extra_sql += f" AND {key} = :x{index}"
     db.query(
         f"UPDATE {table_name} SET deleted_at = :dat, deleted_by = :dby "
         f"WHERE {column} IN ({placeholders}) AND deleted_at IS NULL{extra_sql}",
         **params,
     )
     return len(uids)
 def restore(table_name, **criteria):
     if table_name not in db.tables:
         return 0
     table = db[table_name]
     if not table.has_column("deleted_at"):
         return 0
     rows = [row for row in table.find(**criteria) if row.get("deleted_at")]
     for row in rows:
         table.update({"id": row["id"], "deleted_at": None, "deleted_by": None}, ["id"])
     return len(rows)
 def purge(table_name, **criteria):
     if table_name not in db.tables:
         return 0
     table = db[table_name]
     count = table.count(**criteria)
     table.delete(**criteria)
     return int(count)
 def list_deleted(table_name, page=1, per_page=25):
     if table_name not in db.tables or not db[table_name].has_column("deleted_at"):
         return [], build_pagination(page, 0, per_page)
     table = db[table_name]
     column = table.table.columns.deleted_at
     total = table.count(column.isnot(None))
     pagination = build_pagination(page, total, per_page)
     offset = (pagination["page"] - 1) * pagination["per_page"]
     rows = list(
         table.find(
             column.isnot(None),
             order_by=["-deleted_at"],
             _limit=pagination["per_page"],
             _offset=offset,
         )
     )
     return rows, pagination
 def count_deleted(table_name):
     if table_name not in db.tables or not db[table_name].has_column("deleted_at"):
         return 0
     table = db[table_name]
     return int(table.count(table.table.columns.deleted_at.isnot(None)))
 def restore_event(stamp):
     if not stamp:
         return 0
     restored = 0
     for table_name in SOFT_DELETE_TABLES:
         if table_name in db.tables and db[table_name].has_column("deleted_at"):
             restored += int(
                 db.query(
                     f"SELECT COUNT(*) AS n FROM {table_name} WHERE deleted_at = :s",
                     s=stamp,
                 ).__next__()["n"]
             )
             db.query(
                 f"UPDATE {table_name} SET deleted_at = NULL, deleted_by = NULL "
                 f"WHERE deleted_at = :s",
                 s=stamp,
             )
     return restored
 def purge_event(stamp):
     if not stamp:
         return []
     purged = []
     for table_name in SOFT_DELETE_TABLES:
         if table_name in db.tables and db[table_name].has_column("deleted_at"):
             rows = list(
                 db.query(
                     f"SELECT * FROM {table_name} WHERE deleted_at = :s", s=stamp
                 )
             )
             if rows:
                 purged.append((table_name, rows))
                 db.query(
                     f"DELETE FROM {table_name} WHERE deleted_at = :s", s=stamp
                 )
     return purged