mirror of
https://github.com/nonebot/nonebot2.git
synced 2024-11-24 17:15:05 +08:00
26 lines
954 B
Python
26 lines
954 B
Python
from nonebot import on_command
|
||
from nonebot.rule import to_me
|
||
from nonebot.typing import T_State
|
||
from nonebot.adapters import Bot, Event
|
||
|
||
weather = on_command("天气", rule=to_me(), priority=1)
|
||
|
||
|
||
@weather.handle()
|
||
async def handle_first_receive(bot: Bot, event: Event, state: T_State):
|
||
args = str(event.get_message()).strip() # 首次发送命令时跟随的参数，例：/天气 上海，则args为上海
|
||
print(f"==={args}===")
|
||
if args:
|
||
state["city"] = args # 如果用户发送了参数则直接赋值
|
||
|
||
|
||
@weather.got("city", prompt="你想查询哪个城市的天气呢？")
|
||
async def handle_city(bot: Bot, state: T_State):
|
||
city = state["city"]
|
||
if city not in ["上海", "北京"]:
|
||
await weather.reject("你想查询的城市暂不支持，请重新输入！")
|
||
# weather = await get_weather_from_xxx(city)
|
||
city_weather = "晴天"
|
||
# await bot.send(city_weather)
|
||
await weather.finish(city_weather)
|