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 7 Packages Projects Releases Wiki Activity

🐛 修改PriorityLifecycle字典类名称

Browse Source
This commit is contained in:
HibiKier 2025-04-11 16:49:45 +08:00
parent e84fc89fed
commit b10b277dba
2 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
zhenxun/utils/enum.py
View File

@ -6,7 +6,7 @@ else:
from strenum import StrEnum
class PriorityLifecycle(StrEnum):
class PriorityLifecycleType(StrEnum):
STARTUP = "STARTUP"
"""启动"""
SHUTDOWN = "SHUTDOWN"

12
zhenxun/utils/manager/priority_manager.py
View File

@ -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())