nonebot2/website/versioned_docs/version-2.4.0/appendices/session-state.md
noneflow[bot] 5e86d53e0b 🔖 Release 2.4.0
2024-10-31 13:45:33 +00:00

2.0 KiB
Raw Permalink Blame History

sidebar_position description options
3 会话状态信息
menu
category weight
appendices 40

会话状态

在事件处理流程中,和用户交互的过程即是会话。在会话中,我们可能需要记录一些信息,例如用户的重试次数等等,以便在会话中的不同阶段进行判断和处理。这些信息都可以存储于会话状态中。

NoneBot 中的会话状态是一个字典,可以通过类型 T_State 来获取。字典内可以存储任意类型的数据但是要注意的是NoneBot 本身会在会话状态中存储一些信息,因此不要使用 NoneBot 使用的键名

from nonebot.typing import T_State

@matcher.got("key", prompt="请输入密码")
async def _(state: T_State, key: str = ArgPlainText()):
    if key != "some password":
        try_count = state.get("try_count", 1)
        if try_count >= 3:
            await matcher.finish("密码错误次数过多")
        else:
            state["try_count"] = try_count + 1
            await matcher.reject("密码错误,请重新输入")
    await matcher.finish("密码正确")

会话状态的生命周期与事件处理流程相同,在期间的任何一个事件处理函数都可以进行读写。

from nonebot.typing import T_State

@matcher.handle()
async def _(state: T_State):
    state["key"] = "value"

@matcher.handle()
async def _(state: T_State):
    await matcher.finish(state["key"])

会话状态还可以用于发送动态消息,消息模板在发送时会使用会话状态字典进行渲染。消息模板的使用方法已经在消息处理中介绍过,这里不再赘述。

from nonebot.typing import T_State
from nonebot.adapters import MessageTemplate

@matcher.handle()
async def _(state: T_State):
    state["username"] = "user"

@matcher.got("password", prompt=MessageTemplate("请输入 {username} 的密码"))
async def _():
    await matcher.finish(MessageTemplate("密码为 {password}"))