Diuse/zhenxun_bot
1
0
Fork 0
mirror of https://github.com/zhenxun-org/zhenxun_bot.git synced 2025-12-15 14:22:55 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

✨ 添加关于菜单

Browse Source
This commit is contained in:
HibiKier 2025-03-12 12:33:55 +08:00
parent 11d04efc0e
commit cffc3762f2
1 changed files with 49 additions and 34 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

83
zhenxun/builtin_plugins/web_ui/api/menu/data_source.py
View File

@ -5,6 +5,33 @@ from zhenxun.services.log import logger
from .model import MenuData, MenuItem from .model import MenuData, MenuItem
default_menus = [
MenuItem(
name="仪表盘",
module="dashboard",
router="/dashboard",
icon="dashboard",
default=True,
),
MenuItem(
name="真寻控制台",
module="command",
router="/command",
icon="command",
),
MenuItem(name="插件列表", module="plugin", router="/plugin", icon="plugin"),
MenuItem(name="插件商店", module="store", router="/store", icon="store"),
MenuItem(name="好友/群组", module="manage", router="/manage", icon="user"),
MenuItem(
name="数据库管理",
module="database",
router="/database",
icon="database",
),
MenuItem(name="系统信息", module="system", router="/system", icon="system"),
MenuItem(name="关于我们", module="about", router="/about", icon="about"),
]
class MenuManage: class MenuManage:
def __init__(self) -> None: def __init__(self) -> None:
@ -12,44 +39,32 @@ class MenuManage:
self.menu = [] self.menu = []
if self.file.exists(): if self.file.exists():
try: try:
temp_menu = []
self.menu = json.load(self.file.open(encoding="utf8")) self.menu = json.load(self.file.open(encoding="utf8"))
self_menu_name = [menu["name"] for menu in self.menu]
for module in [m.module for m in default_menus]:
if module in self_menu_name:
temp_menu.append(
MenuItem(
**next(m for m in self.menu if m["module"] == module)
)
)
else:
temp_menu.append(
default_menus[
next(
i
for i, m in enumerate(default_menus)
if m.module == module
)
]
)
self.menu = temp_menu
except Exception as e: except Exception as e:
logger.warning("菜单文件损坏，已重新生成...", "WebUi", e=e) logger.warning("菜单文件损坏，已重新生成...", "WebUi", e=e)
if not self.menu: if not self.menu:
self.menu = [ self.menu = default_menus
MenuItem( self.save()
name="仪表盘",
module="dashboard",
router="/dashboard",
icon="dashboard",
default=True,
),
MenuItem(
name="真寻控制台",
module="command",
router="/command",
icon="command",
),
MenuItem(
name="插件列表", module="plugin", router="/plugin", icon="plugin"
),
MenuItem(
name="插件商店", module="store", router="/store", icon="store"
),
MenuItem(
name="好友/群组", module="manage", router="/manage", icon="user"
),
MenuItem(
name="数据库管理",
module="database",
router="/database",
icon="database",
),
MenuItem(
name="系统信息", module="system", router="/system", icon="system"
),
]
self.save()
def get_menus(self): def get_menus(self):
return MenuData(menus=self.menu) return MenuData(menus=self.menu)