ProjectHentai/nonebot2
6
0
Fork 0
mirror of https://github.com/nonebot/nonebot2.git synced 2025-07-27 08:11:38 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

🎨 impove typing

Browse Source
This commit is contained in:
yanyongyu
2020-12-07 00:06:09 +08:00
parent 629eed08b6
commit 6b3f6a46e6
27 changed files with 60 additions and 137 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
tests/test_plugins/test_weather.py
View File

@ -1,12 +1,13 @@
from nonebot import on_command
from nonebot.rule import to_me
from nonebot.typing import Bot, Event
from nonebot.typing import 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: dict):
async def handle_first_receive(bot: Bot, event: Event, state: State):
args = str(event.message).strip() # 首次发送命令时跟随的参数，例：/天气 上海，则args为上海
print(f"==={args}===")
if args:
@ -14,7 +15,7 @@ async def handle_first_receive(bot: Bot, event: Event, state: dict):
@weather.got("city", prompt="你想查询哪个城市的天气呢？")
async def handle_city(bot: Bot, event: Event, state: dict):
async def handle_city(bot: Bot, event: Event, state: State):
city = state["city"]
if city not in ["上海", "北京"]:
await weather.reject("你想查询的城市暂不支持，请重新输入！")