LQAdminPlatform/test/reinit_rbac_data_fixed.py

#!/usr/bin/env python3
"""
重新初始化RBAC数据 - 修复菜单层级关系混乱问题
清晰的三级结构：父目录菜单 > 功能菜单 > 按钮权限
"""
import pymysql
from dotenv import load_dotenv
import os
import uuid
from datetime import datetime

load_dotenv()

def get_db_connection():
    """获取数据库连接"""
    try:
        config = {
            'host': os.getenv('DB_HOST', 'localhost'),
            'port': int(os.getenv('DB_PORT', 3306)),
            'user': os.getenv('DB_USER', 'root'),
            'password': os.getenv('DB_PASSWORD', 'admin'),
            'database': os.getenv('DB_NAME', 'lq_db'),
            'charset': 'utf8mb4',
            'autocommit': True
        }
        return pymysql.connect(**config)
    except Exception as e:
        print(f"数据库连接失败: {e}")
        return None

def clear_rbac_data(cursor):
    """清理现有RBAC数据"""
    print("🧹 清理现有RBAC数据...")
    
    # 清理关联表
    cursor.execute("DELETE FROM role_permissions")
    cursor.execute("DELETE FROM role_menus")
    cursor.execute("DELETE FROM user_roles")
    
    # 清理主表
    cursor.execute("DELETE FROM permissions")
    cursor.execute("DELETE FROM menus")
    cursor.execute("DELETE FROM roles")
    
    print("   ✅ 清理完成")

def insert_menus(cursor):
    """插入菜单数据 - 清晰的三级结构"""
    print("📋 插入菜单数据（修复层级关系）...")
    
    menus_data = [
        # ==================== 第一级：主菜单（父目录菜单） ====================
        {
            'id': 'dashboard-main',
            'parent_id': None,
            'name': 'dashboard',
            'title': '仪表盘',
            'path': '/dashboard',
            'component': 'dashboard/Index',
            'icon': 'House',
            'sort_order': 1,
            'menu_type': 'menu',
            'is_hidden': False,
            'description': '系统概览和统计信息'
        },
        {
            'id': 'profile-main',
            'parent_id': None,
            'name': 'profile',
            'title': '个人资料',
            'path': '/profile',
            'component': 'user/Profile',
            'icon': 'User',
            'sort_order': 2,
            'menu_type': 'menu',
            'is_hidden': False,
            'description': '个人信息管理'
        },
        {
            'id': 'admin-main',
            'parent_id': None,
            'name': 'admin',
            'title': '系统管理',
            'path': '/admin',
            'component': None,
            'icon': 'Setting',
            'sort_order': 3,
            'menu_type': 'menu',
            'is_hidden': False,
            'description': '系统管理功能目录'
        },
        
        # ==================== 第二级：功能菜单 ====================
        # 系统管理下的功能菜单
        {
            'id': 'admin-dashboard',
            'parent_id': 'admin-main',
            'name': 'admin-dashboard',
            'title': '管理概览',
            'path': '/admin/dashboard',
            'component': 'admin/Dashboard',
            'icon': 'Monitor',
            'sort_order': 1,
            'menu_type': 'menu',
            'is_hidden': False,
            'description': '管理员仪表盘'
        },
        {
            'id': 'user-management',
            'parent_id': 'admin-main',
            'name': 'user-management',
            'title': '用户管理',
            'path': '/admin/users',
            'component': 'admin/Users',
            'icon': 'UserFilled',
            'sort_order': 2,
            'menu_type': 'menu',
            'is_hidden': False,
            'description': '用户信息管理'
        },
        {
            'id': 'role-management',
            'parent_id': 'admin-main',
            'name': 'role-management',
            'title': '角色管理',
            'path': '/admin/roles',
            'component': 'admin/Roles',
            'icon': 'Avatar',
            'sort_order': 3,
            'menu_type': 'menu',
            'is_hidden': False,
            'description': '角色权限管理'
        },
        {
            'id': 'menu-management',
            'parent_id': 'admin-main',
            'name': 'menu-management',
            'title': '菜单管理',
            'path': '/admin/menus',
            'component': 'admin/Menus',
            'icon': 'Menu',
            'sort_order': 4,
            'menu_type': 'menu',
            'is_hidden': False,
            'description': '菜单结构管理'
        },
        {
            'id': 'permission-management',
            'parent_id': 'admin-main',
            'name': 'permission-management',
            'title': '权限管理',
            'path': '/admin/permissions',
            'component': 'admin/Permissions',
            'icon': 'Key',
            'sort_order': 5,
            'menu_type': 'menu',
            'is_hidden': False,
            'description': '权限配置管理'
        },
        {
            'id': 'app-management',
            'parent_id': 'admin-main',
            'name': 'app-management',
            'title': '应用管理',
            'path': '/admin/apps',
            'component': 'admin/Apps',
            'icon': 'Grid',
            'sort_order': 6,
            'menu_type': 'menu',
            'is_hidden': False,
            'description': 'OAuth2应用管理'
        },
        {
            'id': 'log-management',
            'parent_id': 'admin-main',
            'name': 'log-management',
            'title': '系统日志',
            'path': '/admin/logs',
            'component': 'admin/Logs',
            'icon': 'Document',
            'sort_order': 7,
            'menu_type': 'menu',
            'is_hidden': False,
            'description': '系统操作日志'
        },
        {
            'id': 'system-settings',
            'parent_id': 'admin-main',
            'name': 'system-settings',
            'title': '系统设置',
            'path': '/admin/settings',
            'component': 'admin/Settings',
            'icon': 'Tools',
            'sort_order': 8,
            'menu_type': 'menu',
            'is_hidden': False,
            'description': '系统配置设置'
        },
        
        # ==================== 第三级：按钮权限 ====================
        # 用户管理的按钮权限
        {
            'id': 'user-create-btn',
            'parent_id': 'user-management',
            'name': 'user-create',
            'title': '创建用户',
            'path': None,
            'component': None,
            'icon': 'Plus',
            'sort_order': 1,
            'menu_type': 'button',
            'is_hidden': False,
            'description': '创建新用户'
        },
        {
            'id': 'user-edit-btn',
            'parent_id': 'user-management',
            'name': 'user-edit',
            'title': '编辑用户',
            'path': None,
            'component': None,
            'icon': 'Edit',
            'sort_order': 2,
            'menu_type': 'button',
            'is_hidden': False,
            'description': '编辑用户信息'
        },
        {
            'id': 'user-delete-btn',
            'parent_id': 'user-management',
            'name': 'user-delete',
            'title': '删除用户',
            'path': None,
            'component': None,
            'icon': 'Delete',
            'sort_order': 3,
            'menu_type': 'button',
            'is_hidden': False,
            'description': '删除用户'
        },
        {
            'id': 'user-batch-delete-btn',
            'parent_id': 'user-management',
            'name': 'user-batch-delete',
            'title': '批量删除用户',
            'path': None,
            'component': None,
            'icon': 'DeleteFilled',
            'sort_order': 4,
            'menu_type': 'button',
            'is_hidden': False,
            'description': '批量删除用户'
        },
        {
            'id': 'user-assign-role-btn',
            'parent_id': 'user-management',
            'name': 'user-assign-role',
            'title': '分配角色',
            'path': None,
            'component': None,
            'icon': 'Avatar',
            'sort_order': 5,
            'menu_type': 'button',
            'is_hidden': False,
            'description': '为用户分配角色'
        },
        {
            'id': 'user-reset-password-btn',
            'parent_id': 'user-management',
            'name': 'user-reset-password',
            'title': '重置密码',
            'path': None,
            'component': None,
            'icon': 'Key',
            'sort_order': 6,
            'menu_type': 'button',
            'is_hidden': False,
            'description': '重置用户密码'
        },
        
        # 角色管理的按钮权限
        {
            'id': 'role-create-btn',
            'parent_id': 'role-management',
            'name': 'role-create',
            'title': '创建角色',
            'path': None,
            'component': None,
            'icon': 'Plus',
            'sort_order': 1,
            'menu_type': 'button',
            'is_hidden': False,
            'description': '创建新角色'
        },
        {
            'id': 'role-edit-btn',
            'parent_id': 'role-management',
            'name': 'role-edit',
            'title': '编辑角色',
            'path': None,
            'component': None,
            'icon': 'Edit',
            'sort_order': 2,
            'menu_type': 'button',
            'is_hidden': False,
            'description': '编辑角色信息'
        },
        {
            'id': 'role-delete-btn',
            'parent_id': 'role-management',
            'name': 'role-delete',
            'title': '删除角色',
            'path': None,
            'component': None,
            'icon': 'Delete',
            'sort_order': 3,
            'menu_type': 'button',
            'is_hidden': False,
            'description': '删除角色'
        },
        {
            'id': 'role-assign-permission-btn',
            'parent_id': 'role-management',
            'name': 'role-assign-permission',
            'title': '分配权限',
            'path': None,
            'component': None,
            'icon': 'Key',
            'sort_order': 4,
            'menu_type': 'button',
            'is_hidden': False,
            'description': '为角色分配权限'
        },
        {
            'id': 'role-assign-menu-btn',
            'parent_id': 'role-management',
            'name': 'role-assign-menu',
            'title': '分配菜单',
            'path': None,
            'component': None,
            'icon': 'Menu',
            'sort_order': 5,
            'menu_type': 'button',
            'is_hidden': False,
            'description': '为角色分配菜单'
        },
        
        # 菜单管理的按钮权限
        {
            'id': 'menu-create-btn',
            'parent_id': 'menu-management',
            'name': 'menu-create',
            'title': '创建菜单',
            'path': None,
            'component': None,
            'icon': 'Plus',
            'sort_order': 1,
            'menu_type': 'button',
            'is_hidden': False,
            'description': '创建新菜单'
        },
        {
            'id': 'menu-edit-btn',
            'parent_id': 'menu-management',
            'name': 'menu-edit',
            'title': '编辑菜单',
            'path': None,
            'component': None,
            'icon': 'Edit',
            'sort_order': 2,
            'menu_type': 'button',
            'is_hidden': False,
            'description': '编辑菜单信息'
        },
        {
            'id': 'menu-delete-btn',
            'parent_id': 'menu-management',
            'name': 'menu-delete',
            'title': '删除菜单',
            'path': None,
            'component': None,
            'icon': 'Delete',
            'sort_order': 3,
            'menu_type': 'button',
            'is_hidden': False,
            'description': '删除菜单'
        },
        {
            'id': 'menu-sort-btn',
            'parent_id': 'menu-management',
            'name': 'menu-sort',
            'title': '排序菜单',
            'path': None,
            'component': None,
            'icon': 'Sort',
            'sort_order': 4,
            'menu_type': 'button',
            'is_hidden': False,
            'description': '调整菜单排序'
        },
        
        # 权限管理的按钮权限
        {
            'id': 'permission-create-btn',
            'parent_id': 'permission-management',
            'name': 'permission-create',
            'title': '创建权限',
            'path': None,
            'component': None,
            'icon': 'Plus',
            'sort_order': 1,
            'menu_type': 'button',
            'is_hidden': False,
            'description': '创建新权限'
        },
        {
            'id': 'permission-edit-btn',
            'parent_id': 'permission-management',
            'name': 'permission-edit',
            'title': '编辑权限',
            'path': None,
            'component': None,
            'icon': 'Edit',
            'sort_order': 2,
            'menu_type': 'button',
            'is_hidden': False,
            'description': '编辑权限信息'
        },
        {
            'id': 'permission-delete-btn',
            'parent_id': 'permission-management',
            'name': 'permission-delete',
            'title': '删除权限',
            'path': None,
            'component': None,
            'icon': 'Delete',
            'sort_order': 3,
            'menu_type': 'button',
            'is_hidden': False,
            'description': '删除权限'
        },
        
        # 应用管理的按钮权限
        {
            'id': 'app-create-btn',
            'parent_id': 'app-management',
            'name': 'app-create',
            'title': '创建应用',
            'path': None,
            'component': None,
            'icon': 'Plus',
            'sort_order': 1,
            'menu_type': 'button',
            'is_hidden': False,
            'description': '创建OAuth2应用'
        },
        {
            'id': 'app-edit-btn',
            'parent_id': 'app-management',
            'name': 'app-edit',
            'title': '编辑应用',
            'path': None,
            'component': None,
            'icon': 'Edit',
            'sort_order': 2,
            'menu_type': 'button',
            'is_hidden': False,
            'description': '编辑应用信息'
        },
        {
            'id': 'app-delete-btn',
            'parent_id': 'app-management',
            'name': 'app-delete',
            'title': '删除应用',
            'path': None,
            'component': None,
            'icon': 'Delete',
            'sort_order': 3,
            'menu_type': 'button',
            'is_hidden': False,
            'description': '删除应用'
        },
        {
            'id': 'app-view-secret-btn',
            'parent_id': 'app-management',
            'name': 'app-view-secret',
            'title': '查看密钥',
            'path': None,
            'component': None,
            'icon': 'View',
            'sort_order': 4,
            'menu_type': 'button',
            'is_hidden': False,
            'description': '查看应用密钥'
        },
        {
            'id': 'app-reset-secret-btn',
            'parent_id': 'app-management',
            'name': 'app-reset-secret',
            'title': '重置密钥',
            'path': None,
            'component': None,
            'icon': 'RefreshRight',
            'sort_order': 5,
            'menu_type': 'button',
            'is_hidden': False,
            'description': '重置应用密钥'
        },
        
        # 系统日志的按钮权限
        {
            'id': 'log-view-btn',
            'parent_id': 'log-management',
            'name': 'log-view',
            'title': '查看日志',
            'path': None,
            'component': None,
            'icon': 'View',
            'sort_order': 1,
            'menu_type': 'button',
            'is_hidden': False,
            'description': '查看系统日志'
        },
        {
            'id': 'log-export-btn',
            'parent_id': 'log-management',
            'name': 'log-export',
            'title': '导出日志',
            'path': None,
            'component': None,
            'icon': 'Download',
            'sort_order': 2,
            'menu_type': 'button',
            'is_hidden': False,
            'description': '导出系统日志'
        },
        {
            'id': 'log-clear-btn',
            'parent_id': 'log-management',
            'name': 'log-clear',
            'title': '清理日志',
            'path': None,
            'component': None,
            'icon': 'Delete',
            'sort_order': 3,
            'menu_type': 'button',
            'is_hidden': False,
            'description': '清理历史日志'
        },
        
        # 系统设置的按钮权限
        {
            'id': 'settings-view-btn',
            'parent_id': 'system-settings',
            'name': 'settings-view',
            'title': '查看设置',
            'path': None,
            'component': None,
            'icon': 'View',
            'sort_order': 1,
            'menu_type': 'button',
            'is_hidden': False,
            'description': '查看系统设置'
        },
        {
            'id': 'settings-edit-btn',
            'parent_id': 'system-settings',
            'name': 'settings-edit',
            'title': '修改设置',
            'path': None,
            'component': None,
            'icon': 'Edit',
            'sort_order': 2,
            'menu_type': 'button',
            'is_hidden': False,
            'description': '修改系统设置'
        }
    ]
    
    for menu in menus_data:
        cursor.execute("""
            INSERT INTO menus (id, parent_id, name, title, path, component, icon, sort_order, menu_type, is_hidden, is_active, description, created_at, updated_at)
            VALUES (%s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, NOW(), NOW())
        """, (
            menu['id'], menu['parent_id'], menu['name'], menu['title'], 
            menu['path'], menu['component'], menu['icon'], menu['sort_order'],
            menu['menu_type'], menu['is_hidden'], True, menu['description']
        ))
    
    print(f"   ✅ 插入了 {len(menus_data)} 个菜单项")
    
    # 统计各级菜单数量
    main_menus = len([m for m in menus_data if not m['parent_id']])
    function_menus = len([m for m in menus_data if m['parent_id'] and m['menu_type'] == 'menu'])
    button_permissions = len([m for m in menus_data if m['menu_type'] == 'button'])
    
    print(f"   📊 菜单结构统计:")
    print(f"      📁 主菜单（父目录）: {main_menus} 个")
    print(f"      📄 功能菜单: {function_menus} 个")
    print(f"      🔘 按钮权限: {button_permissions} 个")

def insert_roles(cursor):
    """插入角色数据"""
    print("👥 插入角色数据...")
    
    roles_data = [
        {
            'id': str(uuid.uuid4()),
            'name': 'super_admin',
            'display_name': '超级管理员',
            'description': '拥有系统所有权限的超级管理员',
            'is_system': True
        },
        {
            'id': str(uuid.uuid4()),
            'name': 'admin',
            'display_name': '管理员',
            'description': '系统管理员，拥有大部分管理权限',
            'is_system': True
        },
        {
            'id': str(uuid.uuid4()),
            'name': 'user_manager',
            'display_name': '用户管理员',
            'description': '负责用户和角色管理的管理员',
            'is_system': True
        },
        {
            'id': str(uuid.uuid4()),
            'name': 'app_manager',
            'display_name': '应用管理员',
            'description': '负责应用管理的管理员',
            'is_system': True
        },
        {
            'id': str(uuid.uuid4()),
            'name': 'user',
            'display_name': '普通用户',
            'description': '系统普通用户',
            'is_system': True
        }
    ]
    
    role_ids = {}
    for role in roles_data:
        cursor.execute("""
            INSERT INTO roles (id, name, display_name, description, is_active, is_system, created_at, updated_at)
            VALUES (%s, %s, %s, %s, %s, %s, NOW(), NOW())
        """, (role['id'], role['name'], role['display_name'], role['description'], True, role['is_system']))
        role_ids[role['name']] = role['id']
    
    print(f"   ✅ 插入了 {len(roles_data)} 个角色")
    return role_ids

def insert_permissions(cursor):
    """插入权限数据"""
    print("🔐 插入权限数据...")
    
    permissions_data = [
        # 用户管理权限
        {'name': 'user.view', 'display_name': '查看用户', 'resource': 'user', 'action': 'view', 'description': '查看用户列表和详情'},
        {'name': 'user.create', 'display_name': '创建用户', 'resource': 'user', 'action': 'create', 'description': '创建新用户'},
        {'name': 'user.edit', 'display_name': '编辑用户', 'resource': 'user', 'action': 'edit', 'description': '编辑用户信息'},
        {'name': 'user.delete', 'display_name': '删除用户', 'resource': 'user', 'action': 'delete', 'description': '删除用户'},
        {'name': 'user.batch_delete', 'display_name': '批量删除用户', 'resource': 'user', 'action': 'batch_delete', 'description': '批量删除用户'},
        {'name': 'user.assign_role', 'display_name': '分配用户角色', 'resource': 'user', 'action': 'assign_role', 'description': '为用户分配角色'},
        {'name': 'user.reset_password', 'display_name': '重置用户密码', 'resource': 'user', 'action': 'reset_password', 'description': '重置用户密码'},
        
        # 角色管理权限
        {'name': 'role.view', 'display_name': '查看角色', 'resource': 'role', 'action': 'view', 'description': '查看角色列表和详情'},
        {'name': 'role.create', 'display_name': '创建角色', 'resource': 'role', 'action': 'create', 'description': '创建新角色'},
        {'name': 'role.edit', 'display_name': '编辑角色', 'resource': 'role', 'action': 'edit', 'description': '编辑角色信息'},
        {'name': 'role.delete', 'display_name': '删除角色', 'resource': 'role', 'action': 'delete', 'description': '删除角色'},
        {'name': 'role.assign_permission', 'display_name': '分配角色权限', 'resource': 'role', 'action': 'assign_permission', 'description': '为角色分配权限'},
        {'name': 'role.assign_menu', 'display_name': '分配角色菜单', 'resource': 'role', 'action': 'assign_menu', 'description': '为角色分配菜单'},
        
        # 菜单管理权限
        {'name': 'menu.view', 'display_name': '查看菜单', 'resource': 'menu', 'action': 'view', 'description': '查看菜单列表和详情'},
        {'name': 'menu.create', 'display_name': '创建菜单', 'resource': 'menu', 'action': 'create', 'description': '创建新菜单'},
        {'name': 'menu.edit', 'display_name': '编辑菜单', 'resource': 'menu', 'action': 'edit', 'description': '编辑菜单信息'},
        {'name': 'menu.delete', 'display_name': '删除菜单', 'resource': 'menu', 'action': 'delete', 'description': '删除菜单'},
        {'name': 'menu.sort', 'display_name': '排序菜单', 'resource': 'menu', 'action': 'sort', 'description': '调整菜单排序'},
        
        # 权限管理权限
        {'name': 'permission.view', 'display_name': '查看权限', 'resource': 'permission', 'action': 'view', 'description': '查看权限列表和详情'},
        {'name': 'permission.create', 'display_name': '创建权限', 'resource': 'permission', 'action': 'create', 'description': '创建新权限'},
        {'name': 'permission.edit', 'display_name': '编辑权限', 'resource': 'permission', 'action': 'edit', 'description': '编辑权限信息'},
        {'name': 'permission.delete', 'display_name': '删除权限', 'resource': 'permission', 'action': 'delete', 'description': '删除权限'},
        
        # 应用管理权限
        {'name': 'app.view', 'display_name': '查看应用', 'resource': 'app', 'action': 'view', 'description': '查看应用列表和详情'},
        {'name': 'app.create', 'display_name': '创建应用', 'resource': 'app', 'action': 'create', 'description': '创建OAuth2应用'},
        {'name': 'app.edit', 'display_name': '编辑应用', 'resource': 'app', 'action': 'edit', 'description': '编辑应用信息'},
        {'name': 'app.delete', 'display_name': '删除应用', 'resource': 'app', 'action': 'delete', 'description': '删除应用'},
        {'name': 'app.view_secret', 'display_name': '查看应用密钥', 'resource': 'app', 'action': 'view_secret', 'description': '查看应用密钥'},
        {'name': 'app.reset_secret', 'display_name': '重置应用密钥', 'resource': 'app', 'action': 'reset_secret', 'description': '重置应用密钥'},
        
        # 系统管理权限
        {'name': 'system.dashboard', 'display_name': '管理概览', 'resource': 'system', 'action': 'dashboard', 'description': '查看管理概览'},
        {'name': 'system.log_view', 'display_name': '查看系统日志', 'resource': 'system', 'action': 'log_view', 'description': '查看系统操作日志'},
        {'name': 'system.log_export', 'display_name': '导出系统日志', 'resource': 'system', 'action': 'log_export', 'description': '导出系统日志'},
        {'name': 'system.log_clear', 'display_name': '清理系统日志', 'resource': 'system', 'action': 'log_clear', 'description': '清理历史日志'},
        {'name': 'system.settings_view', 'display_name': '查看系统设置', 'resource': 'system', 'action': 'settings_view', 'description': '查看系统配置'},
        {'name': 'system.settings_edit', 'display_name': '修改系统设置', 'resource': 'system', 'action': 'settings_edit', 'description': '修改系统配置'},
    ]
    
    permission_ids = {}
    for perm in permissions_data:
        perm_id = str(uuid.uuid4())
        cursor.execute("""
            INSERT INTO permissions (id, name, display_name, resource, action, description, is_active, created_at, updated_at)
            VALUES (%s, %s, %s, %s, %s, %s, %s, NOW(), NOW())
        """, (perm_id, perm['name'], perm['display_name'], perm['resource'], perm['action'], perm['description'], True))
        permission_ids[perm['name']] = perm_id
    
    print(f"   ✅ 插入了 {len(permissions_data)} 个权限")
    return permission_ids

def assign_role_permissions(cursor, role_ids, permission_ids):
    """分配角色权限"""
    print("🔗 分配角色权限...")
    
    # 超级管理员拥有所有权限
    super_admin_id = role_ids['super_admin']
    for perm_name, perm_id in permission_ids.items():
        cursor.execute("""
            INSERT INTO role_permissions (role_id, permission_id, created_at)
            VALUES (%s, %s, NOW())
        """, (super_admin_id, perm_id))
    
    # 管理员拥有大部分权限（除了系统设置修改）
    admin_id = role_ids['admin']
    admin_permissions = [perm for perm in permission_ids.keys() if not perm.startswith('system.settings_edit')]
    for perm_name in admin_permissions:
        cursor.execute("""
            INSERT INTO role_permissions (role_id, permission_id, created_at)
            VALUES (%s, %s, NOW())
        """, (admin_id, permission_ids[perm_name]))
    
    # 用户管理员只有用户和角色相关权限
    user_manager_id = role_ids['user_manager']
    user_permissions = [perm for perm in permission_ids.keys() if perm.startswith(('user.', 'role.', 'system.dashboard'))]
    for perm_name in user_permissions:
        cursor.execute("""
            INSERT INTO role_permissions (role_id, permission_id, created_at)
            VALUES (%s, %s, NOW())
        """, (user_manager_id, permission_ids[perm_name]))
    
    # 应用管理员只有应用相关权限
    app_manager_id = role_ids['app_manager']
    app_permissions = [perm for perm in permission_ids.keys() if perm.startswith(('app.', 'system.dashboard'))]
    for perm_name in app_permissions:
        cursor.execute("""
            INSERT INTO role_permissions (role_id, permission_id, created_at)
            VALUES (%s, %s, NOW())
        """, (app_manager_id, permission_ids[perm_name]))
    
    print("   ✅ 角色权限分配完成")

def assign_role_menus(cursor, role_ids):
    """分配角色菜单"""
    print("🔗 分配角色菜单...")
    
    # 获取所有菜单ID
    cursor.execute("SELECT id, parent_id, menu_type FROM menus")
    all_menus = cursor.fetchall()
    menu_ids = [menu[0] for menu in all_menus]
    
    # 超级管理员和管理员拥有所有菜单
    for role_name in ['super_admin', 'admin']:
        role_id = role_ids[role_name]
        for menu_id in menu_ids:
            cursor.execute("""
                INSERT INTO role_menus (role_id, menu_id, created_at)
                VALUES (%s, %s, NOW())
            """, (role_id, menu_id))
    
    # 用户管理员只有基础菜单和用户管理相关菜单
    user_manager_id = role_ids['user_manager']
    user_manager_menus = [
        'dashboard-main', 'profile-main', 'admin-main', 'admin-dashboard', 
        'user-management', 'role-management'
    ]
    # 添加用户管理和角色管理相关的按钮权限
    cursor.execute("SELECT id FROM menus WHERE parent_id IN ('user-management', 'role-management')")
    user_buttons = cursor.fetchall()
    user_manager_menus.extend([btn[0] for btn in user_buttons])
    
    for menu_id in user_manager_menus:
        cursor.execute("""
            INSERT INTO role_menus (role_id, menu_id, created_at)
            VALUES (%s, %s, NOW())
        """, (user_manager_id, menu_id))
    
    # 应用管理员只有基础菜单和应用管理相关菜单
    app_manager_id = role_ids['app_manager']
    app_manager_menus = [
        'dashboard-main', 'profile-main', 'admin-main', 'admin-dashboard', 
        'app-management'
    ]
    # 添加应用管理相关的按钮权限
    cursor.execute("SELECT id FROM menus WHERE parent_id = 'app-management'")
    app_buttons = cursor.fetchall()
    app_manager_menus.extend([btn[0] for btn in app_buttons])
    
    for menu_id in app_manager_menus:
        cursor.execute("""
            INSERT INTO role_menus (role_id, menu_id, created_at)
            VALUES (%s, %s, NOW())
        """, (app_manager_id, menu_id))
    
    # 普通用户只有基础菜单
    user_id = role_ids['user']
    user_menus = ['dashboard-main', 'profile-main']
    for menu_id in user_menus:
        cursor.execute("""
            INSERT INTO role_menus (role_id, menu_id, created_at)
            VALUES (%s, %s, NOW())
        """, (user_id, menu_id))
    
    print("   ✅ 角色菜单分配完成")

def assign_user_roles(cursor, role_ids):
    """为现有用户分配角色"""
    print("👤 为用户分配角色...")
    
    # 获取admin用户
    cursor.execute("SELECT id FROM users WHERE username = 'admin'")
    admin_user = cursor.fetchone()
    if admin_user:
        admin_user_id = admin_user[0]
        # 为admin用户分配超级管理员角色
        cursor.execute("""
            INSERT INTO user_roles (user_id, role_id, is_active, created_at)
            VALUES (%s, %s, %s, NOW())
        """, (admin_user_id, role_ids['super_admin'], True))
        print(f"   ✅ 为admin用户分配了超级管理员角色")
    
    # 获取其他用户并分配普通用户角色
    cursor.execute("SELECT id, username FROM users WHERE username != 'admin'")
    other_users = cursor.fetchall()
    for user in other_users:
        user_id, username = user
        cursor.execute("""
            INSERT INTO user_roles (user_id, role_id, is_active, created_at)
            VALUES (%s, %s, %s, NOW())
        """, (user_id, role_ids['user'], True))
        print(f"   ✅ 为用户 {username} 分配了普通用户角色")

def print_menu_structure(cursor):
    """打印菜单结构验证"""
    print("\n🌳 菜单结构验证:")
    print("=" * 60)
    
    # 获取所有菜单
    cursor.execute("""
        SELECT id, parent_id, title, menu_type, sort_order 
        FROM menus 
        ORDER BY 
            CASE WHEN parent_id IS NULL THEN 0 ELSE 1 END,
            sort_order, 
            CASE WHEN menu_type = 'menu' THEN 0 ELSE 1 END
    """)
    all_menus = cursor.fetchall()
    
    # 构建菜单树
    menu_dict = {}
    for menu in all_menus:
        menu_id, parent_id, title, menu_type, sort_order = menu
        menu_dict[menu_id] = {
            'id': menu_id,
            'parent_id': parent_id,
            'title': title,
            'menu_type': menu_type,
            'sort_order': sort_order,
            'children': []
        }
    
    # 建立父子关系
    root_menus = []
    for menu_id, menu_data in menu_dict.items():
        if menu_data['parent_id']:
            if menu_data['parent_id'] in menu_dict:
                menu_dict[menu_data['parent_id']]['children'].append(menu_data)
        else:
            root_menus.append(menu_data)
    
    # 打印树形结构
    def print_menu_tree(menus, level=0):
        for menu in sorted(menus, key=lambda x: x['sort_order']):
            indent = "  " * level
            if menu['menu_type'] == 'menu' and level == 0:
                icon = "📁"
            elif menu['menu_type'] == 'menu':
                icon = "📄"
            else:
                icon = "🔘"
            
            print(f"{indent}{icon} {menu['title']} ({menu['menu_type']})")
            
            if menu['children']:
                # 先显示菜单类型的子项，再显示按钮类型的子项
                menu_children = [c for c in menu['children'] if c['menu_type'] == 'menu']
                button_children = [c for c in menu['children'] if c['menu_type'] == 'button']
                print_menu_tree(menu_children, level + 1)
                print_menu_tree(button_children, level + 1)
    
    print_menu_tree(root_menus)

def reinit_rbac_data():
    """重新初始化RBAC数据 - 修复层级关系"""
    print("🚀 开始重新初始化RBAC数据（修复层级关系）")
    print("=" * 60)
    
    conn = get_db_connection()
    if not conn:
        print("❌ 数据库连接失败")
        return False
    
    cursor = conn.cursor()
    
    try:
        # 1. 清理现有数据
        clear_rbac_data(cursor)
        
        # 2. 插入菜单数据（修复层级关系）
        insert_menus(cursor)
        
        # 3. 插入角色数据
        role_ids = insert_roles(cursor)
        
        # 4. 插入权限数据
        permission_ids = insert_permissions(cursor)
        
        # 5. 分配角色权限
        assign_role_permissions(cursor, role_ids, permission_ids)
        
        # 6. 分配角色菜单
        assign_role_menus(cursor, role_ids)
        
        # 7. 为用户分配角色
        assign_user_roles(cursor, role_ids)
        
        # 8. 打印菜单结构验证
        print_menu_structure(cursor)
        
        conn.commit()
        
        print("\n" + "=" * 60)
        print("🎉 RBAC数据重新初始化完成！（层级关系已修复）")
        print("=" * 60)
        print("📊 数据统计:")
        
        # 统计数据
        cursor.execute("SELECT COUNT(*) FROM menus")
        menu_count = cursor.fetchone()[0]
        cursor.execute("SELECT COUNT(*) FROM menus WHERE menu_type = 'menu' AND parent_id IS NULL")
        main_menu_count = cursor.fetchone()[0]
        cursor.execute("SELECT COUNT(*) FROM menus WHERE menu_type = 'menu' AND parent_id IS NOT NULL")
        function_menu_count = cursor.fetchone()[0]
        cursor.execute("SELECT COUNT(*) FROM menus WHERE menu_type = 'button'")
        button_count = cursor.fetchone()[0]
        cursor.execute("SELECT COUNT(*) FROM roles")
        role_count = cursor.fetchone()[0]
        cursor.execute("SELECT COUNT(*) FROM permissions")
        permission_count = cursor.fetchone()[0]
        
        print(f"   📁 主菜单（父目录）: {main_menu_count} 个")
        print(f"   📄 功能菜单: {function_menu_count} 个")
        print(f"   🔘 按钮权限: {button_count} 个")
        print(f"   📋 菜单总数: {menu_count} 个")
        print(f"   👥 角色数量: {role_count} 个")
        print(f"   🔐 权限数量: {permission_count} 个")
        print("=" * 60)
        
        return True
        
    except Exception as e:
        print(f"❌ 初始化失败: {e}")
        conn.rollback()
        return False
    finally:
        cursor.close()
        conn.close()

def main():
    """主函数"""
    success = reinit_rbac_data()
    if success:
        print("\n🎯 重新初始化成功！菜单层级关系已修复。")
        print("💡 建议重启前端服务以刷新菜单缓存。")
        print("\n🌐 访问地址:")
        print("   前端: http://localhost:3000")
        print("   菜单管理: http://localhost:3000/admin/menus")
        print("\n🔑 登录信息:")
        print("   用户名: admin")
        print("   密码: Admin123456")
    else:
        print("\n❌ 重新初始化失败！")

if __name__ == "__main__":
    main()