HAHAHHA

models/add_tg_web_app_data_column.py

@@ -0,0 +1,55 @@
+#!/usr/bin/env python3
+"""
+为 tg_phone_devices_copy1 表添加 tg_web_app_data 列（若不存在）。
+解决: Unknown column 't1.tg_web_app_data' in 'field list'
+"""
+
+import sys
+from pathlib import Path
+
+project_root = Path(__file__).resolve().parent.parent
+if str(project_root) not in sys.path:
+    sys.path.insert(0, str(project_root))
+
+from models.tg_models import db
+
+TABLE = "tg_phone_devices_copy1"
+COLUMN = "tg_web_app_data"
+SPEC = "VARCHAR(2048) NULL COMMENT 'Blum WebApp tgWebAppData token'"
+
+
+def column_exists(cursor, table, column):
+    cursor.execute(
+        """
+        SELECT COUNT(*) FROM information_schema.COLUMNS
+        WHERE TABLE_SCHEMA = DATABASE()
+          AND TABLE_NAME = %s
+          AND COLUMN_NAME = %s
+        """,
+        (table, column),
+    )
+    return cursor.fetchone()[0] > 0
+
+
+def main():
+    if not db.is_closed():
+        db.connect()
+    try:
+        with db.connection() as conn:
+            cursor = conn.cursor()
+            if column_exists(cursor, TABLE, COLUMN):
+                print(f"表 {TABLE} 已存在列 {COLUMN}，无需添加。")
+                return
+            sql = f"ALTER TABLE `{TABLE}` ADD COLUMN `{COLUMN}` {SPEC};"
+            cursor.execute(sql)
+            print(f"已在表 {TABLE} 添加列 {COLUMN}。")
+    finally:
+        if not db.is_closed():
+            try:
+                db.close()
+            except Exception:
+                pass
+
+
+if __name__ == "__main__":
+    main()