#!/usr/bin/env python3
"""
清理MySQL数据库，重新开始迁移
"""

import pymysql
import sys
from pathlib import Path
from loguru import logger

# 添加项目根目录到路径
current_dir = Path(__file__).parent
sys.path.insert(0, str(current_dir))

from config.mysql_config import MYSQL_CONFIG


def clean_mysql_database():
    """清理MySQL数据库"""
    logger.info("🧹 清理MySQL数据库...")

    try:
        with pymysql.connect(**MYSQL_CONFIG.to_dict()) as conn:
            cursor = conn.cursor()

            # 删除视图
            try:
                cursor.execute("DROP VIEW IF EXISTS latest_signals_view")
                cursor.execute("DROP VIEW IF EXISTS strategy_stats_view")
                logger.info("✅ 删除视图")
            except Exception as e:
                logger.warning(f"删除视图警告: {e}")

            # 删除表（注意外键约束顺序）
            tables = ['pullback_alerts', 'stock_signals', 'scan_sessions', 'strategies']

            for table in tables:
                try:
                    cursor.execute(f"DROP TABLE IF EXISTS {table}")
                    logger.info(f"✅ 删除表: {table}")
                except Exception as e:
                    logger.warning(f"删除表 {table} 警告: {e}")

            conn.commit()
            logger.info("✅ MySQL数据库清理完成")

    except Exception as e:
        logger.error(f"❌ 清理MySQL数据库失败: {e}")
        raise


if __name__ == "__main__":
    clean_mysql_database()