"""
初始化数据库表
"""
import sys
sys.path.append('.')

from app.database import Base, engine
from app.models.risk_detection import (
    DetectionRule,
    DetectionTask,
    RuleExecution,
    TaskExecution,
    DetectionResult
)
import asyncio


async def init_tables():
    """创建所有数据库表"""
    try:
        print("正在创建数据库表...")
        async with engine.begin() as conn:
            # 删除所有表（仅用于开发环境）
            await conn.run_sync(Base.metadata.drop_all)
            # 创建所有表
            await conn.run_sync(Base.metadata.create_all)
        print("✅ 数据库表创建成功！")
        return True
    except Exception as e:
        print(f"❌ 创建数据库表失败: {e}")
        print(f"错误详情: {type(e).__name__}")
        import traceback
        traceback.print_exc()
        return False


if __name__ == "__main__":
    success = asyncio.run(init_tables())
    sys.exit(0 if success else 1)