#!/usr/bin/env python3
"""
数据库初始化脚本
用于创建数据库表和运行初始迁移
"""

import asyncio
import sys
import os

# 添加项目根目录到Python路径
sys.path.append(os.path.dirname(__file__))

from app.core.database import init_db, close_db
from app.core.config import settings


async def main():
    """主函数：初始化数据库"""
    try:
        print(f"正在连接数据库: {settings.DATABASE_URL}")
        print("正在创建数据库表...")
        
        # 初始化数据库表
        await init_db()
        
        print("✅ 数据库表创建成功!")
        print("💡 提示: 如果需要使用Alembic管理迁移，请运行:")
        print("   alembic stamp head")
        print("   alembic revision --autogenerate -m '描述'")
        print("   alembic upgrade head")
        
    except Exception as e:
        print(f"❌ 数据库初始化失败: {e}")
        sys.exit(1)
    finally:
        await close_db()


if __name__ == "__main__":
    asyncio.run(main())