74 lines
2.3 KiB
Python
74 lines
2.3 KiB
Python
|
"""迁移脚本：添加 baodan_conversation_id 字段到会话表。"""
|
|||
|
|
import logging
|
|||
|
|
from insurance.db.compat import db
|
|||
|
|
|
|||
|
|
|
|||
|
|
def migrate():
|
|||
|
|
"""添加 baodan_conversation_id 字段到 insurance_chat_sessions 表。"""
|
|||
|
|
try:
|
|||
|
|
from sqlalchemy import inspect
|
|||
|
|
|
|||
|
|
# 检查字段是否已存在
|
|||
|
|
inspector = inspect(db.engine)
|
|||
|
|
columns = [col['name'] for col in inspector.get_columns('insurance_chat_sessions')]
|
|||
|
|
if 'baodan_conversation_id' in columns:
|
|||
|
|
logging.info("字段 baodan_conversation_id 已存在，跳过添加")
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
# 根据数据库方言执行不同的 SQL
|
|||
|
|
dialect = db.engine.dialect.name
|
|||
|
|
if dialect == 'sqlite':
|
|||
|
|
# SQLite 不支持 COMMENT 语法
|
|||
|
|
db.session.execute(
|
|||
|
|
"ALTER TABLE insurance_chat_sessions "
|
|||
|
|
"ADD COLUMN baodan_conversation_id VARCHAR(128)"
|
|||
|
|
)
|
|||
|
|
else:
|
|||
|
|
# PostgreSQL
|
|||
|
|
db.session.execute(
|
|||
|
|
"ALTER TABLE insurance_chat_sessions "
|
|||
|
|
"ADD COLUMN baodan_conversation_id VARCHAR(128) NULL "
|
|||
|
|
"COMMENT 'BaoDan 返回的 conversation_id，用于保持上下文'"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
db.session.commit()
|
|||
|
|
logging.info("成功添加 baodan_conversation_id 字段")
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
logging.error(f"添加字段失败: {e}")
|
|||
|
|
db.session.rollback()
|
|||
|
|
raise
|
|||
|
|
|
|||
|
|
|
|||
|
|
def downgrade():
|
|||
|
|
"""删除 baodan_conversation_id 字段。"""
|
|||
|
|
try:
|
|||
|
|
from sqlalchemy import inspect
|
|||
|
|
|
|||
|
|
inspector = inspect(db.engine)
|
|||
|
|
columns = [col['name'] for col in inspector.get_columns('insurance_chat_sessions')]
|
|||
|
|
if 'baodan_conversation_id' not in columns:
|
|||
|
|
logging.info("字段 baodan_conversation_id 不存在，跳过删除")
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
db.session.execute(
|
|||
|
|
"ALTER TABLE insurance_chat_sessions DROP COLUMN baodan_conversation_id"
|
|||
|
|
)
|
|||
|
|
db.session.commit()
|
|||
|
|
logging.info("成功删除 baodan_conversation_id 字段")
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
logging.error(f"删除字段失败: {e}")
|
|||
|
|
db.session.rollback()
|
|||
|
|
raise
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
# 直接运行迁移
|
|||
|
|
import sys
|
|||
|
|
sys.path.insert(0, ".")
|
|||
|
|
from app import app
|
|||
|
|
with app.app_context():
|
|||
|
|
migrate()
|
|||
|
|
print("迁移完成")
|