nex_docus/backend/scripts/add_user_role_menus.py

"""
添加用户管理和角色管理菜单到数据库
"""
import sys
import asyncio
from pathlib import Path

# 添加项目根目录到 Python 路径
sys.path.insert(0, str(Path(__file__).parent.parent))

from sqlalchemy import text
from app.core.database import async_session


async def add_user_role_menus():
    """添加用户管理和角色管理菜单"""
    print("正在添加用户管理和角色管理菜单...")

    async with async_session() as session:
        # 1. 检查菜单是否已存在
        result = await session.execute(
            text("SELECT COUNT(*) FROM system_menus WHERE id IN (15, 16)")
        )
        count = result.scalar()

        if count > 0:
            print(f"  菜单已存在（{count}个），跳过添加")
            return

        # 2. 插入用户管理菜单（ID: 15）
        await session.execute(
            text("""
                INSERT INTO system_menus (
                    id, parent_id, menu_name, menu_code, menu_type,
                    path, component, icon, sort_order, visible, status,
                    created_at, updated_at
                ) VALUES (
                    15, 4, '用户管理', 'system:users', 1,
                    '/users', 'UserManagement', 'UserOutlined',
                    1, 1, 1, NOW(), NOW()
                )
            """)
        )
        print("✓ 用户管理菜单添加成功（ID: 15）")

        # 3. 插入角色管理菜单（ID: 16）
        await session.execute(
            text("""
                INSERT INTO system_menus (
                    id, parent_id, menu_name, menu_code, menu_type,
                    path, component, icon, sort_order, visible, status,
                    created_at, updated_at
                ) VALUES (
                    16, 4, '角色管理', 'system:roles', 1,
                    '/roles', 'RoleManagement', 'TeamOutlined',
                    2, 1, 1, NOW(), NOW()
                )
            """)
        )
        print("✓ 角色管理菜单添加成功（ID: 16）")

        # 4. 更新已有菜单的sort_order，避免冲突
        await session.execute(
            text("""
                UPDATE system_menus
                SET sort_order = sort_order + 2
                WHERE parent_id = 4 AND id NOT IN (15, 16) AND sort_order >= 1
            """)
        )
        print("✓ 已调整其他子菜单的排序")

        # 5. 为超级管理员角色分配这两个菜单权限
        result = await session.execute(
            text("SELECT id FROM roles WHERE role_code = 'super_admin'")
        )
        super_admin_role_id = result.scalar()

        if super_admin_role_id:
            # 用户管理菜单
            await session.execute(
                text("""
                    INSERT INTO role_menus (role_id, menu_id, created_at)
                    VALUES (:role_id, 15, NOW())
                """),
                {"role_id": super_admin_role_id}
            )
            # 角色管理菜单
            await session.execute(
                text("""
                    INSERT INTO role_menus (role_id, menu_id, created_at)
                    VALUES (:role_id, 16, NOW())
                """),
                {"role_id": super_admin_role_id}
            )
            print(f"✓ 已为超级管理员角色分配权限")

        # 6. 为管理员角色分配这两个菜单权限（如果存在）
        result = await session.execute(
            text("SELECT id FROM roles WHERE role_code = 'admin'")
        )
        admin_role_id = result.scalar()

        if admin_role_id:
            # 用户管理菜单
            await session.execute(
                text("""
                    INSERT INTO role_menus (role_id, menu_id, created_at)
                    VALUES (:role_id, 15, NOW())
                """),
                {"role_id": admin_role_id}
            )
            # 角色管理菜单
            await session.execute(
                text("""
                    INSERT INTO role_menus (role_id, menu_id, created_at)
                    VALUES (:role_id, 16, NOW())
                """),
                {"role_id": admin_role_id}
            )
            print(f"✓ 已为管理员角色分配权限")

        await session.commit()
        print("\n✓ 用户管理和角色管理菜单添加完成！")


async def main():
    """主函数"""
    print("=" * 80)
    print("添加用户管理和角色管理菜单")
    print("=" * 80)
    print()

    try:
        await add_user_role_menus()
        print()
        print("=" * 80)
        print("✓ 操作完成！现在可以在系统管理菜单中访问用户管理和角色管理了")
        print("=" * 80)
    except Exception as e:
        print(f"\n✗ 操作失败: {str(e)}")
        import traceback
        traceback.print_exc()
        sys.exit(1)


if __name__ == "__main__":
    asyncio.run(main())