#!/usr/bin/env python3
"""
修复 alembic_version 表：如果数据库里记录了不存在的版本号（如 002），
将其重置为当前最新的迁移版本（如 001）。
用法（在项目根目录）：
  python deploy/scripts/fix_alembic_version.py
"""
import os
import sys

sys.path.insert(0, os.path.join(os.path.dirname(__file__), "..", "..", "backend"))

from dotenv import load_dotenv

load_dotenv(os.path.join(os.path.dirname(__file__), "..", "..", ".env"))

from sqlalchemy import create_engine, text

DATABASE_URL_SYNC = os.getenv(
    "DATABASE_URL_SYNC",
    "postgresql://wecom:wecom_secret@localhost:5432/wecom_ai",
)


def main():
    engine = create_engine(DATABASE_URL_SYNC)
    with engine.connect() as conn:
        # 检查 alembic_version 表
        try:
            r = conn.execute(text("SELECT version_num FROM alembic_version"))
            current = r.scalar_one_or_none()
            if current:
                print(f"当前数据库版本: {current}")
                if current == "002":
                    print("检测到版本 002，但本地只有 001。重置为 001...")
                    conn.execute(text("UPDATE alembic_version SET version_num = '001'"))
                    conn.commit()
                    print("已重置为 001。")
                else:
                    print(f"版本 {current} 正常，无需修复。")
            else:
                print("alembic_version 表为空，无需修复。")
        except Exception as e:
            if "does not exist" in str(e) or "relation" in str(e).lower():
                print("alembic_version 表不存在，这是首次迁移前的状态，正常。")
            else:
                print(f"错误: {e}")
                sys.exit(1)


if __name__ == "__main__":
    main()