From b10b277dbab72e61645b9747c4793280db46861e Mon Sep 17 00:00:00 2001 From: HibiKier <775757368@qq.com> Date: Fri, 11 Apr 2025 16:49:45 +0800 Subject: [PATCH] =?UTF-8?q?:bug:=20=E4=BF=AE=E6=94=B9PriorityLifecycle?= =?UTF-8?q?=E5=AD=97=E5=85=B8=E7=B1=BB=E5=90=8D=E7=A7=B0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- zhenxun/utils/enum.py | 2 +- zhenxun/utils/manager/priority_manager.py | 12 ++++++------ 2 files changed, 7 insertions(+), 7 deletions(-) diff --git a/zhenxun/utils/enum.py b/zhenxun/utils/enum.py index f9c974c6..7e816b2b 100644 --- a/zhenxun/utils/enum.py +++ b/zhenxun/utils/enum.py @@ -6,7 +6,7 @@ else: from strenum import StrEnum -class PriorityLifecycle(StrEnum): +class PriorityLifecycleType(StrEnum): STARTUP = "STARTUP" """启动""" SHUTDOWN = "SHUTDOWN" diff --git a/zhenxun/utils/manager/priority_manager.py b/zhenxun/utils/manager/priority_manager.py index 6ab6848d..d42148e0 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 PriorityLifecycle +from zhenxun.utils.enum import PriorityLifecycleType from zhenxun.utils.exception import HookPriorityException driver = nonebot.get_driver() class PriorityLifecycle: - _data: ClassVar[dict[PriorityLifecycle, dict[int, list[Callable]]]] = {} + _data: ClassVar[dict[PriorityLifecycleType, dict[int, list[Callable]]]] = {} @classmethod - def add(cls, hook_type: PriorityLifecycle, func: Callable, priority: int = 5): + def add(cls, hook_type: PriorityLifecycleType, func: Callable, priority: int = 5): if hook_type not in cls._data: cls._data[hook_type] = {} if priority not in cls._data[hook_type]: @@ -25,7 +25,7 @@ class PriorityLifecycle: @classmethod def on_startup(cls, *, priority: int = 5): def wrapper(func): - cls.add(PriorityLifecycle.STARTUP, func, priority) + cls.add(PriorityLifecycleType.STARTUP, func, priority) return func return wrapper @@ -33,7 +33,7 @@ class PriorityLifecycle: @classmethod def on_shutdown(cls, *, priority: int = 5): def wrapper(func): - cls.add(PriorityLifecycle.SHUTDOWN, func, priority) + cls.add(PriorityLifecycleType.SHUTDOWN, func, priority) return func return wrapper @@ -41,7 +41,7 @@ class PriorityLifecycle: @driver.on_startup async def _(): - priority_data = PriorityLifecycle._data.get(PriorityLifecycle.STARTUP) + priority_data = PriorityLifecycle._data.get(PriorityLifecycleType.STARTUP) if not priority_data: return priority_list = sorted(priority_data.keys())