This commit is contained in:
aaron 2026-02-23 11:39:07 +08:00
parent 5a270542ea
commit 736717ba91

View File

@ -59,11 +59,18 @@ class RealTradingService:
from sqlalchemy import text from sqlalchemy import text
Base.metadata.create_all(bind=db_service.engine) Base.metadata.create_all(bind=db_service.engine)
# 创建自动交易开关表 # 创建自动交易开关表(使用简单的文本检查而不是 ORM
db = db_service.get_session() db = db_service.get_session()
try: try:
# 检查表是否存在
result = db.execute(text("""
SELECT name FROM sqlite_master WHERE type='table' AND name='real_trading_settings'
""")).fetchone()
if not result:
# 表不存在,创建表
db.execute(text(""" db.execute(text("""
CREATE TABLE IF NOT EXISTS real_trading_settings ( CREATE TABLE real_trading_settings (
key TEXT PRIMARY KEY, key TEXT PRIMARY KEY,
value TEXT NOT NULL, value TEXT NOT NULL,
updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
@ -73,10 +80,11 @@ class RealTradingService:
# 初始化自动交易开关 # 初始化自动交易开关
db.execute(text(""" db.execute(text("""
INSERT OR IGNORE INTO real_trading_settings (key, value) INSERT INTO real_trading_settings (key, value)
VALUES ('auto_trading_enabled', '0') VALUES ('auto_trading_enabled', '0')
""")) """))
db.commit() db.commit()
logger.info("创建实盘交易设置表")
except Exception as e: except Exception as e:
logger.warning(f"创建设置表失败: {e}") logger.warning(f"创建设置表失败: {e}")
db.rollback() db.rollback()