feat(avatar): add user token accounting

2026-08-25 13:24:02 +08:00
parent 7a0199e685
commit 699bbbde57
15 changed files with 798 additions and 52 deletions
@@ -40,8 +40,14 @@ def init_db():
        ("authorizations", "takeover_mode", "VARCHAR DEFAULT 'immediate'"),
        ("authorizations", "takeover_delay_seconds", "INTEGER DEFAULT 30"),
        ("avatars", "share_token", "VARCHAR DEFAULT NULL"),
+        ("token_account", "user_id", "VARCHAR DEFAULT ''"),
+        ("token_account", "total_granted", "BIGINT DEFAULT 0"),
+        ("token_account", "total_consumed", "BIGINT DEFAULT 0"),
+        ("token_account", "created_at", "TIMESTAMP"),
+        ("token_account", "updated_at", "TIMESTAMP"),
    )
    _normalize_optional_unique_values()
+    _create_token_indexes()


 def _try_add_columns(*cols):
@@ -58,3 +64,11 @@ def _try_add_columns(*cols):
 def _normalize_optional_unique_values():
    with engine.begin() as conn:
        conn.exec_driver_sql("UPDATE avatars SET share_token = NULL WHERE share_token = ''")
+
+
+def _create_token_indexes():
+    with engine.begin() as conn:
+        conn.exec_driver_sql(
+            "CREATE UNIQUE INDEX IF NOT EXISTS ux_token_account_user_id "
+            "ON token_account(user_id) WHERE user_id <> ''"
+        )