ProjectHentai/nonebot2
6
0
Fork 0
mirror of https://github.com/nonebot/nonebot2.git synced 2025-07-16 19:11:00 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

⚗️ add got receive overload #142

Browse Source
This commit is contained in:
yanyongyu
2021-01-06 18:29:24 +08:00
parent a211f09a99
commit b6bf477551
2 changed files with 49 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
nonebot/matcher.py
View File

@ -294,18 +294,27 @@ class Matcher(metaclass=MatcherMeta):
* 无
"""
async def _receive(bot: "Bot", event: "Event",
state: T_State) -> NoReturn:
async def _receive(bot: "Bot", event: "Event") -> NoReturn:
raise PausedException
cls.process_handler(_receive)
if cls.handlers:
# 已有前置handlers则接受一条新的消息，否则视为接收初始消息
cls.append_handler(_receive)
def _decorator(func: T_Handler) -> T_Handler:
cls.process_handler(func)
if not cls.handlers or cls.handlers[-1] is not func:
cls.append_handler(func)
_receive.__params__.update({
"bot":
func.__params__["bot"],
"event":
func.__params__["event"] or _receive.__params__["event"]
})
return func
return _decorator
@ -368,6 +377,25 @@ class Matcher(metaclass=MatcherMeta):
cls.append_handler(wrapper)
wrapper.__params__.update({
"bot":
func.__params__["bot"],
"event":
func.__params__["event"] or wrapper.__params__["event"]
})
_key_getter.__params__.update({
"bot":
func.__params__["bot"],
"event":
func.__params__["event"] or wrapper.__params__["event"]
})
_key_parser.__params__.update({
"bot":
func.__params__["bot"],
"event":
func.__params__["event"] or wrapper.__params__["event"]
})
return func
return _decorator