54 lines
1.6 KiB
Python
54 lines
1.6 KiB
Python
|
|
import asyncio
|
|
import sys
|
|
import os
|
|
|
|
# 将项目根目录添加到 python 路径
|
|
project_root = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
|
|
sys.path.append(project_root)
|
|
|
|
from Config.Config import DB_URL
|
|
from DbKit.Db import Db
|
|
from sqlalchemy import text
|
|
import logging
|
|
|
|
logging.basicConfig(level=logging.INFO)
|
|
logger = logging.getLogger(__name__)
|
|
|
|
async def update_schema():
|
|
"""
|
|
为 t_station_status_scd 表添加 market_price 字段
|
|
"""
|
|
db = Db(db_url=DB_URL)
|
|
await db.init_db()
|
|
|
|
# 检查字段是否已存在
|
|
check_sql = "SHOW COLUMNS FROM t_station_status_scd LIKE 'market_price';"
|
|
|
|
# 添加字段的 SQL
|
|
alter_sql = "ALTER TABLE t_station_status_scd ADD COLUMN market_price DECIMAL(10, 4) COMMENT '挂牌价快照' AFTER pro_price;"
|
|
|
|
async with await db.get_session() as session:
|
|
try:
|
|
result = await session.execute(text(check_sql))
|
|
column_exists = result.fetchone() is not None
|
|
|
|
if not column_exists:
|
|
logger.info("正在为 t_station_status_scd 添加 market_price 字段...")
|
|
await session.execute(text(alter_sql))
|
|
await session.commit()
|
|
logger.info("字段 market_price 添加成功。")
|
|
else:
|
|
logger.info("字段 market_price 已存在，跳过。")
|
|
|
|
except Exception as e:
|
|
logger.error(f"执行 ALTER TABLE 时出错: {e}")
|
|
await session.rollback()
|
|
finally:
|
|
await session.close()
|
|
|
|
await db.engine.dispose()
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(update_schema())
|