#!/usr/bin/env python3
"""
数据库初始化脚本
创建数据库表结构并初始化基础数据
"""
import sys
import os
from pathlib import Path

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

from app.database import DatabaseManager
from app.config import Config


def create_database():
    """创建数据库"""
    print("正在创建数据库...")

    # 创建数据库管理器，连接到MySQL服务器（不指定数据库）
    db_manager = DatabaseManager()

    try:
        with db_manager.get_connection() as conn:
            cursor = conn.cursor()

            # 创建数据库
            cursor.execute(f"CREATE DATABASE IF NOT EXISTS {Config.MYSQL_DATABASE} CHARACTER SET utf8mb4 COLLATE utf8mb4_unicode_ci")
            print(f"✓ 数据库 {Config.MYSQL_DATABASE} 创建成功")

            cursor.close()

    except Exception as e:
        print(f"✗ 创建数据库失败: {e}")
        return False

    return True


def create_tables():
    """创建数据表"""
    print("正在创建数据表...")

    # 读取SQL脚本
    schema_file = project_root / "database_schema.sql"
    if not schema_file.exists():
        print(f"✗ 数据库表结构文件不存在: {schema_file}")
        return False

    with open(schema_file, 'r', encoding='utf-8') as f:
        sql_content = f.read()

    db_manager = DatabaseManager()

    try:
        with db_manager.get_connection() as conn:
            cursor = conn.cursor()

            # 分割SQL语句并执行
            statements = [stmt.strip() for stmt in sql_content.split(';') if stmt.strip()]

            for statement in statements:
                if statement:
                    try:
                        cursor.execute(statement)
                    except Exception as e:
                        # 忽略表已存在的错误
                        if "already exists" not in str(e):
                            print(f"警告: 执行SQL语句失败: {statement[:50]}... 错误: {e}")

            conn.commit()
            print("✓ 数据表创建成功")
            cursor.close()

    except Exception as e:
        print(f"✗ 创建数据表失败: {e}")
        return False

    return True


def test_connection():
    """测试数据库连接"""
    print("正在测试数据库连接...")

    try:
        from app.dao import StockDAO, WatchlistDAO, AIAnalysisDAO, ConfigDAO

        # 测试各个DAO
        stock_dao = StockDAO()
        watchlist_dao = WatchlistDAO()
        ai_dao = AIAnalysisDAO()
        config_dao = ConfigDAO()

        # 获取数据库状态
        stock_count = stock_dao.get_stock_count()
        watchlist_count = watchlist_dao.get_watchlist_count()
        ai_count = ai_dao.get_analysis_count()

        print(f"✓ 数据库连接成功")
        print(f"  - 股票数量: {stock_count}")
        print(f"  - 监控列表: {watchlist_count}")
        print(f"  - AI分析: {ai_count}")

        return True

    except Exception as e:
        print(f"✗ 数据库连接失败: {e}")
        return False


def main():
    """主函数"""
    print("=" * 60)
    print("股票监控系统数据库初始化")
    print("=" * 60)
    print(f"数据库主机: {Config.MYSQL_HOST}:{Config.MYSQL_PORT}")
    print(f"数据库名称: {Config.MYSQL_DATABASE}")
    print(f"数据库用户: {Config.MYSQL_USER}")
    print("=" * 60)

    # 1. 创建数据库
    if not create_database():
        print("数据库创建失败，初始化终止")
        return False

    # 2. 创建数据表
    if not create_tables():
        print("数据表创建失败，初始化终止")
        return False

    # 3. 测试连接
    if not test_connection():
        print("数据库连接测试失败，初始化终止")
        return False

    print("\n" + "=" * 60)
    print("数据库初始化完成！")
    print("=" * 60)
    print("\n下一步操作：")
    print("1. 运行数据迁移脚本: python migrate_to_database.py")
    print("2. 启动应用系统")
    print("=" * 60)

    return True


if __name__ == "__main__":
    try:
        success = main()
        sys.exit(0 if success else 1)
    except KeyboardInterrupt:
        print("\n数据库初始化被用户中断")
        sys.exit(1)
    except Exception as e:
        print(f"\n数据库初始化过程中发生错误: {e}")
        sys.exit(1)