110 lines
4.2 KiB
Python
110 lines
4.2 KiB
Python
|
#!/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 recreate_workorder_menu():
|
|||
|
|
engine = create_async_engine(DATABASE_URL, echo=False)
|
|||
|
|
|
|||
|
|
async with engine.connect() as conn:
|
|||
|
|
print("=" * 80)
|
|||
|
|
print("Recreating Work Order Claim Menu")
|
|||
|
|
print("=" * 80)
|
|||
|
|
|
|||
|
|
# 1. 找到测试工单目录（ID:2098）
|
|||
|
|
print("\n[1] Finding Test Work Order directory...")
|
|||
|
|
result = await conn.execute(text("""
|
|||
|
|
SELECT menu_id, menu_name, path FROM sys_menu WHERE menu_id = 2098
|
|||
|
|
"""))
|
|||
|
|
row = result.fetchone()
|
|||
|
|
if row:
|
|||
|
|
parent_id = row[0]
|
|||
|
|
print(f" Found: ID:{row[0]} | {row[1]} | Path:{row[2]}")
|
|||
|
|
else:
|
|||
|
|
print(" [ERROR] Test Work Order directory not found!")
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
# 2. 检查是否已存在工单领取菜单
|
|||
|
|
print("\n[2] Checking if work order claim menu exists...")
|
|||
|
|
result = await conn.execute(text("""
|
|||
|
|
SELECT COUNT(*) FROM sys_menu WHERE path = 'claim' AND parent_id = 2098
|
|||
|
|
"""))
|
|||
|
|
count = result.scalar()
|
|||
|
|
|
|||
|
|
if count > 0:
|
|||
|
|
print(" Work order claim menu already exists")
|
|||
|
|
else:
|
|||
|
|
# 3. 创建工单领取菜单
|
|||
|
|
print("\n[3] Creating work order claim 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 ('工单领取', 2098, 3, 'claim', 'workorder/claim/index',
|
|||
|
|
0, 0, 'C', '0', '0', 'workorder:claim:list', 'el-icon-s-claim', 'admin', NOW())
|
|||
|
|
"""))
|
|||
|
|
await conn.commit()
|
|||
|
|
print(" [OK] Work order claim menu created")
|
|||
|
|
|
|||
|
|
# 获取新创建的菜单ID
|
|||
|
|
result = await conn.execute(text("""
|
|||
|
|
SELECT menu_id FROM sys_menu WHERE path = 'claim' AND parent_id = 2098
|
|||
|
|
"""))
|
|||
|
|
claim_menu_id = result.scalar()
|
|||
|
|
|
|||
|
|
# 4. 创建按钮权限
|
|||
|
|
print("\n[4] Creating button permissions...")
|
|||
|
|
buttons = [
|
|||
|
|
('工单领取查询', 'workorder:claim:query', 1),
|
|||
|
|
('工单领取', 'workorder:claim:claim', 2),
|
|||
|
|
('批量领取', 'workorder:claim:batch', 3),
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
for btn_name, btn_perm, order 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}', {claim_menu_id}, {order}, '#', NULL,
|
|||
|
|
0, 0, 'F', '0', '0', '{btn_perm}', NULL, 'admin', NOW())
|
|||
|
|
"""))
|
|||
|
|
await conn.commit()
|
|||
|
|
print(" [OK] Button permissions created")
|
|||
|
|
|
|||
|
|
# 5. 验证
|
|||
|
|
print("\n[5] Verification:")
|
|||
|
|
result = await conn.execute(text("""
|
|||
|
|
SELECT m.menu_id, m.menu_name, m.path, m.component, m.menu_type, p.menu_name as parent_name
|
|||
|
|
FROM sys_menu m
|
|||
|
|
LEFT JOIN sys_menu p ON m.parent_id = p.menu_id
|
|||
|
|
WHERE m.parent_id = 2098
|
|||
|
|
ORDER BY m.order_num
|
|||
|
|
"""))
|
|||
|
|
rows = result.fetchall()
|
|||
|
|
print(f" Children of Test Work Order directory:")
|
|||
|
|
for row in rows:
|
|||
|
|
print(f" ID:{row[0]} | {row[1]} | Path:{row[2]} | Type:{row[4]}")
|
|||
|
|
|
|||
|
|
print("\n" + "=" * 80)
|
|||
|
|
print("Recreate Complete")
|
|||
|
|
print("=" * 80)
|
|||
|
|
print("\nPlease logout and login again to refresh the menu")
|
|||
|
|
|
|||
|
|
await engine.dispose()
|
|||
|
|
|
|||
|
|
if __name__ == '__main__':
|
|||
|
|
asyncio.run(recreate_workorder_menu())
|