nonebot2/docs/guide/creating-a-plugin.md
2020-11-09 12:33:17 +08:00

11 KiB
Raw Blame History

创建插件

如果之前使用 nb-cli 生成了项目结构,那我们已经有了一个空的插件目录 Awesome-Bot/awesome_bot/plugins,并且它已在 bot.py 中被加载,我们现在可以开始创建插件了!

插件通常有两种形式,下面分别介绍

单文件形式

在插件目录下创建名为 foo.py 的 Python 文件,暂时留空,此时目录结构如下:

:::vue AweSome-Bot ├── awesome_bot │ └── plugins │ └── foo.py ├── .env ├── .env.dev ├── .env.prod ├── .gitignore ├── bot.py ├── docker-compose.yml ├── Dockerfile ├── pyproject.toml └── README.md :::

这个时候它已经可以被称为一个插件了,尽管它还什么都没做。

包形式(推荐)

在插件目录下创建文件夹 foo,并在该文件夹下创建文件 __init__.py,此时目录结构如下:

:::vue AweSome-Bot ├── awesome_bot │ └── plugins │ └── foo │ └── __init__.py ├── .env ├── .env.dev ├── .env.prod ├── .gitignore ├── bot.py ├── docker-compose.yml ├── Dockerfile ├── pyproject.toml └── README.md :::

这个时候 foo 就是一个合法的 Python 包了,同时也是合法的 NoneBot 插件,插件内容可以在 __init__.py 中编写。

推荐结构(仅供参考)

:::vue foo ├── __init__.py ├── config.py ├── data_source.py └── model.py :::

__init__.py

在该文件中编写各类事件响应及处理逻辑。

config.py

在该文件中使用 pydantic 定义插件所需要的配置项以及类型。

示例:

from pydantic import BaseSetting


class Config(BaseSetting):

    # plugin custom config
    plugin_setting: str = "default"

    class Config:
        extra = "ignore"

并在 __init__.py 文件中添加以下行

import nonebot
from .config import Config

global_config = nonebot.get_bot().config
plugin_config = Config(**global_config.dict())

此时就可以通过 plugin_config.plugin_setting 获取到插件所需要的配置项了。

data_source.py

在该文件中编写数据获取函数。

:::warning 警告 数据获取应尽量使用异步处理!例如使用 httpx 而非 requests :::

model.py

在该文件中编写数据库模型。

编写真正的内容

好了,现在插件已经可以正确加载,我们可以开始编写命令的实际代码了。在 weather.py 中添加如下代码:

from nonebot import on_command
from nonebot.rule import to_me
from nonebot.adapters.cqhttp import Bot, Event

weather = on_command("天气", rule=to_me(), priority=5)


@weather.handle()
async def handle_first_receive(bot: Bot, event: Event, state: dict):
    args = str(event.message).strip()  # 首次发送命令时跟随的参数,例:/天气 上海则args为上海
    if args:
        state["city"] = args  # 如果用户发送了参数则直接赋值


@weather.got("city", prompt="你想查询哪个城市的天气呢?")
async def handle_city(bot: Bot, event: Event, state: dict):
    city = state["city"]
    if city not in ["上海", "北京"]:
        await weather.reject("你想查询的城市暂不支持,请重新输入!")
    city_weather = await get_weather(city)
    await weather.finish(city_weather)


async def get_weather(city: str):
    return f"{city}的天气是..."

为了简单起见,我们在这里的例子中没有接入真实的天气数据,但要接入也非常简单,你可以使用中国天气网、和风天气等网站提供的 API。

下面我们来说明这段代码是如何工作的。

:::tip 提示 从这里开始,你需要对 Python 的 asyncio 编程有所了解,因为 NoneBot 是完全基于 asyncio 的,具体可以参考 廖雪峰的 Python 教程 :::

注册一个 事件响应器

from nonebot import on_command
from nonebot.rule import to_me
from nonebot.permission import Permission

weather = on_command("天气", rule=to_me(), permission=Permission(), priority=5)

在上方代码中,我们注册了一个事件响应器 Matcher,它由几个部分组成:

  1. on_command 注册一个消息类型的命令处理器
  2. "天气" 指定 command 参数 - 命令名
  3. rule 补充事件响应器的匹配规则
  4. priority 事件响应器优先级
  5. block 是否阻止事件传递

其他详细配置可以参考 API 文档,下面我们详细说明各个部分:

事件响应器类型 type

事件响应器类型其实就是对应 Event.type NoneBot 提供了一个基础类型事件响应器 on() 以及一些内置的事件响应器。

  • on("事件类型"): 基础事件响应器,第一个参数为事件类型,空字符串表示不限
  • on_metaevent() ~ on("meta_event"): 元事件响应器
  • on_message() ~ on("message"): 消息事件响应器
  • on_request() ~ on("request"): 请求事件响应器
  • on_notice() ~ on("notice"): 通知事件响应器
  • on_startswith(str) ~ on("message", startswith(str)): 消息开头匹配处理器
  • on_endswith(str) ~ on("message", endswith(str)): 消息结尾匹配处理器
  • on_command(str|tuple) ~ on("message", command(str|tuple)): 命令处理器
  • on_regex(pattern_str) ~ on("message", regex(pattern_str)): 正则匹配处理器

匹配规则 rule

事件响应器的匹配规则即 Rule,由非负个 RuleChecker 组成,当所有 RuleChecker 返回 True 时匹配成功。这些 RuleChecker 的形式如下:

async def check(bot: Bot, event: Event, state: dict) -> bool:
    return True

def check(bot: Bot, event: Event, state: dict) -> bool:
    return True

RuleRuleChecker 之间可以使用 与 & 互相组合:

from nonebot.rule import Rule

Rule(async_checker1) & sync_checker & async_checker2

:::danger 警告 Rule(*checkers) 只接受 async function或使用 nonebot.utils.run_sync 自行包裹 sync function。在使用 与 &NoneBot 会自动包裹 sync function :::

优先级 priority

事件响应器的优先级代表事件响应器的执行顺序,同一优先级的事件响应器会 同时执行!

:::tip 提示 使用 nonebot-test 可以看到当前所有事件响应器的执行流程,有助理解事件响应流程!

pip install nonebot2[test]

:::

阻断 block

当有任意事件响应器发出了阻止事件传递信号时,该事件将不再会传递给下一优先级,直接结束处理。

NoneBot 内置的事件响应器中,所有 message 类的事件响应器默认会阻断事件传递,其他则不会。

编写事件处理函数 Handler

@weather.handle()
async def handle_first_receive(bot: Bot, event: Event, state: dict):
    args = str(event.message).strip()  # 首次发送命令时跟随的参数,例:/天气 上海则args为上海
    if args:
        state["city"] = args  # 如果用户发送了参数则直接赋值


@weather.got("city", prompt="你想查询哪个城市的天气呢?")
async def handle_city(bot: Bot, event: Event, state: dict):
    city = state["city"]
    if city not in ["上海", "北京"]:
        await weather.reject("你想查询的城市暂不支持,请重新输入!")
    city_weather = await get_weather(city)
    await weather.finish(city_weather)

在上面的代码中,我们给 weather 事件响应器添加了两个事件处理函数:handle_first_receive, handle_city

其中有几个要点,我们一一解释:

添加一个事件处理函数

在事件响应器响应事件时,事件处理函数会依次顺序执行,也就是说,与添加顺序一致。

我们可以使用 @matcher.handle() 装饰器来简单地为该事件响应器添加一个处理函数。

同时NoneBot 内置了几种添加事件处理函数方式以方便处理:

  • @matcher.receive(): 指示 NoneBot 接收一条新的用户消息以继续执行后续处理函数。
  • @matcher.got(key, [prompt="请输入key"], [args_parser=function]): 指示 NoneBot 当 state 中不存在 key 时向用户发送 prompt 等待用户回复并赋值给 state[key]

这些装饰器可以套娃使用!例如:

@matcher.got("key1")
@matcher.got("key2")
async def handle(bot: Bot, event: Event, state: dict):
    pass

事件处理函数参数

事件处理函数类型为 Callable[[Bot, Event, dict], Union[Awaitable[None], Awaitable[NoReturn]]]

参数分别为:

  1. nonebot.typing.Bot: 即事件上报连接对应的 Bot 对象,为 BaseBot 的子类。特别注意,此处的类型注释可以替换为指定的 Bot 类型,例如:nonebot.adapters.cqhttp.Bot,只有在上报事件的 Bot 类型与类型注释相符时才会执行该处理函数!可用于多平台进行不同的处理。
  2. nonebot.typing.Event: 即上报事件对象,可以获取到上报的所有信息。
  3. state: 状态字典,可以存储任意的信息

处理事件

在事件处理函数中,我们只需要对 event 做出相应的处理,存入状态字典 state 中,或者向用户发送消息、调用某个机器人 API 等等。

在 NoneBot 中,提供了几种特殊的处理函数:

@matcher.args_parser

这是一个装饰器,装饰一个函数来使它成为参数的默认解析函数,当使用 matcher.got(xxx, [args_parser]) 获取到一条消息时,会运行 matcher.gotargs_parser ,如果不存在则运行 @matcher.args_parser

matcher.pause

这个函数用于结束当前事件处理函数,强制接收一条新的消息再运行下一个消息处理函数

matcher.reject

这个函数用于结束当前事件处理函数,强制接收一条新的消息再再次运行当前消息处理函数

matcher.finish

这个函数用于直接结束当前事件处理。

以上三个函数都拥有一个参数 prompt,用于向用户发送一条消息。

结语

至此,相信你已经能够写出一个基础的插件了,更多的用法将会在 进阶 部分进行介绍,这里给出几个小提示:

  • 请千万注意事件处理器的优先级设定
  • 在匹配规则中请勿使用耗时极长的函数
  • 同一个用户可以跨群(私聊)继续他的事件处理(除非做出权限限制,将在后续介绍)