#!/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


def apply_extended_schema():
    """应用扩展的数据库结构"""
    print("正在应用扩展的数据库结构...")

    # 读取SQL脚本
    schema_file = project_root / "database_schema_extended.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)
                        print(f"✓ 执行成功: {statement[:50]}...")
                    except Exception as e:
                        # 忽略表已存在的错误
                        if "already exists" not in str(e) and "Duplicate" not in str(e):
                            print(f"⚠️ 警告: 执行SQL语句失败: {statement[:50]}... 错误: {e}")

            conn.commit()
            print("✓ 扩展数据库结构应用成功")
            cursor.close()

        # 验证新表是否创建成功
        verify_tables_created()

        return True

    except Exception as e:
        print(f"✗ 应用扩展数据库结构失败: {e}")
        return False


def verify_tables_created():
    """验证新表是否创建成功"""
    print("正在验证新表是否创建成功...")

    db_manager = DatabaseManager()
    new_tables = [
        'industries', 'sectors', 'kline_data', 'stock_sector_relations',
        'market_statistics', 'data_update_tasks', 'hot_stocks'
    ]

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

            for table_name in new_tables:
                cursor.execute(f"SHOW TABLES LIKE '{table_name}'")
                exists = cursor.fetchone()

                if exists:
                    print(f"✓ 表 {table_name} 创建成功")
                else:
                    print(f"✗ 表 {table_name} 创建失败")

            cursor.close()

    except Exception as e:
        print(f"✗ 验证表创建失败: {e}")


def main():
    """主函数"""
    print("=" * 60)
    print("应用扩展数据库结构")
    print("=" * 60)

    success = apply_extended_schema()

    if success:
        print("\n" + "=" * 60)
        print("扩展数据库结构应用完成！")
        print("=" * 60)
        print("\n下一步操作：")
        print("1. 启动应用系统: python run.py")
        print("2. 访问 http://localhost:8000/api/market/stocks 查看股票列表")
        print("3. 访问 http://localhost:8000/api/market/sync 同步市场数据")
        print("=" * 60)
    else:
        print("✗ 扩展数据库结构应用失败，请检查错误信息")

    return success


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)