Maas-Model-platform/backend/scripts/drop_log_table_fkeys.py

#!/usr/bin/env python3
"""
清理 api_call_log 和 ai_message 表的所有外键约束

使用方法：
    python scripts/drop_log_table_fkeys.py
"""

import sys
from pathlib import Path

sys.path.append(str(Path(__file__).parent.parent))

from sqlalchemy import text
from app.database import engine

SQL_FILE = Path(__file__).parent.parent / "migrations" / "drop_log_table_fkeys.sql"


def main():
    print("开始清理外键约束...")

    sql = SQL_FILE.read_text(encoding="utf-8")

    try:
        with engine.begin() as conn:
            conn.execute(text(sql))
        print("✅ 外键清理完成")
        print("   - aigcspace.api_call_log 所有外键已移除")
        print("   - aigcspace.ai_message 所有外键已移除")
    except Exception as e:
        print(f"❌ 执行失败: {e}")
        sys.exit(1)


if __name__ == "__main__":
    main()