#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
创建初始用户脚本
用于创建管理员账户
"""

import sys
import os

# 将backend目录添加到Python路径
sys.path.insert(0, os.path.join(os.path.dirname(__file__), 'backend'))

from database import SessionLocal, init_db
from models import User

def create_initial_user():
    """创建初始用户"""
    # 初始化数据库
    init_db()

    # 创建会话
    db = SessionLocal()

    try:
        # 检查是否已有用户
        existing_user = db.query(User).first()

        if existing_user:
            print(f"用户已存在: {existing_user.username}")
            return

        # 创建默认管理员用户
        username = input("请输入用户名 (默认: admin): ").strip() or "admin"
        password = input("请输入密码 (默认: admin123): ").strip() or "admin123"

        user = User(username=username)
        user.set_password(password)

        db.add(user)
        db.commit()

        print(f"用户创建成功!")
        print(f"用户名: {username}")
        print(f"请妥善保管密码!")
    finally:
        db.close()

if __name__ == '__main__':
    create_initial_user()