#!/usr/bin/env python3
"""恢复测试分类到基础数据管理下"""

import asyncio
from sqlalchemy.ext.asyncio import create_async_engine
from sqlalchemy import text

DB_CONFIG = {
    'host': '123.57.81.127',
    'port': 3306,
    'user': 'cpy_admin',
    'password': 'Tgzz2025+',
    'database': 'ruoyi-fastapi'
}

DATABASE_URL = (
    f"mysql+asyncmy://{DB_CONFIG['user']}:{DB_CONFIG['password']}@"
    f"{DB_CONFIG['host']}:{DB_CONFIG['port']}/{DB_CONFIG['database']}"
)

async def fix_test_category_position():
    engine = create_async_engine(DATABASE_URL, echo=False)
    
    async with engine.connect() as conn:
        print("=" * 80)
        print("Restoring Test Category Position")
        print("=" * 80)
        
        # 1. 找到基础数据管理目录
        print("\n[1] Finding Base Data Management directory...")
        result = await conn.execute(text("""
            SELECT menu_id, menu_name FROM sys_menu WHERE menu_id = 2036
        """))
        row = result.fetchone()
        if row:
            base_data_id = row[0]
            print(f"  Found: ID:{row[0]} | {row[1]}")
        else:
            print("  [ERROR] Base Data Management directory not found!")
            return
        
        # 2. 将测试分类移回基础数据管理下
        print("\n[2] Moving Test Category back to Base Data Management...")
        await conn.execute(text(f"""
            UPDATE sys_menu 
            SET parent_id = {base_data_id}, order_num = 1
            WHERE path = 'test_category' AND menu_type = 'C'
        """))
        await conn.commit()
        print("  [OK] Test Category moved back to Base Data Management")
        
        # 3. 验证
        print("\n[3] Verification:")
        result = await conn.execute(text(f"""
            SELECT m.menu_id, m.menu_name, m.path, m.order_num, p.menu_name as parent_name
            FROM sys_menu m
            LEFT JOIN sys_menu p ON m.parent_id = p.menu_id
            WHERE m.path = 'test_category'
        """))
        row = result.fetchone()
        if row:
            print(f"  Test Category: ID:{row[0]} | {row[1]}")
            print(f"    Parent: {row[4]} (ID: {row[0]})")
            print(f"    Order: {row[3]}")
        
        print("\n" + "=" * 80)
        print("Restore Complete")
        print("=" * 80)
    
    await engine.dispose()

if __name__ == '__main__':
    asyncio.run(fix_test_category_position())