#!/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 add_menu():
    engine = create_async_engine(DATABASE_URL, echo=False)
    
    async with engine.connect() as conn:
        print("=" * 80)
        print("Adding Test Form Menu")
        print("=" * 80)
        
        # 找到测试配置管理目录ID (ID:2059)
        parent_id = 2059
        
        if not parent_id:
            print("[ERROR] Test Config directory not found")
            return
        
        print(f"[1] Found Test Config directory: {parent_id}")
        
        # 检查是否已存在
        result = await conn.execute(text("""
            SELECT COUNT(*) FROM sys_menu WHERE path = 'test_form'
        """))
        if result.scalar() > 0:
            print("[INFO] Test Form menu already exists")
            return
        
        # 添加测试表单管理菜单
        print("[2] Adding Test Form menu...")
        await conn.execute(text(f"""
            INSERT INTO sys_menu (menu_name, parent_id, order_num, path, component, 
                is_frame, is_cache, menu_type, visible, status, perms, icon, create_by, create_time)
            VALUES ('表单管理', {parent_id}, 6, 'test_form', 'system/test_form/index', 
                0, 0, 'C', '0', '0', 'system:test_form:list', 'el-icon-document', 'admin', NOW())
        """))
        await conn.commit()
        
        # 获取新菜单ID
        result = await conn.execute(text("""
            SELECT menu_id FROM sys_menu WHERE path = 'test_form'
        """))
        menu_id = result.scalar()
        
        # 添加按钮权限
        buttons = [
            ('表单查询', 'system:test_form:query'),
            ('表单新增', 'system:test_form:add'),
            ('表单修改', 'system:test_form:edit'),
            ('表单删除', 'system:test_form:remove'),
        ]
        
        for btn_name, btn_perm in buttons:
            await conn.execute(text(f"""
                INSERT INTO sys_menu (menu_name, parent_id, order_num, path, component, 
                    is_frame, is_cache, menu_type, visible, status, perms, icon, create_by, create_time)
                VALUES ('{btn_name}', {menu_id}, 1, '#', NULL, 
                    0, 0, 'F', '0', '0', '{btn_perm}', NULL, 'admin', NOW())
            """))
        await conn.commit()
        
        print(f"[OK] Test Form menu added (ID: {menu_id})")
        
        print("\n" + "=" * 80)
    
    await engine.dispose()

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