事件处理函数重载
当我们在编写 NoneBot2 应用时,常常会遇到这样一个问题:该怎么让同一类型的不同事件执行不同的响应逻辑?又或者如何让不同的 bot
针对同一类型的事件作出不同响应?
针对这个问题, NoneBot2 提供一个便捷而高效的解决方案:事件处理函数重载机制。简单地说,handler
(事件处理函数)会根据其参数的 type hints
(PEP484 类型标注)来对相对应的 bot
和 event
进行响应,并且会忽略不符合其参数类型标注的情况。
提示
如果想了解更多关于 inspect
标准库的信息,可以查看官方文档。
下面,我们会以 OneBot 适配器中的群聊消息事件和私聊消息事件为例,对该机制的应用进行简单的介绍。
一个例子
首先,我们需要导入需要的方法、类型。
from nonebot import on_command
from nonebot.adapters.onebot.v11 import Bot, GroupMessageEvent, PrivateMessageEvent
之后,我们可以注册一个 Matcher
来响应消息事件。
matcher = on_command("test_overload")
最后,我们编写不同的 handler
并编写不同的类型标注来实现事件处理函数重载:
@matcher.handle()
async def _(bot: Bot, event: GroupMessageEvent):
await matcher.send("群聊消息事件响应成功!")
@matcher.handle()
async def _(bot: Bot, event: PrivateMessageEvent):
await matcher.send("私聊消息事件响应成功!")
此时,我们可以在群聊或私聊中对我们的机器人发送 test_overload
,它会在不同的场景做出不同的应答。
这样一个简单的事件处理函数重载就完成了。
进阶
事件处理函数重载机制同样支持被 matcher.got
等装饰器装饰的函数。例如:
@matcher.got("key1", prompt="群事件提问")
async def _(bot: Bot, event: GroupMessageEvent):
await matcher.send("群聊消息事件响应成功!")
@matcher.got("key2", prompt="私聊事件提问")
async def _(bot: Bot, event: PrivateMessageEvent):
await matcher.send("私聊消息事件响应成功!")
只有触发事件符合的函数才会触发装饰器。
注意
bot 和 event 参数具有最高的检查优先级,因此,如果参数类型不符合,所有的依赖项 Depends
等都不会被执行。