forked from bot/app
64 lines
1.6 KiB
Python
64 lines
1.6 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""
|
|
Copyright (C) 2020-2024 LiteyukiStudio. All Rights Reserved
|
|
|
|
@Time : 2024/8/19 下午10:51
|
|
@Author : snowykami
|
|
@Email : snowykami@outlook.com
|
|
@File : matcher.py
|
|
@Software: PyCharm
|
|
"""
|
|
import traceback
|
|
from typing import Any, TypeAlias, Callable, Coroutine
|
|
|
|
from liteyuki.message.event import MessageEvent
|
|
from liteyuki.message.rule import Rule
|
|
|
|
EventHandler: TypeAlias = Callable[[MessageEvent], Coroutine[None, None, Any]]
|
|
|
|
|
|
class Matcher:
|
|
def __init__(self, rule: Rule, priority: int, block: bool):
|
|
"""
|
|
匹配器
|
|
Args:
|
|
rule: 规则
|
|
priority: 优先级 >= 0
|
|
block: 是否阻断后续优先级更低的匹配器
|
|
"""
|
|
self.rule = rule
|
|
self.priority = priority
|
|
self.block = block
|
|
self.handlers: list[EventHandler] = []
|
|
|
|
def __str__(self):
|
|
return f"Matcher(rule={self.rule}, priority={self.priority}, block={self.block})"
|
|
|
|
def handle(self) -> Callable[[EventHandler], EventHandler]:
|
|
"""
|
|
添加处理函数，装饰器
|
|
Returns:
|
|
装饰器 handler
|
|
"""
|
|
def decorator(handler: EventHandler) -> EventHandler:
|
|
self.handlers.append(handler)
|
|
return handler
|
|
|
|
return decorator
|
|
|
|
async def run(self, event: MessageEvent) -> None:
|
|
"""
|
|
运行处理函数
|
|
Args:
|
|
event:
|
|
Returns:
|
|
"""
|
|
if not await self.rule(event):
|
|
return
|
|
|
|
for handler in self.handlers:
|
|
try:
|
|
await handler(event)
|
|
except Exception:
|
|
traceback.print_exc()
|