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

Merge branch 'feature/webui-config' of https://github.com/HibiKier/zhenxun_bot into feature/webui-config

Browse Source
This commit is contained in:
HibiKier 2025-05-11 05:28:31 +08:00
commit f583306dae
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 from strenum import StrEnum
class PriorityLifecycle(StrEnum): class PriorityLifecycleType(StrEnum):
STARTUP = "STARTUP" STARTUP = "STARTUP"
"""启动""" """启动"""
SHUTDOWN = "SHUTDOWN" 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 nonebot.utils import is_coroutine_callable
from zhenxun.services.log import logger from zhenxun.services.log import logger
from zhenxun.utils.enum import PriorityLifecycle from zhenxun.utils.enum import PriorityLifecycleType
from zhenxun.utils.exception import HookPriorityException from zhenxun.utils.exception import HookPriorityException
driver = nonebot.get_driver() driver = nonebot.get_driver()
class PriorityLifecycle: class PriorityLifecycle:
_data: ClassVar[dict[PriorityLifecycle, dict[int, list[Callable]]]] = {} _data: ClassVar[dict[PriorityLifecycleType, dict[int, list[Callable]]]] = {}
@classmethod @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: if hook_type not in cls._data:
cls._data[hook_type] = {} cls._data[hook_type] = {}
if priority not in cls._data[hook_type]: if priority not in cls._data[hook_type]:
@ -25,7 +25,7 @@ class PriorityLifecycle:
@classmethod @classmethod
def on_startup(cls, *, priority: int = 5): def on_startup(cls, *, priority: int = 5):
def wrapper(func): def wrapper(func):
cls.add(PriorityLifecycle.STARTUP, func, priority) cls.add(PriorityLifecycleType.STARTUP, func, priority)
return func return func
return wrapper return wrapper
@ -33,7 +33,7 @@ class PriorityLifecycle:
@classmethod @classmethod
def on_shutdown(cls, *, priority: int = 5): def on_shutdown(cls, *, priority: int = 5):
def wrapper(func): def wrapper(func):
cls.add(PriorityLifecycle.SHUTDOWN, func, priority) cls.add(PriorityLifecycleType.SHUTDOWN, func, priority)
return func return func
return wrapper return wrapper
@ -41,7 +41,7 @@ class PriorityLifecycle:
@driver.on_startup @driver.on_startup
async def _(): async def _():
priority_data = PriorityLifecycle._data.get(PriorityLifecycle.STARTUP) priority_data = PriorityLifecycle._data.get(PriorityLifecycleType.STARTUP)
if not priority_data: if not priority_data:
return return
priority_list = sorted(priority_data.keys()) priority_list = sorted(priority_data.keys())