Как остановить bot.register_next_step_handler?

Question

[micromeow]

Доброго времени суток.
Нужен код, который начинает брать от пользователя информацию из телеграм бота, которую пользователь вводит в чат, для дальнейшей обработки.

Есть такая реализация через bot.register_next_step_handler.

def ur_name(message):
    msg = bot.send_message(message.chat.id, 'Здравствуйте, введите пожалуйста ваше имя') #запрашиваем имя
    bot.register_next_step_handler(msg, ur_secondname) #отправляем ответ в следующую функцию

def ur_secondname(message):
    user_data = {} #создаем словарь
    user_data['firstname'] = message.text #записываем имя в словарь
    msg = bot.send_message(message.chat.id, 'Введите пожалуйста вашу фамилию') #запрашиваем фамилию
    bot.register_next_step_handler(msg, ur_phone, user_data) #отправляем ответ в следующую функцию, где так же записываем фамилию и просим телефон и так далее

В общем все работает, но есть одно но, например если пользователь сделает опечатку или захочет остановить процесс, то последующие команды как /stop или повторный вызов любой команды будет восприниматься как текст и записываться в словарь, что с этим делать? Возможно есть способ брать информацию от пользователя получше?

Comments

[Алан Гибизов]

См.п.3.4

Answer 1

[Михаил Кростелев]

как один из вариантов, делать проверку на слова триггеры:

def ur_secondname(message):
    if message.text.lower() in ['отмена', 'stop', '/stop', 'cancel']:
        bot.send_message(message.chat.id, 'Отмена пользователем!')
    else:
        user_data = {} #создаем словарь
        ...

Comments

[micromeow]

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

[Михаил Кростелев]

micromeow, сначала лучше сделать рабочий вариант, потом уже делать код чистым.
можно проверку вывести в отдельную функцию, если отмена планируется на всех шагах. Убрать else можно.

def check_for_cancel(text):
    return text.lower() in ['отмена', 'stop', '/stop', 'cancel']


def ur_secondname(message):
    if check_for_cancel(message.text):
        bot.send_message(message.chat.id, 'Отмена пользователем!')
        return
        
    user_data = {} #создаем словарь
    ...

[micromeow]

Михаил Кростелев, Спасибо большое!

[Михаил Кростелев]

micromeow, не за что) Хотите еще чище, оформите проверку как декоратор

Answer 2

[soremix]

Сделать кнопку «Отмена», проверять в начале каждой функции текст сообщения, если он == отмена, то можно просто выйти из функции, отослать другое сообщение, отправить стартовое сообщение и тд и тп

Comments

[micromeow]

спасибо, я сомневался, что это правильное решение, теперь так и поступлю