nonebot2/tests/test_plugins/test_package/test_command.py

25 lines
893 B
Python
Raw Normal View History

2020-08-23 02:45:26 +00:00
from nonebot.rule import to_me
2020-12-17 13:09:30 +00:00
from nonebot.typing import T_State
2020-08-17 08:09:41 +00:00
from nonebot.plugin import on_command
2020-12-17 13:09:30 +00:00
from nonebot.adapters import Bot, Event
2020-08-17 08:09:41 +00:00
2020-08-23 02:45:26 +00:00
test_command = on_command("帮助", to_me())
2020-08-17 08:09:41 +00:00
@test_command.handle()
2020-12-17 13:09:30 +00:00
async def test_handler(bot: Bot, event: Event, state: T_State):
args = str(event.get_message()).strip()
2020-08-24 09:59:36 +00:00
print("[!] Command:", state["_prefix"], "Args:", args)
2020-08-23 02:45:26 +00:00
if args:
state["help"] = args
else:
2020-08-26 09:47:36 +00:00
await bot.send(message="命令:\n1. test1\n2. test2", event=event)
2020-08-23 02:45:26 +00:00
@test_command.got("help", prompt="你要帮助的命令是?")
2020-12-17 13:09:30 +00:00
async def test_handler(bot: Bot, event: Event, state: T_State):
2020-08-23 02:45:26 +00:00
print("[!] Command 帮助:", state["help"])
if state["help"] not in ["test1", "test2"]:
await test_command.reject(f"{state['help']} 不支持,请重新输入!")
2020-08-26 09:47:36 +00:00
await bot.send(message=f"{state['help']} 帮助:\n...", event=event)