#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
创建管理员账户脚本
"""
from src.flask_prompt_master import create_app, db
from src.flask_prompt_master.admin.models.admin_user import AdminUser

def create_admin_user():
    """创建管理员用户"""
    app = create_app()
    
    with app.app_context():
        try:
            # 检查是否已存在管理员
            admin = AdminUser.query.filter_by(username='admin').first()
            if admin:
                print("✅ 管理员账户已存在")
                return
            
            # 创建管理员账户
            admin = AdminUser(
                username='admin',
                password='admin123',
                email='admin@example.com',
                role='super_admin'
            )
            
            db.session.add(admin)
            db.session.commit()
            
            print("✅ 管理员账户创建成功")
            print("用户名: admin")
            print("密码: admin123")
            print("请及时修改默认密码！")
            
        except Exception as e:
            print(f"❌ 创建管理员账户失败: {str(e)}")
            db.session.rollback()

if __name__ == '__main__':
    create_admin_user()