import re
from nonebot import on_command, CommandSession


@on_command('weather', aliases=['天气', '查天气', '天气查询'])
async def weather(session: CommandSession):
    city = session.get('city', prompt='你想查哪个城市呢？')
    date = session.get('date', prompt='你想查哪一天呢？(格式:20200427)')
    await session.send('你查询的情况如下' + city)
    await session.send('你想查询的日期' + date)


@weather.args_parser
async def _(session: CommandSession):
    if session.is_first_run:
        return

    if session.current_key == 'date':
        if not re.fullmatch(r'\d{8}', session.current_arg_text):
            session.pause('日期格式有误，请重新输入')
    session.args[session.current_key] = session.current_arg_text