summaryrefslogtreecommitdiffstatshomepage
path: root/src/infini/handler.py
blob: 06d5c5fa2e2178816d43cb1a38e8e1391e776fd0 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
from abc import ABCMeta, abstractmethod
from .event import MatcherEvent
from .typing import Dict

__all__ = ["Result", "Handler"]


class Result(metaclass=ABCMeta):
    """规则包运行结果基类"""

    event: str
    status: bool
    kwargs: dict = {}

    def __init__(self, event: str, status: bool, **kwargs) -> None:
        self.event = event
        self.status = status
        self.kwargs = kwargs


class Handler:
    """规则包业务基类"""

    name: str
    priority: int = 0

    def __init_subclass__(cls) -> None:
        handlers.regist(cls.name, cls())

    @abstractmethod
    def process(self, event: MatcherEvent) -> Result:
        raise NotImplementedError


class Handlers:
    """规则包业务集合"""

    _handlers: Dict[str, Handler] = {}

    def regist(self, name: str, handler: Handler) -> None:
        self._handlers[name.lower()] = handler

    def match(self, name: str) -> Handler | None:
        return self._handlers.get(name.lower())


handlers = Handlers()