125 lines
3.4 KiB
Python
125 lines
3.4 KiB
Python
#!/usr/bin/env python
|
||
# -*- coding: utf-8 -*-
|
||
import pymysql
|
||
|
||
# 数据库连接配置
|
||
db_config = {
|
||
'host': '1.15.149.240',
|
||
'user': 'zhibo',
|
||
'password': 'zCETFpGMwYN3CNeH',
|
||
'database': 'zhibo',
|
||
'charset': 'utf8mb4'
|
||
}
|
||
|
||
# 连接数据库
|
||
conn = pymysql.connect(**db_config)
|
||
cursor = conn.cursor(pymysql.cursors.DictCursor)
|
||
|
||
# 查询所有显示的菜单
|
||
sql = """
|
||
SELECT id, pid, name, component, menu_type, sort, is_show
|
||
FROM eb_system_menu
|
||
WHERE is_show = 1
|
||
ORDER BY pid, sort, id
|
||
"""
|
||
|
||
cursor.execute(sql)
|
||
all_menus = cursor.fetchall()
|
||
|
||
# 构建菜单树
|
||
menu_tree = {}
|
||
for menu in all_menus:
|
||
if menu['pid'] == 0:
|
||
if menu['id'] not in menu_tree:
|
||
menu_tree[menu['id']] = {'info': menu, 'children': []}
|
||
else:
|
||
menu_tree[menu['id']]['info'] = menu
|
||
|
||
# 添加子菜单
|
||
for menu in all_menus:
|
||
if menu['pid'] != 0:
|
||
parent_id = menu['pid']
|
||
if parent_id not in menu_tree:
|
||
menu_tree[parent_id] = {'info': None, 'children': []}
|
||
menu_tree[parent_id]['children'].append(menu)
|
||
|
||
# 生成Markdown文档
|
||
output = """# 系统完整菜单结构
|
||
|
||
> 生成时间：2024-12-04
|
||
> 说明：本文档记录了系统中所有一级菜单及其子菜单的完整层级结构
|
||
|
||
---
|
||
|
||
## 📋 菜单统计
|
||
|
||
- **一级菜单总数**: {total_level1}
|
||
- **子菜单总数**: {total_children}
|
||
- **菜单总数**: {total_all}
|
||
|
||
---
|
||
|
||
## 🗂️ 完整菜单结构
|
||
|
||
"""
|
||
|
||
# 统计
|
||
total_level1 = len([m for m in all_menus if m['pid'] == 0])
|
||
total_children = len([m for m in all_menus if m['pid'] != 0])
|
||
total_all = len(all_menus)
|
||
|
||
output = output.format(
|
||
total_level1=total_level1,
|
||
total_children=total_children,
|
||
total_all=total_all
|
||
)
|
||
|
||
# 按sort排序一级菜单
|
||
sorted_level1 = sorted(
|
||
[(k, v) for k, v in menu_tree.items() if v['info'] is not None],
|
||
key=lambda x: (x[1]['info']['sort'], x[1]['info']['id'])
|
||
)
|
||
|
||
# 生成菜单列表
|
||
for idx, (menu_id, menu_data) in enumerate(sorted_level1, 1):
|
||
menu_info = menu_data['info']
|
||
children = menu_data['children']
|
||
|
||
# 一级菜单标题
|
||
output += f"\n### {idx}. {menu_info['name']}\n\n"
|
||
output += f"- **菜单ID**: {menu_info['id']}\n"
|
||
output += f"- **路由路径**: `{menu_info['component']}`\n"
|
||
output += f"- **菜单类型**: {menu_info['menu_type']}\n"
|
||
output += f"- **排序**: {menu_info['sort']}\n"
|
||
|
||
if children:
|
||
output += f"- **子菜单数量**: {len(children)}\n\n"
|
||
output += "#### 子菜单列表：\n\n"
|
||
# 按sort排序子菜单
|
||
sorted_children = sorted(children, key=lambda x: (x['sort'], x['id']))
|
||
for child_idx, child in enumerate(sorted_children, 1):
|
||
output += f"{child_idx}. **{child['name']}**\n"
|
||
output += f" - ID: {child['id']}\n"
|
||
output += f" - 路径: `{child['component']}`\n"
|
||
output += f" - 类型: {child['menu_type']}\n"
|
||
output += f" - 排序: {child['sort']}\n\n"
|
||
else:
|
||
output += f"- **子菜单数量**: 0（无子菜单）\n\n"
|
||
|
||
output += "---\n"
|
||
|
||
# 保存文件
|
||
output_file = r'小张\行动\完整菜单结构.md'
|
||
with open(output_file, 'w', encoding='utf-8') as f:
|
||
f.write(output)
|
||
|
||
print(f"✅ 菜单结构已生成到文件: {output_file}")
|
||
print(f"📊 统计信息:")
|
||
print(f" - 一级菜单: {total_level1} 个")
|
||
print(f" - 子菜单: {total_children} 个")
|
||
print(f" - 总计: {total_all} 个")
|
||
|
||
# 关闭连接
|
||
cursor.close()
|
||
conn.close()
|