From 33e43789ce5bea31a347bb4bff84eebd3f9f660a Mon Sep 17 00:00:00 2001 From: aaron <> Date: Tue, 3 Mar 2026 16:51:17 +0800 Subject: [PATCH] update --- .../scripts/migrate_add_margin_leverage.py | 22 ++++++++++++++++--- frontend/trading.html | 15 +++++++++++-- 2 files changed, 32 insertions(+), 5 deletions(-) diff --git a/backend/scripts/migrate_add_margin_leverage.py b/backend/scripts/migrate_add_margin_leverage.py index 65cbfb4..4cb8428 100644 --- a/backend/scripts/migrate_add_margin_leverage.py +++ b/backend/scripts/migrate_add_margin_leverage.py @@ -26,16 +26,32 @@ def check_column_exists(table_name, column_name): return column_name in columns +def create_tables_if_not_exist(): + """创建表(如果不存在)""" + from app.models.paper_trading import PaperOrder as POTable + from app.models.database import Base + + # 创建所有表 + Base.metadata.create_all(bind=db_service.engine) + logger.info("✅ 数据库表已创建") + + def migrate(): """执行迁移""" try: - with db_service.engine.connect() as conn: - # 检查表是否存在 + # 首先确保表存在 + inspector = inspect(db_service.engine) + if 'paper_orders' not in inspector.get_table_names(): + logger.info("⚠️ paper_orders 表不存在,先创建表...") + create_tables_if_not_exist() + # 重新检查 inspector = inspect(db_service.engine) if 'paper_orders' not in inspector.get_table_names(): - logger.error("❌ paper_orders 表不存在") + logger.error("❌ 创建表失败") return False + with db_service.engine.connect() as conn: + logger.info("开始迁移 paper_orders 表...") # 检查并添加 margin 列 diff --git a/frontend/trading.html b/frontend/trading.html index 16a1dd8..45ffc73 100644 --- a/frontend/trading.html +++ b/frontend/trading.html @@ -1380,6 +1380,11 @@