Upload database_sql.py

database_sql.py

@@ -109,18 +109,17 @@ def init_sql_db():
 def _auto_migrate_p7_fields():
     """
     🔄 P7后悔模式：自动迁移新增字段
-    检查并添加 ownerships 表的新字段
+    检查并添加 ownerships 表和 transactions 表的新字段
     """
     from sqlalchemy import inspect
     
     try:
         inspector = inspect(engine)
         
-        # 检查 ownerships 表是否存在
+        # ========== 1. 迁移 ownerships 表 ==========
         if 'ownerships' in inspector.get_table_names():
             columns = [col['name'] for col in inspector.get_columns('ownerships')]
             
-            # 添加 P7 新增字段
             with engine.connect() as conn:
                 if 'price_paid' not in columns:
                     if 'sqlite' in SQLALCHEMY_DATABASE_URL:
@@ -141,8 +140,30 @@ def _auto_migrate_p7_fields():
                     logger.info("迁移完成: 添加 ownerships.refunded_at 字段")
                 
                 conn.commit()
+        
+        # ========== 2. 迁移 transactions 表（提现相关新字段） ==========
+        if 'transactions' in inspector.get_table_names():
+            columns = [col['name'] for col in inspector.get_columns('transactions')]
+            
+            with engine.connect() as conn:
+                # 定义 transactions 表的新列
+                new_columns = {
+                    'alipay_account': 'VARCHAR',
+                    'real_name': 'VARCHAR',
+                    'withdraw_status': 'VARCHAR',
+                    'payment_order_id': 'VARCHAR',
+                    'net_amount': 'INTEGER',
+                }
+                
+                for col_name, col_type in new_columns.items():
+                    if col_name not in columns:
+                        conn.execute(text(f"ALTER TABLE transactions ADD COLUMN {col_name} {col_type}"))
+                        logger.info(f"[DB Migration] 添加列 transactions.{col_name}")
+                
+                conn.commit()
+                
     except Exception as e:
-        logger.warning(f"P7字段迁移跳过 (可能已存在): {e}")
+        logger.warning(f"字段迁移跳过 (可能已存在): {e}")
 
 
 def get_db():