58 lines
2.0 KiB
Python
58 lines
2.0 KiB
Python
"""使用 Flask 应用执行数据库迁移（兼容 SQLite 和 PostgreSQL）"""
|
||
import sys
|
||
import os
|
||
|
||
# 添加项目根目录到 Python 路径
|
||
sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..', 'api'))
|
||
|
||
from insurance.app import app
|
||
from insurance.db.compat import db
|
||
|
||
def add_column():
|
||
"""添加 baodan_conversation_id 字段。"""
|
||
with app.app_context():
|
||
try:
|
||
from sqlalchemy import text, inspect
|
||
|
||
# 获取数据库方言
|
||
dialect = db.engine.dialect.name
|
||
print(f"数据库方言: {dialect}")
|
||
|
||
# 检查字段是否已存在
|
||
inspector = inspect(db.engine)
|
||
columns = [col['name'] for col in inspector.get_columns('insurance_chat_sessions')]
|
||
if 'baodan_conversation_id' in columns:
|
||
print("字段 baodan_conversation_id 已存在，跳过添加")
|
||
return
|
||
|
||
# 根据数据库类型执行不同的 SQL
|
||
if dialect == 'sqlite':
|
||
# SQLite 使用 ALTER TABLE ADD COLUMN
|
||
db.session.execute(text(
|
||
"ALTER TABLE insurance_chat_sessions "
|
||
"ADD COLUMN baodan_conversation_id VARCHAR(128)"
|
||
))
|
||
else:
|
||
# PostgreSQL 使用 ALTER TABLE ADD COLUMN
|
||
db.session.execute(text(
|
||
"ALTER TABLE insurance_chat_sessions "
|
||
"ADD COLUMN baodan_conversation_id VARCHAR(128) NULL"
|
||
))
|
||
|
||
db.session.commit()
|
||
print("成功添加 baodan_conversation_id 字段")
|
||
|
||
# 验证字段已添加
|
||
inspector = inspect(db.engine)
|
||
columns = [col['name'] for col in inspector.get_columns('insurance_chat_sessions')]
|
||
if 'baodan_conversation_id' in columns:
|
||
print("验证成功: 字段已添加到数据库")
|
||
|
||
except Exception as e:
|
||
print(f"添加字段失败: {e}")
|
||
db.session.rollback()
|
||
raise
|
||
|
||
if __name__ == "__main__":
|
||
add_column()
|