From e84fc89fed0ecb8114346f2a3837de0afb610a83 Mon Sep 17 00:00:00 2001
From: HibiKier <775757368@qq.com>
Date: Fri, 4 Apr 2025 04:44:37 +0800
Subject: [PATCH] =?UTF-8?q?:art:=20=E6=98=BE=E7=A4=BA=E6=B7=BB=E5=8A=A0?=
 =?UTF-8?q?=E4=BC=98=E5=85=88=E7=BA=A7?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 zhenxun/builtin_plugins/__init__.py           |  4 ++--
 .../admin/welcome_message/data_source.py      |  4 ++--
 zhenxun/builtin_plugins/init/init_config.py   |  4 ++--
 zhenxun/builtin_plugins/init/init_plugin.py   |  4 ++--
 zhenxun/builtin_plugins/init/init_task.py     |  4 ++--
 zhenxun/builtin_plugins/scripts.py            |  6 +++---
 zhenxun/builtin_plugins/sign_in/utils.py      |  4 ++--
 zhenxun/builtin_plugins/web_ui/__init__.py    |  4 ++--
 .../web_ui/api/menu/data_source.py            | 19 ++++++++-----------
 .../web_ui/api/tabs/dashboard/data_source.py  |  4 ++--
 .../web_ui/api/tabs/database/__init__.py      |  4 ++--
 zhenxun/builtin_plugins/web_ui/config.py      |  8 +++++++-
 zhenxun/services/db_context.py                |  4 ++--
 zhenxun/services/plugin_init.py               |  4 ++--
 zhenxun/utils/_build_mat.py                   |  7 ++++++-
 zhenxun/utils/enum.py                         |  9 +++++++--
 zhenxun/utils/github_utils/models.py          |  7 ++++++-
 zhenxun/utils/manager/priority_manager.py     | 18 +++++++++---------
 18 files changed, 68 insertions(+), 50 deletions(-)

diff --git a/zhenxun/builtin_plugins/__init__.py b/zhenxun/builtin_plugins/__init__.py
index 7bc82012..028e7558 100644
--- a/zhenxun/builtin_plugins/__init__.py
+++ b/zhenxun/builtin_plugins/__init__.py
@@ -16,7 +16,7 @@ from zhenxun.models.sign_user import SignUser
 from zhenxun.models.user_console import UserConsole
 from zhenxun.services.log import logger
 from zhenxun.utils.decorator.shop import shop_register
-from zhenxun.utils.manager.priority_manager import HookPriorityManager
+from zhenxun.utils.manager.priority_manager import PriorityLifecycle
 from zhenxun.utils.manager.resource_manager import ResourceManager
 from zhenxun.utils.platform import PlatformUtils
 
@@ -71,7 +71,7 @@ from public.bag_users t1
 """
 
 
-@HookPriorityManager.on_startup()
+@PriorityLifecycle.on_startup()
 async def _():
     await ResourceManager.init_resources()
     """签到与用户的数据迁移"""
diff --git a/zhenxun/builtin_plugins/admin/welcome_message/data_source.py b/zhenxun/builtin_plugins/admin/welcome_message/data_source.py
index f5ab5ecd..0714e4b4 100644
--- a/zhenxun/builtin_plugins/admin/welcome_message/data_source.py
+++ b/zhenxun/builtin_plugins/admin/welcome_message/data_source.py
@@ -14,7 +14,7 @@ from zhenxun.services.log import logger
 from zhenxun.utils._build_image import BuildImage
 from zhenxun.utils._image_template import ImageTemplate
 from zhenxun.utils.http_utils import AsyncHttpx
-from zhenxun.utils.manager.priority_manager import HookPriorityManager
+from zhenxun.utils.manager.priority_manager import PriorityLifecycle
 from zhenxun.utils.platform import PlatformUtils
 
 BASE_PATH = DATA_PATH / "welcome_message"
@@ -92,7 +92,7 @@ def migrate(path: Path):
         json.dump(new_data, f, ensure_ascii=False, indent=4)
 
 
-@HookPriorityManager.on_startup()
+@PriorityLifecycle.on_startup()
 def _():
     """数据迁移
 
diff --git a/zhenxun/builtin_plugins/init/init_config.py b/zhenxun/builtin_plugins/init/init_config.py
index 169e4068..eef63635 100644
--- a/zhenxun/builtin_plugins/init/init_config.py
+++ b/zhenxun/builtin_plugins/init/init_config.py
@@ -11,7 +11,7 @@ from zhenxun.configs.config import Config
 from zhenxun.configs.path_config import DATA_PATH
 from zhenxun.configs.utils import RegisterConfig
 from zhenxun.services.log import logger
-from zhenxun.utils.manager.priority_manager import HookPriorityManager
+from zhenxun.utils.manager.priority_manager import PriorityLifecycle
 
 _yaml = YAML(pure=True)
 _yaml.allow_unicode = True
@@ -103,7 +103,7 @@ def _generate_simple_config(exists_module: list[str]):
         temp_file.unlink()
 
 
-@HookPriorityManager.on_startup(0)
+@PriorityLifecycle.on_startup(priority=0)
 def _():
     """
     初始化插件数据配置
diff --git a/zhenxun/builtin_plugins/init/init_plugin.py b/zhenxun/builtin_plugins/init/init_plugin.py
index 510fc5a2..bc3eda99 100644
--- a/zhenxun/builtin_plugins/init/init_plugin.py
+++ b/zhenxun/builtin_plugins/init/init_plugin.py
@@ -20,7 +20,7 @@ from zhenxun.utils.enum import (
     PluginLimitType,
     PluginType,
 )
-from zhenxun.utils.manager.priority_manager import HookPriorityManager
+from zhenxun.utils.manager.priority_manager import PriorityLifecycle
 
 from .manager import manager
 
@@ -96,7 +96,7 @@ async def _handle_setting(
         )
 
 
-@HookPriorityManager.on_startup()
+@PriorityLifecycle.on_startup()
 async def _():
     """
     初始化插件数据配置
diff --git a/zhenxun/builtin_plugins/init/init_task.py b/zhenxun/builtin_plugins/init/init_task.py
index 50475ccd..8857e665 100644
--- a/zhenxun/builtin_plugins/init/init_task.py
+++ b/zhenxun/builtin_plugins/init/init_task.py
@@ -10,7 +10,7 @@ from zhenxun.models.group_console import GroupConsole
 from zhenxun.models.task_info import TaskInfo
 from zhenxun.services.log import logger
 from zhenxun.utils.common_utils import CommonUtils
-from zhenxun.utils.manager.priority_manager import HookPriorityManager
+from zhenxun.utils.manager.priority_manager import PriorityLifecycle
 
 driver: Driver = nonebot.get_driver()
 
@@ -133,7 +133,7 @@ async def create_schedule(task: Task):
         logger.error(f"动态创建定时任务 {task.name}({task.module}) 失败", e=e)
 
 
-@HookPriorityManager.on_startup()
+@PriorityLifecycle.on_startup()
 async def _():
     """
     初始化插件数据配置
diff --git a/zhenxun/builtin_plugins/scripts.py b/zhenxun/builtin_plugins/scripts.py
index 02516b64..9cc2949e 100644
--- a/zhenxun/builtin_plugins/scripts.py
+++ b/zhenxun/builtin_plugins/scripts.py
@@ -10,12 +10,12 @@ from zhenxun.configs.path_config import TEXT_PATH
 from zhenxun.models.group_console import GroupConsole
 from zhenxun.services.log import logger
 from zhenxun.utils.http_utils import AsyncHttpx
-from zhenxun.utils.manager.priority_manager import HookPriorityManager
+from zhenxun.utils.manager.priority_manager import PriorityLifecycle
 
 driver: Driver = nonebot.get_driver()
 
 
-@HookPriorityManager.on_startup()
+@PriorityLifecycle.on_startup()
 async def update_city():
     """
     部分插件需要中国省份城市
@@ -62,7 +62,7 @@ async def _():
     await update_city()
 
 
-@HookPriorityManager.on_startup()
+@PriorityLifecycle.on_startup()
 async def _():
     """开启/禁用插件格式修改"""
     _, is_create = await GroupConsole.get_or_create(group_id=133133133)
diff --git a/zhenxun/builtin_plugins/sign_in/utils.py b/zhenxun/builtin_plugins/sign_in/utils.py
index 84c0d8cd..3bd20c99 100644
--- a/zhenxun/builtin_plugins/sign_in/utils.py
+++ b/zhenxun/builtin_plugins/sign_in/utils.py
@@ -16,7 +16,7 @@ from zhenxun.models.sign_log import SignLog
 from zhenxun.models.sign_user import SignUser
 from zhenxun.utils.http_utils import AsyncHttpx
 from zhenxun.utils.image_utils import BuildImage
-from zhenxun.utils.manager.priority_manager import HookPriorityManager
+from zhenxun.utils.manager.priority_manager import PriorityLifecycle
 from zhenxun.utils.platform import PlatformUtils
 
 from .config import (
@@ -55,7 +55,7 @@ LG_MESSAGE = [
 ]
 
 
-@HookPriorityManager.on_startup()
+@PriorityLifecycle.on_startup()
 async def init_image():
     SIGN_RESOURCE_PATH.mkdir(parents=True, exist_ok=True)
     SIGN_TODAY_CARD_PATH.mkdir(exist_ok=True, parents=True)
diff --git a/zhenxun/builtin_plugins/web_ui/__init__.py b/zhenxun/builtin_plugins/web_ui/__init__.py
index 4df596d1..8030c71b 100644
--- a/zhenxun/builtin_plugins/web_ui/__init__.py
+++ b/zhenxun/builtin_plugins/web_ui/__init__.py
@@ -10,7 +10,7 @@ from zhenxun.configs.config import Config as gConfig
 from zhenxun.configs.utils import PluginExtraData, RegisterConfig
 from zhenxun.services.log import logger, logger_
 from zhenxun.utils.enum import PluginType
-from zhenxun.utils.manager.priority_manager import HookPriorityManager
+from zhenxun.utils.manager.priority_manager import PriorityLifecycle
 
 from .api.configure import router as configure_router
 from .api.logs import router as ws_log_routes
@@ -94,7 +94,7 @@ WsApiRouter.include_router(status_routes)
 WsApiRouter.include_router(chat_routes)
 
 
-@HookPriorityManager.on_startup(0)
+@PriorityLifecycle.on_startup(priority=0)
 async def _():
     try:
 
diff --git a/zhenxun/builtin_plugins/web_ui/api/menu/data_source.py b/zhenxun/builtin_plugins/web_ui/api/menu/data_source.py
index 1696c61c..e54bf9e5 100644
--- a/zhenxun/builtin_plugins/web_ui/api/menu/data_source.py
+++ b/zhenxun/builtin_plugins/web_ui/api/menu/data_source.py
@@ -33,7 +33,7 @@ default_menus = [
 ]
 
 
-class MenuManage:
+class MenuManager:
     def __init__(self) -> None:
         self.file = DATA_PATH / "web_ui" / "menu.json"
         self.menu = []
@@ -50,15 +50,7 @@ class MenuManage:
                             )
                         )
                     else:
-                        temp_menu.append(
-                            default_menus[
-                                next(
-                                    i
-                                    for i, m in enumerate(default_menus)
-                                    if m.module == module
-                                )
-                            ]
-                        )
+                        temp_menu.append(self.__get_menu_model(module))
                 self.menu = temp_menu
             except Exception as e:
                 logger.warning("菜单文件损坏，已重新生成...", "WebUi", e=e)
@@ -66,6 +58,11 @@ class MenuManage:
             self.menu = default_menus
         self.save()
 
+    def __get_menu_model(self, module: str):
+        return default_menus[
+            next(i for i, m in enumerate(default_menus) if m.module == module)
+        ]
+
     def get_menus(self):
         return MenuData(menus=self.menu)
 
@@ -76,4 +73,4 @@ class MenuManage:
             json.dump(temp, f, ensure_ascii=False, indent=4)
 
 
-menu_manage = MenuManage()
+menu_manage = MenuManager()
diff --git a/zhenxun/builtin_plugins/web_ui/api/tabs/dashboard/data_source.py b/zhenxun/builtin_plugins/web_ui/api/tabs/dashboard/data_source.py
index 535cac22..1853b8bd 100644
--- a/zhenxun/builtin_plugins/web_ui/api/tabs/dashboard/data_source.py
+++ b/zhenxun/builtin_plugins/web_ui/api/tabs/dashboard/data_source.py
@@ -13,7 +13,7 @@ from zhenxun.models.bot_connect_log import BotConnectLog
 from zhenxun.models.chat_history import ChatHistory
 from zhenxun.models.statistics import Statistics
 from zhenxun.services.log import logger
-from zhenxun.utils.manager.priority_manager import HookPriorityManager
+from zhenxun.utils.manager.priority_manager import PriorityLifecycle
 from zhenxun.utils.platform import PlatformUtils
 
 from ....base_model import BaseResultModel, QueryModel
@@ -32,7 +32,7 @@ driver: Driver = nonebot.get_driver()
 CONNECT_TIME = 0
 
 
-@HookPriorityManager.on_startup()
+@PriorityLifecycle.on_startup()
 async def _():
     global CONNECT_TIME
     CONNECT_TIME = int(time.time())
diff --git a/zhenxun/builtin_plugins/web_ui/api/tabs/database/__init__.py b/zhenxun/builtin_plugins/web_ui/api/tabs/database/__init__.py
index a05d8953..b01c3f8c 100644
--- a/zhenxun/builtin_plugins/web_ui/api/tabs/database/__init__.py
+++ b/zhenxun/builtin_plugins/web_ui/api/tabs/database/__init__.py
@@ -8,7 +8,7 @@ from zhenxun.configs.config import BotConfig
 from zhenxun.models.plugin_info import PluginInfo
 from zhenxun.models.task_info import TaskInfo
 from zhenxun.services.log import logger
-from zhenxun.utils.manager.priority_manager import HookPriorityManager
+from zhenxun.utils.manager.priority_manager import PriorityLifecycle
 
 from ....base_model import BaseResultModel, QueryModel, Result
 from ....utils import authentication
@@ -22,7 +22,7 @@ router = APIRouter(prefix="/database")
 driver: Driver = nonebot.get_driver()
 
 
-@HookPriorityManager.on_startup()
+@PriorityLifecycle.on_startup()
 async def _():
     for plugin in nonebot.get_loaded_plugins():
         module = plugin.name
diff --git a/zhenxun/builtin_plugins/web_ui/config.py b/zhenxun/builtin_plugins/web_ui/config.py
index bddcb062..4a88aad9 100644
--- a/zhenxun/builtin_plugins/web_ui/config.py
+++ b/zhenxun/builtin_plugins/web_ui/config.py
@@ -1,6 +1,12 @@
+import sys
+
 from fastapi.middleware.cors import CORSMiddleware
 import nonebot
-from strenum import StrEnum
+
+if sys.version_info >= (3, 11):
+    from enum import StrEnum
+else:
+    from strenum import StrEnum
 
 from zhenxun.configs.path_config import DATA_PATH, TEMP_PATH
 
diff --git a/zhenxun/services/db_context.py b/zhenxun/services/db_context.py
index c774f912..33678965 100644
--- a/zhenxun/services/db_context.py
+++ b/zhenxun/services/db_context.py
@@ -6,7 +6,7 @@ from tortoise.models import Model as Model_
 
 from zhenxun.configs.config import BotConfig
 from zhenxun.utils.exception import HookPriorityException
-from zhenxun.utils.manager.priority_manager import HookPriorityManager
+from zhenxun.utils.manager.priority_manager import PriorityLifecycle
 
 from .log import logger
 
@@ -48,7 +48,7 @@ class DbConnectError(Exception):
     pass
 
 
-@HookPriorityManager.on_startup(1)
+@PriorityLifecycle.on_startup(priority=1)
 async def init():
     if not BotConfig.db_url:
         # raise DbUrlIsNode("数据库配置为空，请在.env.dev中配置DB_URL...")
diff --git a/zhenxun/services/plugin_init.py b/zhenxun/services/plugin_init.py
index 4993950f..1e573f6e 100644
--- a/zhenxun/services/plugin_init.py
+++ b/zhenxun/services/plugin_init.py
@@ -6,7 +6,7 @@ from nonebot.utils import is_coroutine_callable
 from pydantic import BaseModel
 
 from zhenxun.services.log import logger
-from zhenxun.utils.manager.priority_manager import HookPriorityManager
+from zhenxun.utils.manager.priority_manager import PriorityLifecycle
 
 driver = nonebot.get_driver()
 
@@ -101,6 +101,6 @@ class PluginInitManager:
                     logger.error(f"执行: {module_path}:remove 失败", e=e)
 
 
-@HookPriorityManager.on_startup()
+@PriorityLifecycle.on_startup()
 async def _():
     await PluginInitManager.install_all()
diff --git a/zhenxun/utils/_build_mat.py b/zhenxun/utils/_build_mat.py
index de73e69d..a3de3087 100644
--- a/zhenxun/utils/_build_mat.py
+++ b/zhenxun/utils/_build_mat.py
@@ -1,12 +1,17 @@
 from io import BytesIO
 from pathlib import Path
 import random
+import sys
 
 from pydantic import BaseModel, Field
-from strenum import StrEnum
 
 from ._build_image import BuildImage
 
+if sys.version_info >= (3, 11):
+    from enum import StrEnum
+else:
+    from strenum import StrEnum
+
 
 class MatType(StrEnum):
     LINE = "LINE"
diff --git a/zhenxun/utils/enum.py b/zhenxun/utils/enum.py
index a68d8918..f9c974c6 100644
--- a/zhenxun/utils/enum.py
+++ b/zhenxun/utils/enum.py
@@ -1,7 +1,12 @@
-from strenum import StrEnum
+import sys
+
+if sys.version_info >= (3, 11):
+    from enum import StrEnum
+else:
+    from strenum import StrEnum
 
 
-class HookPriorityType(StrEnum):
+class PriorityLifecycle(StrEnum):
     STARTUP = "STARTUP"
     """启动"""
     SHUTDOWN = "SHUTDOWN"
diff --git a/zhenxun/utils/github_utils/models.py b/zhenxun/utils/github_utils/models.py
index e3e5dfe3..fb690616 100644
--- a/zhenxun/utils/github_utils/models.py
+++ b/zhenxun/utils/github_utils/models.py
@@ -1,13 +1,18 @@
 import contextlib
+import sys
 from typing import Protocol
 
 from aiocache import cached
 from nonebot.compat import model_dump
 from pydantic import BaseModel, Field
-from strenum import StrEnum
 
 from zhenxun.utils.http_utils import AsyncHttpx
 
+if sys.version_info >= (3, 11):
+    from enum import StrEnum
+else:
+    from strenum import StrEnum
+
 from .const import (
     CACHED_API_TTL,
     GIT_API_COMMIT_FORMAT,
diff --git a/zhenxun/utils/manager/priority_manager.py b/zhenxun/utils/manager/priority_manager.py
index 0cb454b3..6ab6848d 100644
--- a/zhenxun/utils/manager/priority_manager.py
+++ b/zhenxun/utils/manager/priority_manager.py
@@ -5,17 +5,17 @@ import nonebot
 from nonebot.utils import is_coroutine_callable
 
 from zhenxun.services.log import logger
-from zhenxun.utils.enum import HookPriorityType
+from zhenxun.utils.enum import PriorityLifecycle
 from zhenxun.utils.exception import HookPriorityException
 
 driver = nonebot.get_driver()
 
 
-class HookPriorityManager:
-    _data: ClassVar[dict[HookPriorityType, dict[int, list[Callable]]]] = {}
+class PriorityLifecycle:
+    _data: ClassVar[dict[PriorityLifecycle, dict[int, list[Callable]]]] = {}
 
     @classmethod
-    def add(cls, hook_type: HookPriorityType, func: Callable, priority: int = 5):
+    def add(cls, hook_type: PriorityLifecycle, func: Callable, priority: int = 5):
         if hook_type not in cls._data:
             cls._data[hook_type] = {}
         if priority not in cls._data[hook_type]:
@@ -23,17 +23,17 @@ class HookPriorityManager:
         cls._data[hook_type][priority].append(func)
 
     @classmethod
-    def on_startup(cls, priority: int = 5):
+    def on_startup(cls, *, priority: int = 5):
         def wrapper(func):
-            cls.add(HookPriorityType.STARTUP, func, priority)
+            cls.add(PriorityLifecycle.STARTUP, func, priority)
             return func
 
         return wrapper
 
     @classmethod
-    def on_shutdown(cls, priority: int = 5):
+    def on_shutdown(cls, *, priority: int = 5):
         def wrapper(func):
-            cls.add(HookPriorityType.SHUTDOWN, func, priority)
+            cls.add(PriorityLifecycle.SHUTDOWN, func, priority)
             return func
 
         return wrapper
@@ -41,7 +41,7 @@ class HookPriorityManager:
 
 @driver.on_startup
 async def _():
-    priority_data = HookPriorityManager._data.get(HookPriorityType.STARTUP)
+    priority_data = PriorityLifecycle._data.get(PriorityLifecycle.STARTUP)
     if not priority_data:
         return
     priority_list = sorted(priority_data.keys())