update

2026-07-23 01:15:04 +02:00
parent 582e37d176
commit b534a496fd
79 changed files with 4086 additions and 0 deletions
@@ -0,0 +1,126 @@
+# retoor <retoor@molodetz.nl>
+
+from __future__ import annotations
+
+from datetime import date, datetime
+from pathlib import Path
+from typing import Any
+
+WRITE_PREFIXES = (
+    "set_",
+    "add_",
+    "delete_",
+    "create_",
+    "update_",
+    "upsert_",
+    "record_",
+    "migrate_",
+    "backfill_",
+    "mark_",
+    "invalidate_",
+    "ensure_",
+    "restore",
+    "purge",
+    "revoke_",
+    "recompute_",
+    "soft_delete",
+    "bump_",
+    "init_",
+    "clear_settings",
+)
+
+WRITE_EXACT = frozenset(
+    {
+        "soft_delete_in",
+        "restore_event",
+        "purge_event",
+        "delete_engagement",
+        "soft_delete_engagement",
+        "delete_fork_relations",
+        "soft_delete_fork_relations",
+        "delete_attachments",
+        "delete_attachment_record",
+        "_delete_attachment_file",
+        "_index",
+        "_drop_index",
+        "_uid_index",
+        "_ensure_cache_state",
+        "_refresh_query_planner_stats",
+        "_backfill_gamification",
+    }
+)
+
+REMOTE_TABLE_MARKER = "__remote_table__"
+
+
+def is_write(name: str) -> bool:
+    if name in WRITE_EXACT:
+        return True
+    return any(name.startswith(prefix) for prefix in WRITE_PREFIXES)
+
+
+SQL_WRITE_KEYWORDS = frozenset(
+    {"INSERT", "UPDATE", "DELETE", "REPLACE", "CREATE", "ALTER", "DROP"}
+)
+
+
+def is_write_sql(sql: str) -> bool:
+    first_word = sql.strip().split(None, 1)[0].upper() if sql.strip() else ""
+    return first_word in SQL_WRITE_KEYWORDS
+
+
+def encode_value(value: Any) -> Any:
+    if value is None or isinstance(value, (bool, int, float, str)):
+        return value
+    if type(value).__name__ == "RemoteTable":
+        return {REMOTE_TABLE_MARKER: value._name}
+    if isinstance(value, (datetime, date)):
+        return value.isoformat()
+    if isinstance(value, Path):
+        return str(value)
+    if isinstance(value, frozenset):
+        return [encode_value(item) for item in value]
+    if isinstance(value, set):
+        return [encode_value(item) for item in value]
+    if isinstance(value, tuple):
+        return [encode_value(item) for item in value]
+    if isinstance(value, list):
+        return [encode_value(item) for item in value]
+    if isinstance(value, dict):
+        return {str(key): encode_value(item) for key, item in value.items()}
+    if hasattr(value, "items") and callable(value.items):
+        try:
+            return {str(key): encode_value(item) for key, item in value.items()}
+        except TypeError:
+            pass
+    return str(value)
+
+
+def encode_args(args: tuple | list, kwargs: dict) -> tuple[list, dict]:
+    return [encode_value(item) for item in args], {
+        str(key): encode_value(value) for key, value in kwargs.items()
+    }
+
+
+def decode_value(value: Any) -> Any:
+    if isinstance(value, list):
+        return [decode_value(item) for item in value]
+    if isinstance(value, dict):
+        return {key: decode_value(item) for key, item in value.items()}
+    return value
+
+
+def decode_arg(value: Any) -> Any:
+    if isinstance(value, list):
+        return [decode_arg(item) for item in value]
+    if isinstance(value, dict):
+        if set(value) == {REMOTE_TABLE_MARKER}:
+            from devplacepy.db_client import get_table
+
+            return get_table(value[REMOTE_TABLE_MARKER])
+        return {key: decode_arg(item) for key, item in value.items()}
+    return value
+
+
+def encode_result(value: Any) -> Any:
+    return encode_value(value)