weather.py

import random

from classes import Message
from monitor.plugin.on import on_command

# priority 为优先级，数值越低优先级越高，block 是否阻断消息继续传递，默认 True，为 False 时还需继续传递至下一层事件处理
weather = on_command("天气", priority=2, block=True)


@weather.handle()
async def handle_first_receive(message: Message, state):
    args = message.strip(state)
    if args:
        if len(args) == 2:
            state["city"], state["time"] = args  # 如果用户发送了参数则直接赋值
        else:
            state['city'] = args[0]


# got 获取参数的值，prompt 为查询不到该参数时自动回复的问题
@weather.got("city", prompt="你想查询哪个城市的天气呢？")
async def handle_city(message, state):
    city = state["city"]  # got 回复会存储到当前state中
    if city not in ["上海", "北京"]:
        # reject 发送一条消息给当前交互用户并暂停事件响应器，在接收用户新的一条消息后重新运行当前处理函数
        await weather.reject("你想查询的城市暂不支持，请重新输入！")


# got 获取参数的值，prompt 为查询不到该参数时自动回复的问题
@weather.got("time", prompt="你想查询几号的？")
async def handle_time(message, state):
    time = state["time"]
    city = state["city"]
    city_weather = await get_weather(city, time)
    await weather.finish(city_weather)


async def get_weather(city: str, time: str):
    _weather = random.choice(['晴天', '雨天', '多云', '大风'])
    return f"{city}{time}的天气是{_weather}"