"""
检查和修复角色的 is_system 字段
"""
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 check_and_fix_roles():
    """检查和修复角色的 is_system 字段"""
    print("正在检查角色数据...")

    async with async_session() as session:
        # 查询所有角色
        result = await session.execute(
            text("SELECT id, role_name, role_code, is_system FROM roles ORDER BY id")
        )
        roles = result.fetchall()

        print("\n当前角色列表:")
        print("-" * 80)
        print(f"{'ID':<5} {'角色名称':<20} {'角色编码':<20} {'是否系统角色':<15}")
        print("-" * 80)
        for role in roles:
            is_system_text = "是" if role[3] == 1 else "否"
            print(f"{role[0]:<5} {role[1]:<20} {role[2]:<20} {is_system_text:<15}")
        print("-" * 80)

        # 修复建议：只有 super_admin 应该是系统角色
        print("\n开始修复角色 is_system 字段...")

        # 将 super_admin 设置为系统角色
        await session.execute(
            text("UPDATE roles SET is_system = 1 WHERE role_code = 'super_admin'")
        )
        print("✓ 已将 super_admin 设置为系统角色")

        # 将其他角色设置为非系统角色
        await session.execute(
            text("UPDATE roles SET is_system = 0 WHERE role_code != 'super_admin'")
        )
        print("✓ 已将其他角色设置为非系统角色")

        await session.commit()

        # 再次查询验证
        result = await session.execute(
            text("SELECT id, role_name, role_code, is_system FROM roles ORDER BY id")
        )
        roles = result.fetchall()

        print("\n修复后的角色列表:")
        print("-" * 80)
        print(f"{'ID':<5} {'角色名称':<20} {'角色编码':<20} {'是否系统角色':<15}")
        print("-" * 80)
        for role in roles:
            is_system_text = "是" if role[3] == 1 else "否"
            print(f"{role[0]:<5} {role[1]:<20} {role[2]:<20} {is_system_text:<15}")
        print("-" * 80)

        print("\n✓ 角色数据修复完成！")
        print("\n说明:")
        print("  - super_admin (超级管理员): 系统角色，不允许修改权限")
        print("  - admin (管理员): 非系统角色，可以修改权限")
        print("  - user (普通用户): 非系统角色，可以修改权限")


async def main():
    """主函数"""
    print("=" * 80)
    print("检查和修复角色 is_system 字段")
    print("=" * 80)
    print()

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