"""初始化 MySQL 数据库"""
import sys
import pymysql

# 先创建数据库（如果不存在）
conn = pymysql.connect(
    host='localhost',
    port=3306,
    user='root',
    password='',
    charset='utf8mb4'
)

try:
    with conn.cursor() as cursor:
        cursor.execute("CREATE DATABASE IF NOT EXISTS marketdata CHARACTER SET utf8mb4 COLLATE utf8mb4_unicode_ci")
        print("[OK] 数据库 'marketdata' 创建成功或已存在")
    conn.commit()
finally:
    conn.close()

# 使用 SQLAlchemy 创建表
import os
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))

from app.repositories.database import init_db, engine
from sqlalchemy import inspect

print("\n正在创建数据表...")
init_db()

# 验证表是否创建成功
inspector = inspect(engine)
tables = inspector.get_table_names()
print(f"\n[OK] 已创建 {len(tables)} 个表:")
for table in sorted(tables):
    print(f"   - {table}")

print("\n[OK] MySQL 数据库初始化完成！")