mirror of
https://github.com/nonebot/nonebot2.git
synced 2024-12-12 14:35:45 +08:00
945 lines
31 KiB
Python
945 lines
31 KiB
Python
from collections.abc import Iterable
|
|
from contextlib import AsyncExitStack, contextmanager
|
|
from contextvars import ContextVar
|
|
from dataclasses import dataclass
|
|
from datetime import datetime, timedelta
|
|
import inspect
|
|
from pathlib import Path
|
|
import sys
|
|
from types import ModuleType
|
|
from typing import ( # noqa: UP035
|
|
TYPE_CHECKING,
|
|
Any,
|
|
Callable,
|
|
ClassVar,
|
|
NoReturn,
|
|
Optional,
|
|
Type,
|
|
TypeVar,
|
|
Union,
|
|
overload,
|
|
)
|
|
from typing_extensions import Self
|
|
import warnings
|
|
|
|
from exceptiongroup import BaseExceptionGroup, catch
|
|
|
|
from nonebot.consts import (
|
|
ARG_KEY,
|
|
LAST_RECEIVE_KEY,
|
|
RECEIVE_KEY,
|
|
REJECT_CACHE_TARGET,
|
|
REJECT_TARGET,
|
|
)
|
|
from nonebot.dependencies import Dependent, Param
|
|
from nonebot.exception import (
|
|
FinishedException,
|
|
PausedException,
|
|
RejectedException,
|
|
SkippedException,
|
|
StopPropagation,
|
|
)
|
|
from nonebot.internal.adapter import (
|
|
Bot,
|
|
Event,
|
|
Message,
|
|
MessageSegment,
|
|
MessageTemplate,
|
|
)
|
|
from nonebot.internal.params import (
|
|
ArgParam,
|
|
BotParam,
|
|
DefaultParam,
|
|
DependParam,
|
|
Depends,
|
|
EventParam,
|
|
MatcherParam,
|
|
StateParam,
|
|
)
|
|
from nonebot.internal.permission import Permission, User
|
|
from nonebot.internal.rule import Rule
|
|
from nonebot.log import logger
|
|
from nonebot.typing import (
|
|
T_DependencyCache,
|
|
T_Handler,
|
|
T_PermissionUpdater,
|
|
T_State,
|
|
T_TypeUpdater,
|
|
)
|
|
from nonebot.utils import classproperty, flatten_exception_group
|
|
|
|
from . import matchers
|
|
|
|
if TYPE_CHECKING:
|
|
from nonebot.plugin import Plugin
|
|
|
|
T = TypeVar("T")
|
|
|
|
current_bot: ContextVar[Bot] = ContextVar("current_bot")
|
|
current_event: ContextVar[Event] = ContextVar("current_event")
|
|
current_matcher: ContextVar["Matcher"] = ContextVar("current_matcher")
|
|
current_handler: ContextVar[Dependent[Any]] = ContextVar("current_handler")
|
|
|
|
|
|
@dataclass
|
|
class MatcherSource:
|
|
"""Matcher 源代码上下文信息"""
|
|
|
|
plugin_id: Optional[str] = None
|
|
"""事件响应器所在插件标识符"""
|
|
module_name: Optional[str] = None
|
|
"""事件响应器所在插件模块的路径名"""
|
|
lineno: Optional[int] = None
|
|
"""事件响应器所在行号"""
|
|
|
|
@property
|
|
def plugin(self) -> Optional["Plugin"]:
|
|
"""事件响应器所在插件"""
|
|
from nonebot.plugin import get_plugin
|
|
|
|
if self.plugin_id is not None:
|
|
return get_plugin(self.plugin_id)
|
|
|
|
@property
|
|
def plugin_name(self) -> Optional[str]:
|
|
"""事件响应器所在插件名"""
|
|
return self.plugin and self.plugin.name
|
|
|
|
@property
|
|
def module(self) -> Optional[ModuleType]:
|
|
if self.module_name is not None:
|
|
return sys.modules.get(self.module_name)
|
|
|
|
@property
|
|
def file(self) -> Optional[Path]:
|
|
if self.module is not None and (file := inspect.getsourcefile(self.module)):
|
|
return Path(file).absolute()
|
|
|
|
|
|
class MatcherMeta(type):
|
|
if TYPE_CHECKING:
|
|
type: str
|
|
_source: Optional[MatcherSource]
|
|
module_name: Optional[str]
|
|
|
|
def __repr__(self) -> str:
|
|
return (
|
|
f"{self.__name__}(type={self.type!r}"
|
|
+ (f", module={self.module_name}" if self.module_name else "")
|
|
+ (
|
|
f", lineno={self._source.lineno}"
|
|
if self._source and self._source.lineno is not None
|
|
else ""
|
|
)
|
|
+ ")"
|
|
)
|
|
|
|
|
|
class Matcher(metaclass=MatcherMeta):
|
|
"""事件响应器类"""
|
|
|
|
_source: ClassVar[Optional[MatcherSource]] = None
|
|
|
|
type: ClassVar[str] = ""
|
|
"""事件响应器类型"""
|
|
rule: ClassVar[Rule] = Rule()
|
|
"""事件响应器匹配规则"""
|
|
permission: ClassVar[Permission] = Permission()
|
|
"""事件响应器触发权限"""
|
|
handlers: ClassVar[list[Dependent[Any]]] = []
|
|
"""事件响应器拥有的事件处理函数列表"""
|
|
priority: ClassVar[int] = 1
|
|
"""事件响应器优先级"""
|
|
block: bool = False
|
|
"""事件响应器是否阻止事件传播"""
|
|
temp: ClassVar[bool] = False
|
|
"""事件响应器是否为临时"""
|
|
expire_time: ClassVar[Optional[datetime]] = None
|
|
"""事件响应器过期时间点"""
|
|
|
|
_default_state: ClassVar[T_State] = {}
|
|
"""事件响应器默认状态"""
|
|
|
|
_default_type_updater: ClassVar[Optional[Dependent[str]]] = None
|
|
"""事件响应器类型更新函数"""
|
|
_default_permission_updater: ClassVar[Optional[Dependent[Permission]]] = None
|
|
"""事件响应器权限更新函数"""
|
|
|
|
HANDLER_PARAM_TYPES: ClassVar[tuple[Type[Param], ...]] = ( # noqa: UP006
|
|
DependParam,
|
|
BotParam,
|
|
EventParam,
|
|
StateParam,
|
|
ArgParam,
|
|
MatcherParam,
|
|
DefaultParam,
|
|
)
|
|
|
|
def __init__(self):
|
|
self.remain_handlers: list[Dependent[Any]] = self.handlers.copy()
|
|
self.state = self._default_state.copy()
|
|
|
|
def __repr__(self) -> str:
|
|
return (
|
|
f"{self.__class__.__name__}(type={self.type!r}"
|
|
+ (f", module={self.module_name}" if self.module_name else "")
|
|
+ (
|
|
f", lineno={self._source.lineno}"
|
|
if self._source and self._source.lineno is not None
|
|
else ""
|
|
)
|
|
+ ")"
|
|
)
|
|
|
|
@classmethod
|
|
def new(
|
|
cls,
|
|
type_: str = "",
|
|
rule: Optional[Rule] = None,
|
|
permission: Optional[Permission] = None,
|
|
handlers: Optional[list[Union[T_Handler, Dependent[Any]]]] = None,
|
|
temp: bool = False,
|
|
priority: int = 1,
|
|
block: bool = False,
|
|
*,
|
|
plugin: Optional["Plugin"] = None,
|
|
module: Optional[ModuleType] = None,
|
|
source: Optional[MatcherSource] = None,
|
|
expire_time: Optional[Union[datetime, timedelta]] = None,
|
|
default_state: Optional[T_State] = None,
|
|
default_type_updater: Optional[Union[T_TypeUpdater, Dependent[str]]] = None,
|
|
default_permission_updater: Optional[
|
|
Union[T_PermissionUpdater, Dependent[Permission]]
|
|
] = None,
|
|
) -> Type[Self]: # noqa: UP006
|
|
"""
|
|
创建一个新的事件响应器,并存储至 `matchers <#matchers>`_
|
|
|
|
参数:
|
|
type_: 事件响应器类型,与 `event.get_type()` 一致时触发,空字符串表示任意
|
|
rule: 匹配规则
|
|
permission: 权限
|
|
handlers: 事件处理函数列表
|
|
temp: 是否为临时事件响应器,即触发一次后删除
|
|
priority: 响应优先级
|
|
block: 是否阻止事件向更低优先级的响应器传播
|
|
plugin: **Deprecated.** 事件响应器所在插件
|
|
module: **Deprecated.** 事件响应器所在模块
|
|
source: 事件响应器源代码上下文信息
|
|
expire_time: 事件响应器最终有效时间点,过时即被删除
|
|
default_state: 默认状态 `state`
|
|
default_type_updater: 默认事件类型更新函数
|
|
default_permission_updater: 默认会话权限更新函数
|
|
|
|
返回:
|
|
Type[Matcher]: 新的事件响应器类
|
|
"""
|
|
if plugin is not None:
|
|
warnings.warn(
|
|
(
|
|
"Pass `plugin` context info to create Matcher is deprecated. "
|
|
"Use `source` instead."
|
|
),
|
|
DeprecationWarning,
|
|
)
|
|
if module is not None:
|
|
warnings.warn(
|
|
(
|
|
"Pass `module` context info to create Matcher is deprecated. "
|
|
"Use `source` instead."
|
|
),
|
|
DeprecationWarning,
|
|
)
|
|
source = source or (
|
|
MatcherSource(
|
|
plugin_id=plugin and plugin.id_,
|
|
module_name=module and module.__name__,
|
|
)
|
|
if plugin is not None or module is not None
|
|
else None
|
|
)
|
|
|
|
NewMatcher = type(
|
|
cls.__name__,
|
|
(cls,),
|
|
{
|
|
"_source": source,
|
|
"type": type_,
|
|
"rule": rule or Rule(),
|
|
"permission": permission or Permission(),
|
|
"handlers": (
|
|
[
|
|
(
|
|
handler
|
|
if isinstance(handler, Dependent)
|
|
else Dependent[Any].parse(
|
|
call=handler, allow_types=cls.HANDLER_PARAM_TYPES
|
|
)
|
|
)
|
|
for handler in handlers
|
|
]
|
|
if handlers
|
|
else []
|
|
),
|
|
"temp": temp,
|
|
"expire_time": (
|
|
expire_time
|
|
and (
|
|
expire_time
|
|
if isinstance(expire_time, datetime)
|
|
else datetime.now() + expire_time
|
|
)
|
|
),
|
|
"priority": priority,
|
|
"block": block,
|
|
"_default_state": default_state or {},
|
|
"_default_type_updater": (
|
|
default_type_updater
|
|
and (
|
|
default_type_updater
|
|
if isinstance(default_type_updater, Dependent)
|
|
else Dependent[str].parse(
|
|
call=default_type_updater,
|
|
allow_types=cls.HANDLER_PARAM_TYPES,
|
|
)
|
|
)
|
|
),
|
|
"_default_permission_updater": (
|
|
default_permission_updater
|
|
and (
|
|
default_permission_updater
|
|
if isinstance(default_permission_updater, Dependent)
|
|
else Dependent[Permission].parse(
|
|
call=default_permission_updater,
|
|
allow_types=cls.HANDLER_PARAM_TYPES,
|
|
)
|
|
)
|
|
),
|
|
},
|
|
)
|
|
|
|
logger.trace(f"Define new matcher {NewMatcher}")
|
|
|
|
matchers[priority].append(NewMatcher)
|
|
|
|
return NewMatcher # type: ignore
|
|
|
|
@classmethod
|
|
def destroy(cls) -> None:
|
|
"""销毁当前的事件响应器"""
|
|
matchers[cls.priority].remove(cls)
|
|
|
|
@classproperty
|
|
def plugin(cls) -> Optional["Plugin"]:
|
|
"""事件响应器所在插件"""
|
|
return cls._source and cls._source.plugin
|
|
|
|
@classproperty
|
|
def plugin_id(cls) -> Optional[str]:
|
|
"""事件响应器所在插件标识符"""
|
|
return cls._source and cls._source.plugin_id
|
|
|
|
@classproperty
|
|
def plugin_name(cls) -> Optional[str]:
|
|
"""事件响应器所在插件名"""
|
|
return cls._source and cls._source.plugin_name
|
|
|
|
@classproperty
|
|
def module(cls) -> Optional[ModuleType]:
|
|
"""事件响应器所在插件模块"""
|
|
return cls._source and cls._source.module
|
|
|
|
@classproperty
|
|
def module_name(cls) -> Optional[str]:
|
|
"""事件响应器所在插件模块路径"""
|
|
return cls._source and cls._source.module_name
|
|
|
|
@classmethod
|
|
async def check_perm(
|
|
cls,
|
|
bot: Bot,
|
|
event: Event,
|
|
stack: Optional[AsyncExitStack] = None,
|
|
dependency_cache: Optional[T_DependencyCache] = None,
|
|
) -> bool:
|
|
"""检查是否满足触发权限
|
|
|
|
参数:
|
|
bot: Bot 对象
|
|
event: 上报事件
|
|
stack: 异步上下文栈
|
|
dependency_cache: 依赖缓存
|
|
|
|
返回:
|
|
是否满足权限
|
|
"""
|
|
event_type = event.get_type()
|
|
return event_type == (cls.type or event_type) and await cls.permission(
|
|
bot, event, stack, dependency_cache
|
|
)
|
|
|
|
@classmethod
|
|
async def check_rule(
|
|
cls,
|
|
bot: Bot,
|
|
event: Event,
|
|
state: T_State,
|
|
stack: Optional[AsyncExitStack] = None,
|
|
dependency_cache: Optional[T_DependencyCache] = None,
|
|
) -> bool:
|
|
"""检查是否满足匹配规则
|
|
|
|
参数:
|
|
bot: Bot 对象
|
|
event: 上报事件
|
|
state: 当前状态
|
|
stack: 异步上下文栈
|
|
dependency_cache: 依赖缓存
|
|
|
|
返回:
|
|
是否满足匹配规则
|
|
"""
|
|
event_type = event.get_type()
|
|
return event_type == (cls.type or event_type) and await cls.rule(
|
|
bot, event, state, stack, dependency_cache
|
|
)
|
|
|
|
@classmethod
|
|
def type_updater(cls, func: T_TypeUpdater) -> T_TypeUpdater:
|
|
"""装饰一个函数来更改当前事件响应器的默认响应事件类型更新函数
|
|
|
|
参数:
|
|
func: 响应事件类型更新函数
|
|
"""
|
|
cls._default_type_updater = Dependent[str].parse(
|
|
call=func, allow_types=cls.HANDLER_PARAM_TYPES
|
|
)
|
|
return func
|
|
|
|
@classmethod
|
|
def permission_updater(cls, func: T_PermissionUpdater) -> T_PermissionUpdater:
|
|
"""装饰一个函数来更改当前事件响应器的默认会话权限更新函数
|
|
|
|
参数:
|
|
func: 会话权限更新函数
|
|
"""
|
|
cls._default_permission_updater = Dependent[Permission].parse(
|
|
call=func, allow_types=cls.HANDLER_PARAM_TYPES
|
|
)
|
|
return func
|
|
|
|
@classmethod
|
|
def append_handler(
|
|
cls, handler: T_Handler, parameterless: Optional[Iterable[Any]] = None
|
|
) -> Dependent[Any]:
|
|
handler_ = Dependent[Any].parse(
|
|
call=handler,
|
|
parameterless=parameterless,
|
|
allow_types=cls.HANDLER_PARAM_TYPES,
|
|
)
|
|
cls.handlers.append(handler_)
|
|
return handler_
|
|
|
|
@classmethod
|
|
def handle(
|
|
cls, parameterless: Optional[Iterable[Any]] = None
|
|
) -> Callable[[T_Handler], T_Handler]:
|
|
"""装饰一个函数来向事件响应器直接添加一个处理函数
|
|
|
|
参数:
|
|
parameterless: 非参数类型依赖列表
|
|
"""
|
|
|
|
def _decorator(func: T_Handler) -> T_Handler:
|
|
cls.append_handler(func, parameterless=parameterless)
|
|
return func
|
|
|
|
return _decorator
|
|
|
|
@classmethod
|
|
def receive(
|
|
cls, id: str = "", parameterless: Optional[Iterable[Any]] = None
|
|
) -> Callable[[T_Handler], T_Handler]:
|
|
"""装饰一个函数来指示 NoneBot 在接收用户新的一条消息后继续运行该函数
|
|
|
|
参数:
|
|
id: 消息 ID
|
|
parameterless: 非参数类型依赖列表
|
|
"""
|
|
|
|
async def _receive(event: Event, matcher: "Matcher") -> None:
|
|
matcher.set_target(RECEIVE_KEY.format(id=id))
|
|
if matcher.get_target() == RECEIVE_KEY.format(id=id):
|
|
matcher.set_receive(id, event)
|
|
return
|
|
if matcher.get_receive(id, ...) is not ...:
|
|
return
|
|
await matcher.reject()
|
|
|
|
_parameterless = (Depends(_receive), *(parameterless or ()))
|
|
|
|
def _decorator(func: T_Handler) -> T_Handler:
|
|
if cls.handlers and cls.handlers[-1].call is func:
|
|
func_handler = cls.handlers[-1]
|
|
new_handler = Dependent(
|
|
call=func_handler.call,
|
|
params=func_handler.params,
|
|
parameterless=Dependent.parse_parameterless(
|
|
tuple(_parameterless), cls.HANDLER_PARAM_TYPES
|
|
)
|
|
+ func_handler.parameterless,
|
|
)
|
|
cls.handlers[-1] = new_handler
|
|
else:
|
|
cls.append_handler(func, parameterless=_parameterless)
|
|
|
|
return func
|
|
|
|
return _decorator
|
|
|
|
@classmethod
|
|
def got(
|
|
cls,
|
|
key: str,
|
|
prompt: Optional[Union[str, Message, MessageSegment, MessageTemplate]] = None,
|
|
parameterless: Optional[Iterable[Any]] = None,
|
|
) -> Callable[[T_Handler], T_Handler]:
|
|
"""装饰一个函数来指示 NoneBot 获取一个参数 `key`
|
|
|
|
当要获取的 `key` 不存在时接收用户新的一条消息再运行该函数,
|
|
如果 `key` 已存在则直接继续运行
|
|
|
|
参数:
|
|
key: 参数名
|
|
prompt: 在参数不存在时向用户发送的消息
|
|
parameterless: 非参数类型依赖列表
|
|
"""
|
|
|
|
async def _key_getter(event: Event, matcher: "Matcher"):
|
|
matcher.set_target(ARG_KEY.format(key=key))
|
|
if matcher.get_target() == ARG_KEY.format(key=key):
|
|
matcher.set_arg(key, event.get_message())
|
|
return
|
|
if matcher.get_arg(key, ...) is not ...:
|
|
return
|
|
await matcher.reject(prompt)
|
|
|
|
_parameterless = (Depends(_key_getter), *(parameterless or ()))
|
|
|
|
def _decorator(func: T_Handler) -> T_Handler:
|
|
if cls.handlers and cls.handlers[-1].call is func:
|
|
func_handler = cls.handlers[-1]
|
|
new_handler = Dependent(
|
|
call=func_handler.call,
|
|
params=func_handler.params,
|
|
parameterless=Dependent.parse_parameterless(
|
|
tuple(_parameterless), cls.HANDLER_PARAM_TYPES
|
|
)
|
|
+ func_handler.parameterless,
|
|
)
|
|
cls.handlers[-1] = new_handler
|
|
else:
|
|
cls.append_handler(func, parameterless=_parameterless)
|
|
|
|
return func
|
|
|
|
return _decorator
|
|
|
|
@classmethod
|
|
async def send(
|
|
cls,
|
|
message: Union[str, Message, MessageSegment, MessageTemplate],
|
|
**kwargs: Any,
|
|
) -> Any:
|
|
"""发送一条消息给当前交互用户
|
|
|
|
参数:
|
|
message: 消息内容
|
|
kwargs: {ref}`nonebot.adapters.Bot.send` 的参数,
|
|
请参考对应 adapter 的 bot 对象 api
|
|
"""
|
|
bot = current_bot.get()
|
|
event = current_event.get()
|
|
state = current_matcher.get().state
|
|
if isinstance(message, MessageTemplate):
|
|
_message = message.format(**state)
|
|
else:
|
|
_message = message
|
|
return await bot.send(event=event, message=_message, **kwargs)
|
|
|
|
@classmethod
|
|
async def finish(
|
|
cls,
|
|
message: Optional[Union[str, Message, MessageSegment, MessageTemplate]] = None,
|
|
**kwargs,
|
|
) -> NoReturn:
|
|
"""发送一条消息给当前交互用户并结束当前事件响应器
|
|
|
|
参数:
|
|
message: 消息内容
|
|
kwargs: {ref}`nonebot.adapters.Bot.send` 的参数,
|
|
请参考对应 adapter 的 bot 对象 api
|
|
"""
|
|
if message is not None:
|
|
await cls.send(message, **kwargs)
|
|
raise FinishedException
|
|
|
|
@classmethod
|
|
async def pause(
|
|
cls,
|
|
prompt: Optional[Union[str, Message, MessageSegment, MessageTemplate]] = None,
|
|
**kwargs,
|
|
) -> NoReturn:
|
|
"""发送一条消息给当前交互用户并暂停事件响应器,在接收用户新的一条消息后继续下一个处理函数
|
|
|
|
参数:
|
|
prompt: 消息内容
|
|
kwargs: {ref}`nonebot.adapters.Bot.send` 的参数,
|
|
请参考对应 adapter 的 bot 对象 api
|
|
"""
|
|
if prompt is not None:
|
|
await cls.send(prompt, **kwargs)
|
|
raise PausedException
|
|
|
|
@classmethod
|
|
async def reject(
|
|
cls,
|
|
prompt: Optional[Union[str, Message, MessageSegment, MessageTemplate]] = None,
|
|
**kwargs,
|
|
) -> NoReturn:
|
|
"""最近使用 `got` / `receive` 接收的消息不符合预期,
|
|
发送一条消息给当前交互用户并将当前事件处理流程中断在当前位置,在接收用户新的一个事件后从头开始执行当前处理函数
|
|
|
|
参数:
|
|
prompt: 消息内容
|
|
kwargs: {ref}`nonebot.adapters.Bot.send` 的参数,
|
|
请参考对应 adapter 的 bot 对象 api
|
|
"""
|
|
if prompt is not None:
|
|
await cls.send(prompt, **kwargs)
|
|
raise RejectedException
|
|
|
|
@classmethod
|
|
async def reject_arg(
|
|
cls,
|
|
key: str,
|
|
prompt: Optional[Union[str, Message, MessageSegment, MessageTemplate]] = None,
|
|
**kwargs,
|
|
) -> NoReturn:
|
|
"""最近使用 `got` 接收的消息不符合预期,
|
|
发送一条消息给当前交互用户并将当前事件处理流程中断在当前位置,在接收用户新的一条消息后从头开始执行当前处理函数
|
|
|
|
参数:
|
|
key: 参数名
|
|
prompt: 消息内容
|
|
kwargs: {ref}`nonebot.adapters.Bot.send` 的参数,
|
|
请参考对应 adapter 的 bot 对象 api
|
|
"""
|
|
matcher = current_matcher.get()
|
|
matcher.set_target(ARG_KEY.format(key=key))
|
|
if prompt is not None:
|
|
await cls.send(prompt, **kwargs)
|
|
raise RejectedException
|
|
|
|
@classmethod
|
|
async def reject_receive(
|
|
cls,
|
|
id: str = "",
|
|
prompt: Optional[Union[str, Message, MessageSegment, MessageTemplate]] = None,
|
|
**kwargs,
|
|
) -> NoReturn:
|
|
"""最近使用 `receive` 接收的消息不符合预期,
|
|
发送一条消息给当前交互用户并将当前事件处理流程中断在当前位置,在接收用户新的一个事件后从头开始执行当前处理函数
|
|
|
|
参数:
|
|
id: 消息 id
|
|
prompt: 消息内容
|
|
kwargs: {ref}`nonebot.adapters.Bot.send` 的参数,
|
|
请参考对应 adapter 的 bot 对象 api
|
|
"""
|
|
matcher = current_matcher.get()
|
|
matcher.set_target(RECEIVE_KEY.format(id=id))
|
|
if prompt is not None:
|
|
await cls.send(prompt, **kwargs)
|
|
raise RejectedException
|
|
|
|
@classmethod
|
|
def skip(cls) -> NoReturn:
|
|
"""跳过当前事件处理函数,继续下一个处理函数
|
|
|
|
通常在事件处理函数的依赖中使用。
|
|
"""
|
|
raise SkippedException
|
|
|
|
@overload
|
|
def get_receive(self, id: str) -> Union[Event, None]: ...
|
|
|
|
@overload
|
|
def get_receive(self, id: str, default: T) -> Union[Event, T]: ...
|
|
|
|
def get_receive(
|
|
self, id: str, default: Optional[T] = None
|
|
) -> Optional[Union[Event, T]]:
|
|
"""获取一个 `receive` 事件
|
|
|
|
如果没有找到对应的事件,返回 `default` 值
|
|
"""
|
|
return self.state.get(RECEIVE_KEY.format(id=id), default)
|
|
|
|
def set_receive(self, id: str, event: Event) -> None:
|
|
"""设置一个 `receive` 事件"""
|
|
self.state[RECEIVE_KEY.format(id=id)] = event
|
|
self.state[LAST_RECEIVE_KEY] = event
|
|
|
|
@overload
|
|
def get_last_receive(self) -> Union[Event, None]: ...
|
|
|
|
@overload
|
|
def get_last_receive(self, default: T) -> Union[Event, T]: ...
|
|
|
|
def get_last_receive(
|
|
self, default: Optional[T] = None
|
|
) -> Optional[Union[Event, T]]:
|
|
"""获取最近一次 `receive` 事件
|
|
|
|
如果没有事件,返回 `default` 值
|
|
"""
|
|
return self.state.get(LAST_RECEIVE_KEY, default)
|
|
|
|
@overload
|
|
def get_arg(self, key: str) -> Union[Message, None]: ...
|
|
|
|
@overload
|
|
def get_arg(self, key: str, default: T) -> Union[Message, T]: ...
|
|
|
|
def get_arg(
|
|
self, key: str, default: Optional[T] = None
|
|
) -> Optional[Union[Message, T]]:
|
|
"""获取一个 `got` 消息
|
|
|
|
如果没有找到对应的消息,返回 `default` 值
|
|
"""
|
|
return self.state.get(ARG_KEY.format(key=key), default)
|
|
|
|
def set_arg(self, key: str, message: Message) -> None:
|
|
"""设置一个 `got` 消息"""
|
|
self.state[ARG_KEY.format(key=key)] = message
|
|
|
|
def set_target(self, target: str, cache: bool = True) -> None:
|
|
if cache:
|
|
self.state[REJECT_CACHE_TARGET] = target
|
|
else:
|
|
self.state[REJECT_TARGET] = target
|
|
|
|
@overload
|
|
def get_target(self) -> Union[str, None]: ...
|
|
|
|
@overload
|
|
def get_target(self, default: T) -> Union[str, T]: ...
|
|
|
|
def get_target(self, default: Optional[T] = None) -> Optional[Union[str, T]]:
|
|
return self.state.get(REJECT_TARGET, default)
|
|
|
|
def stop_propagation(self):
|
|
"""阻止事件传播"""
|
|
self.block = True
|
|
|
|
async def update_type(
|
|
self,
|
|
bot: Bot,
|
|
event: Event,
|
|
stack: Optional[AsyncExitStack] = None,
|
|
dependency_cache: Optional[T_DependencyCache] = None,
|
|
) -> str:
|
|
updater = self.__class__._default_type_updater
|
|
return (
|
|
await updater(
|
|
bot=bot,
|
|
event=event,
|
|
state=self.state,
|
|
matcher=self,
|
|
stack=stack,
|
|
dependency_cache=dependency_cache,
|
|
)
|
|
if updater
|
|
else "message"
|
|
)
|
|
|
|
async def update_permission(
|
|
self,
|
|
bot: Bot,
|
|
event: Event,
|
|
stack: Optional[AsyncExitStack] = None,
|
|
dependency_cache: Optional[T_DependencyCache] = None,
|
|
) -> Permission:
|
|
if updater := self.__class__._default_permission_updater:
|
|
return await updater(
|
|
bot=bot,
|
|
event=event,
|
|
state=self.state,
|
|
matcher=self,
|
|
stack=stack,
|
|
dependency_cache=dependency_cache,
|
|
)
|
|
return Permission(User.from_event(event, perm=self.permission))
|
|
|
|
async def resolve_reject(self):
|
|
handler = current_handler.get()
|
|
self.remain_handlers.insert(0, handler)
|
|
if REJECT_CACHE_TARGET in self.state:
|
|
self.state[REJECT_TARGET] = self.state[REJECT_CACHE_TARGET]
|
|
|
|
@contextmanager
|
|
def ensure_context(self, bot: Bot, event: Event):
|
|
b_t = current_bot.set(bot)
|
|
e_t = current_event.set(event)
|
|
m_t = current_matcher.set(self)
|
|
try:
|
|
yield
|
|
finally:
|
|
current_bot.reset(b_t)
|
|
current_event.reset(e_t)
|
|
current_matcher.reset(m_t)
|
|
|
|
async def simple_run(
|
|
self,
|
|
bot: Bot,
|
|
event: Event,
|
|
state: T_State,
|
|
stack: Optional[AsyncExitStack] = None,
|
|
dependency_cache: Optional[T_DependencyCache] = None,
|
|
):
|
|
logger.trace(
|
|
f"{self} run with incoming args: "
|
|
f"bot={bot}, event={event!r}, state={state!r}"
|
|
)
|
|
|
|
def _handle_stop_propagation(exc_group: BaseExceptionGroup[StopPropagation]):
|
|
self.block = True
|
|
|
|
with self.ensure_context(bot, event):
|
|
try:
|
|
with catch({StopPropagation: _handle_stop_propagation}):
|
|
# Refresh preprocess state
|
|
self.state.update(state)
|
|
|
|
while self.remain_handlers:
|
|
handler = self.remain_handlers.pop(0)
|
|
current_handler.set(handler)
|
|
logger.debug(f"Running handler {handler}")
|
|
|
|
def _handle_skipped(
|
|
exc_group: BaseExceptionGroup[SkippedException],
|
|
):
|
|
logger.debug(f"Handler {handler} skipped")
|
|
|
|
with catch({SkippedException: _handle_skipped}):
|
|
await handler(
|
|
matcher=self,
|
|
bot=bot,
|
|
event=event,
|
|
state=self.state,
|
|
stack=stack,
|
|
dependency_cache=dependency_cache,
|
|
)
|
|
finally:
|
|
logger.info(f"{self} running complete")
|
|
|
|
# 运行handlers
|
|
async def run(
|
|
self,
|
|
bot: Bot,
|
|
event: Event,
|
|
state: T_State,
|
|
stack: Optional[AsyncExitStack] = None,
|
|
dependency_cache: Optional[T_DependencyCache] = None,
|
|
):
|
|
exc: Optional[Union[FinishedException, RejectedException, PausedException]] = (
|
|
None
|
|
)
|
|
|
|
def _handle_special_exception(
|
|
exc_group: BaseExceptionGroup[
|
|
Union[FinishedException, RejectedException, PausedException]
|
|
],
|
|
):
|
|
nonlocal exc
|
|
excs = list(flatten_exception_group(exc_group))
|
|
if len(excs) > 1:
|
|
logger.warning(
|
|
"Multiple session control exceptions occurred. "
|
|
"NoneBot will choose the proper one."
|
|
)
|
|
finished_exc = next(
|
|
(e for e in excs if isinstance(e, FinishedException)),
|
|
None,
|
|
)
|
|
rejected_exc = next(
|
|
(e for e in excs if isinstance(e, RejectedException)),
|
|
None,
|
|
)
|
|
paused_exc = next(
|
|
(e for e in excs if isinstance(e, PausedException)),
|
|
None,
|
|
)
|
|
exc = finished_exc or rejected_exc or paused_exc
|
|
elif isinstance(
|
|
excs[0], (FinishedException, RejectedException, PausedException)
|
|
):
|
|
exc = excs[0]
|
|
|
|
with catch(
|
|
{
|
|
(
|
|
FinishedException,
|
|
RejectedException,
|
|
PausedException,
|
|
): _handle_special_exception
|
|
}
|
|
):
|
|
await self.simple_run(bot, event, state, stack, dependency_cache)
|
|
|
|
if isinstance(exc, FinishedException):
|
|
pass
|
|
elif isinstance(exc, RejectedException):
|
|
await self.resolve_reject()
|
|
type_ = await self.update_type(bot, event, stack, dependency_cache)
|
|
permission = await self.update_permission(
|
|
bot, event, stack, dependency_cache
|
|
)
|
|
|
|
self.new(
|
|
type_,
|
|
Rule(),
|
|
permission,
|
|
self.remain_handlers,
|
|
temp=True,
|
|
priority=0,
|
|
block=True,
|
|
source=self.__class__._source,
|
|
expire_time=bot.config.session_expire_timeout,
|
|
default_state=self.state,
|
|
default_type_updater=self.__class__._default_type_updater,
|
|
default_permission_updater=self.__class__._default_permission_updater,
|
|
)
|
|
elif isinstance(exc, PausedException):
|
|
type_ = await self.update_type(bot, event, stack, dependency_cache)
|
|
permission = await self.update_permission(
|
|
bot, event, stack, dependency_cache
|
|
)
|
|
|
|
self.new(
|
|
type_,
|
|
Rule(),
|
|
permission,
|
|
self.remain_handlers,
|
|
temp=True,
|
|
priority=0,
|
|
block=True,
|
|
source=self.__class__._source,
|
|
expire_time=bot.config.session_expire_timeout,
|
|
default_state=self.state,
|
|
default_type_updater=self.__class__._default_type_updater,
|
|
default_permission_updater=self.__class__._default_permission_updater,
|
|
)
|