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

from backend.app import create_app
from backend.models import db, User

def create_initial_user():
    """创建初始用户"""
    app = create_app()

    with app.app_context():
        # 检查是否已有用户
        existing_user = User.query.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.session.add(user)
        db.session.commit()

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

if __name__ == '__main__':
    create_initial_user()