"""
更新系统管理菜单的路径
"""
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 update_menu_paths():
    """更新菜单路径"""
    print("正在更新系统管理菜单路径...")

    async with async_session() as session:
        # 1. 更新角色权限管理菜单路径
        result = await session.execute(
            text("""
                UPDATE system_menus
                SET path = '/system/permissions', component = 'System/Permissions'
                WHERE id = 14
            """)
        )
        print(f"✓ 角色权限管理菜单路径已更新: /system/permissions")

        # 2. 更新用户管理菜单路径
        result = await session.execute(
            text("""
                UPDATE system_menus
                SET path = '/system/users', component = 'System/Users'
                WHERE id = 15
            """)
        )
        print(f"✓ 用户管理菜单路径已更新: /system/users")

        # 3. 更新角色管理菜单路径
        result = await session.execute(
            text("""
                UPDATE system_menus
                SET path = '/system/roles', component = 'System/Roles'
                WHERE id = 16
            """)
        )
        print(f"✓ 角色管理菜单路径已更新: /system/roles")

        await session.commit()
        print("\n✓ 所有菜单路径更新完成！")


async def main():
    """主函数"""
    print("=" * 80)
    print("更新系统管理菜单路径")
    print("=" * 80)
    print()

    try:
        await update_menu_paths()
        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())