#!/usr/bin/env python3
"""重置数据库脚本"""

from app import create_app
from database import db
from models import User, Server, Script, ExecuteHistory, ExecuteResult

def reset_database():
    """重置数据库"""
    app = create_app()
    
    with app.app_context():
        # 删除所有表
        db.drop_all()
        print("已删除所有表")
        
        # 创建所有表
        db.create_all()
        print("已创建所有表")
        
        # 创建默认管理员用户
        admin_user = User(
            username='admin',
            real_name='系统管理员',
            email='admin@example.com',
            role='admin'
        )
        admin_user.set_password('admin123')
        admin_user.set_permissions(['all'])
        
        db.session.add(admin_user)
        db.session.commit()
        print("已创建默认管理员用户 (admin/admin123)")

if __name__ == '__main__':
    reset_database()
    print("数据库重置完成！")