Как правильно использовать FSM в aiogram для обработчика следующего сообщения?

Question

[Gajlk]

Я хочу создать аналог register_next_step_handler из telebot для alogram. Как я уже понял, для этого я должен использовать FSM. Однако я не могу понять, что мне следует делать на этом примере:

@bot.message_handler(func=lambda message: message.text == " Prohibit publication for a while")
def prohibit_sending_messages_button(message):
    msg = bot.send_message(message.chat.id, "Enter the blocking time in minutes (0 for cancellation; multiply by 60/1440 to indicate hours/days):")
    bot.register_next_step_handler(msg, block_sending_for_minutes)

def block_sending_for_minutes(message):
    global acceptable_time
    # Change acceptable_time value to actual time + chosen time (can be mathematical expression)
    acceptable_time = message.date + eval(message.text) * 60
    bot.send_message(message.chat.id, f"The ban on sending messages will be in effect until {datetime.fromtimestamp(acceptable_time).strftime("%H:%M:%S %d/%m/%Y")}")

Я пробовал написать это:

@router.message(is_admin & F.text == " Prohibit publication for a while")
async def prohibit_sending_messages_button(message: Message, state: FSMContext) -> None:
    await state.set_state(Form.blocking_time)
    await message.answer("Enter the blocking time in minutes (0 for cancellation; multiply by 60/1440 to indicate hours/days):")


@router.message(Form.blocking_time)
async def block_sending_for_minutes(message: Message) -> None:
    global acceptable_time
    # Change acceptable_time value to actual time + chosen time (can be mathematical expression)
    acceptable_time = message.date + eval(await message.text) * 60
    await message.answer(f"The ban on sending messages will be in effect until {datetime.fromtimestamp(acceptable_time).strftime("%H:%M:%S %d/%m/%Y")}")

Мои вопросы:

1. Нужно ли мне создавать для этого отдельный класс, если да, то должен ли я поместить все классы FSM в отдельный файл и как правильно назвать класс и переменную?
2. Нужно ли мне использовать state.clear(), если да, то как мне получить state, если нет, то получается, что я просто больше не использую этот класс?

Answer 1

[Wispik]

1. Нужно ли мне создавать для этого отдельный класс

Да, для стейтов задается отдельный класс

2. Нужно ли мне использовать state.clear()

Конечно нужно, если хочешь сбросить стейты.

если да, то как мне получить state

Также, как ты его получил в первой функции, когда устанавливал значение.
state: FSMContext

Вот почитай, все вопросы должны отпасть

global acceptable_time
глобальные переменные зло. Ты же понимаешь, что значение этой переменной одно для всех юзеров?
eval(await message.text)
а это вообще кошмар)) тут надо просто message.text к инту привести, но не евалом же это делать xD

Comments

[Gajlk]

Я не совсем понимаю, как использовать FSM именно как аналог функции из telebot register_next_step_handler.
То есть я создаю отдельный класс для каждого раза, когда мне требуется назначить обработчика следующего сообщения, после чего не использую state.clear(), потому что данных не сохранял? Или я могу создать один класс, с переменной допустим next_step, который я буду использовать для всех случаев, где в telebot я бы использовал register_next_step_handler, также без state.clear()?

глобальные переменные зло. Ты же понимаешь, что значение этой переменной одно для всех юзеров?

Да, понимаю. Это переменная для блокировки отправки сообщений всеми пользователями на время

а это вообще кошмар)) тут надо просто message.text к инту привести, но не евалом же это делать xD

Это старый код, который я переписываю, плюс это сделано для того, чтобы админ мог написать количество минут через 3 * 60, да и функция только для админов.

[Wispik]

Gajlk, смотри в register_next_step_handler ты передаешь функцию, которая должна выполниться на след шаге. А в aiogram ты устанавливаешь state await state.set_state(Form.blocking_time)
и дальше будет выполняться функция, которая обрабатывает этот state
@router.message(Form.blocking_time)
это по сути все отличие.
А state.clear() нужно использовать для того, чтобы сбросить стейт

Прочитай по ссылке, что я в ответе скинул. По мне там довольно понятно все расписано. И просто попытайся понять логику работы fsm, а не проводить параллели с телеботом

[Gajlk]

Wispik, ну я лично понял, что если использовать fsm конкретно для этого, то можно сделать так

class NextMessageHandler(StatesGroup):
    next_step = State()

После чего я просто везде, где мне надо будет назначить обработчик следующего сообщения, буду использовать этот класс через set.state(NextMessageHandler.next_step) без state.clear() потому что данные я не сохраняю. Поправьте если я что-то неправильно понял конкретно для моей задачи

[Wispik]

Gajlk, Неправильно.
state.clear() сбрасывает не только данные, но и сам state - то есть это надо обязательно использовать
Смотри еще раз попробую объяснить:
в register_next_step_handler ты указываешь функцию, которая выполнится ОДИН РАЗ при следующем сообщении от пользователя
в FSM ты меняешь state(состояние) бота и получается функция под @router.message(Form.blocking_time) будет выполняться ПОСТОЯННО, пока ты не изменишь state на новый или не сбросишь его через state.clear()

[Gajlk]

Wispik, я понял, спасибо, еще один вопрос. Корректно ли будет создать один класс с одной переменной для всех одноуровневых fsm, то есть где мне нужно просто назначить обработчик следующего сообщения и возможно передать ему одно значение?

[Wispik]

Gajlk, это будет работать, при условии, что обработчик след сообщения одинаковый. Но хотя бы переменные логично назвать, потому что вот это кошмар:

class NextMessageHandler(StatesGroup):
    next_step = State()

А в плане корректности, надо понимать, что FSM - это состояние бота, а не просто обработка след. шага. Поэтому в идеале, каждый стейт должен быть описан