from app import create_app, db
from models import User

def init_database():
    """初始化数据库和创建默认管理员用户"""
    app = create_app()
    
    with app.app_context():
        # 创建所有表
        db.create_all()
        
        # 检查是否已有管理员用户
        admin = User.query.filter_by(username='admin').first()
        if not admin:
            # 创建默认管理员
            admin = User(
                username='admin',
                real_name='系统管理员',
                email='admin@example.com',
                role='admin'
            )
            admin.set_password('admin123')
            admin.set_permissions([
                'server.view', 'server.manage',
                'script.view', 'script.manage',
                'execute.run', 'user.manage'
            ])
            
            db.session.add(admin)
            db.session.commit()
            
            print('默认管理员用户创建成功')
            print('用户名: admin')
            print('密码: admin123')
        else:
            print('管理员用户已存在')

if __name__ == '__main__':
    init_database()