"""migrations/data_fixes.py — 数据修正迁移（修复脏数据、变更枚举值）"""


def migrate_fix_task_status():
    """修正 project_tasks 中非法的 status 值"""
    from flask_app import db, _exec, mysql, logger

    conn = db()
    try:
        fixes = [
            "UPDATE project_tasks SET status='未开始' WHERE status='' OR status IS NULL",
            "UPDATE project_tasks SET status='已结束' WHERE status='done'",
            "UPDATE project_tasks SET status='进行中' WHERE status='验收中'",
        ]
        for sql in fixes:
            try:
                cur = _exec(conn, sql)
                affected = cur.rowcount
                cur.close()
                if affected:
                    print(f"[migrate] 修正 {affected} 条任务状态")
            except mysql.connector.Error as e:
                logger.warning(f"task status fix skipped: {e}")
        conn.commit()
    finally:
        conn.close()


def migrate_rename_tenant():
    """工作台重命名：无界·无界 → 学会·无界"""
    from flask_app import db, _exec, mysql

    conn = db()
    try:
        tables = ["user_tenants", "sales_leads", "follow_up_records", "business_proposals",
                  "operation_projects", "product_versions", "finance_records", "project_tasks",
                  "project_finances"]
        for table in tables:
            try:
                cur = _exec(conn, f"UPDATE {table} SET tenant='学会·无界' WHERE tenant='无界·无界'")
                affected = cur.rowcount
                cur.close()
                if affected:
                    print(f"[migrate] {table}: {affected} 条记录 tenant 改为 '学会·无界'")
            except mysql.connector.Error:
                pass
        conn.commit()
    finally:
        conn.close()


def migrate_drop_product_fields():
    """删除 product_versions 表的 owner / platform / feature_list 字段"""
    from flask_app import db, mysql

    conn = db()
    try:
        for col in ["owner", "platform", "feature_list"]:
            cur = conn.cursor(dictionary=True)
            cur.execute("SHOW COLUMNS FROM product_versions LIKE %s", (col,))
            exists = cur.fetchone()
            cur.close()
            if exists:
                try:
                    cur = conn.cursor()
                    cur.execute(f"ALTER TABLE product_versions DROP COLUMN {col}")
                    cur.close()
                    conn.commit()
                    print(f"[migrate] product_versions.{col} 列已删除")
                except mysql.connector.Error as e:
                    print(f"[migrate] 删除 {col} 失败: {e}")
    finally:
        conn.close()