"""
管理员用户初始化脚本

创建默认管理员用户，如果已存在则更新密码和APIkey
"""
import sys
from pathlib import Path

sys.path.insert(0, str(Path(__file__).parent.parent))

from app.database import SessionLocal
from app.services.user_service import UserService


def init_admin():
    """初始化管理员用户"""
    db = SessionLocal()
    
    try:
        user_service = UserService(db)
        user_service.init_admin_user()
        print("管理员用户初始化完成")
        print("  用户名: admin")
        print("  密码: admin123")
        print("  邮箱: wxcz@wxcz.com")
        print("  APIkey: sk-db36651a5eb147919ce1035e2b471971")
    except Exception as e:
        print(f"初始化失败: {e}")
    finally:
        db.close()


if __name__ == "__main__":
    init_admin()