"""手动添加 baodan_conversation_id 字段到会话表"""
import psycopg2

# 数据库连接配置
DB_CONFIG = {
    "host": "localhost",
    "port": 5432,
    "database": "baodan",
    "user": "postgres",
    "password": "taiyi1224",
}

def add_column():
    """添加 baodan_conversation_id 字段。"""
    conn = None
    try:
        # 连接数据库
        conn = psycopg2.connect(**DB_CONFIG)
        cur = conn.cursor()

        # 检查字段是否已存在
        cur.execute("""
            SELECT column_name
            FROM information_schema.columns
            WHERE table_name = 'insurance_chat_sessions'
            AND column_name = 'baodan_conversation_id'
        """)
        if cur.fetchone():
            print("字段 baodan_conversation_id 已存在，跳过添加")
            return

        # 添加新字段
        cur.execute("""
            ALTER TABLE insurance_chat_sessions
            ADD COLUMN baodan_conversation_id VARCHAR(128) NULL
        """)
        conn.commit()

        print("成功添加 baodan_conversation_id 字段")

        # 验证字段已添加
        cur.execute("""
            SELECT column_name, data_type, is_nullable
            FROM information_schema.columns
            WHERE table_name = 'insurance_chat_sessions'
            AND column_name = 'baodan_conversation_id'
        """)
        result = cur.fetchone()
        if result:
            print(f"验证成功: 字段={result[0]}, 类型={result[1]}, 可为空={result[2]}")

    except Exception as e:
        print(f"添加字段失败: {e}")
        if conn:
            conn.rollback()
        raise
    finally:
        if conn:
            conn.close()

if __name__ == "__main__":
    add_column()