#!/usr/bin/env python3
"""
数据库初始化脚本
"""
import sys
import os

# 添加项目路径
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))

from app.database import init_db, SessionLocal
from app.init_data import init_default_data


def main():
    print("🚀 开始初始化数据库...")
    print("=" * 50)
    
    # 1. 创建表结构
    print("\n📦 创建数据库表...")
    init_db()
    print("✅ 数据库表创建完成")
    
    # 2. 初始化默认数据
    print("\n📝 初始化默认数据...")
    db = SessionLocal()
    try:
        init_default_data(db)
    finally:
        db.close()
    
    print("\n" + "=" * 50)
    print("🎉 数据库初始化完成！")
    print("\n你可以使用以下命令启动后端服务：")
    print("  uvicorn app.main:app --reload")


if __name__ == "__main__":
    main()