Как сделать время ожидания ответа на вопрос Aiogram, с использованием State?

Question

[Willesp]

Например: Бот спросил ваше имя и будет ждать ответа на вопрос 60 секунд, если ответа нет он прощается, если ответ есть удаляет задачу.

Изначально я реализовал это через Asyncio -> create_task, если ответ получен завершаю задачу.
Мой код:

async def cancel_wait(task_id):
    for task in asyncio.all_tasks():
        if task.get_name() == task_id:
            task.cancel()

# Ранее я запустил задачу так
@dp.message_handler(commands=['start'])
async def start_bot(message: types.Message, state: FSMContext):
    await User.name.set()
    asyncio.create_task(wait_message(message, state), name=message.from_user.id)
    await message.answer(text="Привет")
# callback
async def wait_message(message: types.Message, state: FSMContext):
    await asyncio.sleep(60)
    await state.finish()
    await message.answer("Жаль, что ты так и не ответил...")

Но я не уверен что это правильная реализация, к тому же если бота перезапустить то все таски сбросятся...

Answer 1

[Даниил Рыжов]

@dp.message_handler(state=Add.test_state)
@dp.message_handler(commands=['start'])
async def start_bot(message: types.Message, state: FSMContext):
    if message.text == '/start':
        await message.answer(text="Отправь мне свое имя")
        await Add.test_state.set() # Ваш state
        await asyncio.sleep(5) # 5 сек спим
        try:
            data = await state.get_data()
            if data['get_name'] == 'true':
                pass
        except KeyError: 
            # Если пользователь не ответил или за это время state завершился, получаем KeyError
            await message.answer(f'Жаль, что ты не ответил')
            await state.finish()
    else:
        await state.update_data(get_name='true')
        await message.answer(f'Твое имя {message.text}')
        # Установите следующие состояние
        # Если вы завершите состояние, то тогда бот ответит  'Жаль, что ты не ответил'

Как отработало

Comments

[enabl3]

А как тут просто завершить состояние, что бы не было ответа?

else:
        await state.update_data(get_name='true')
        await message.answer(f'Твое имя {message.text}')
        # Установите следующие состояние
        # Если вы завершите состояние, то тогда бот ответит  'Жаль, что ты не ответил'

По идее тут уже поступил ответ и если этого достаточно можно просто await state.finish() но все равно приходит уже ненужный ответ

[markinmarch]

этот код хорошо работает и к сожалению - это единственное, что работает
я даже пытался использовать Timer из библиотеки threading, но получалась какая-то лажа

Answer 2

[Abraham lary]

используйте asyncio.sleep(60) а потом обработайте

Comments

[Abraham lary]

сначала ответил а потом дочитал код