"""使用 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()