Как остановить handler?

Question

[xx_RuBiCoN_xx]

Есть файл start.py. Вот часть кода из него:

#Всякий нужный код до этого момента
    elif message.text == text42:
        handle_adm_btn1(bot, message)
    elif message.text == text43:
        handle_adm_btn2(bot, message)
#Всякий нужный код после этого момента

При этом есть ещё два файла: userAdd.py и userDelete.py
В них написанны функции по добавлению и удалению пользователя в таблице MySQL.

Вот часть кода из userAdd.py:

def handle_table_selection(bot, message, table_name):
    # Отправляем сообщение с запросом ввода номера
    bot.send_message(message.chat.id, text40)
    # Устанавливаем состояние пользователя в ожидание ввода номера
    bot.register_next_step_handler(message, lambda msg: handle_number_input(bot, msg, table_name))

def handle_number_input(bot, message, table_name):
    number = message.text
    cursor = mydb.cursor()
    sql_query = f"INSERT INTO {table_name} (phone_number, is_try) VALUES (%s, %s)"
    values = (number, 1)
    cursor.execute(sql_query, values)
    mydb.commit()
    cursor.close()
    bot.send_message(message.chat.id, text41)

Вот часть кода из userDelete.py:

def handle_table_selection2(bot, message, table_name):
    # Отправляем сообщение с запросом ввода номера
    bot.send_message(message.chat.id, text40)
    # Устанавливаем состояние пользователя в ожидание ввода номера
    bot.register_next_step_handler(message, lambda msg: handle_number_input2(bot, msg, table_name))

def handle_number_input2(bot, message, table_name):
    number = message.text
    cursor = mydb.cursor()
    sql_query = f"DELETE FROM {table_name} WHERE phone_number = %s"
    values = (number,)
    cursor.execute(sql_query, values)
    mydb.commit()
    cursor.close()
    bot.send_message(message.chat.id, text45)

Проблема заключается в том, что если я сначала добавляю пользователя в базу через userAdd, а потом хочу удалить другого юзера из базы - то на моменте

bot.register_next_step_handler(message, lambda msg: handle_number_input2(bot, msg, table_name))

происходит тот хандлер, который я использовал первым.

То есть, если я сначала добавил юзера, а потом удаляю - то именно в этом месте во время удаления вызовется хандлер добавления юзера. И наоборот, соответственно.

Изза этого ты можешь делать только то, что выбрал первым.

Вопрос как остановить этот хандлер или как изменить функцию, чтобы этого не происходило?

Comments

[shurshur]

bot.cancel_step_handler(message)

или:

bot.cancel_step_handler_by_chat_id(message.chat.id)

[xx_RuBiCoN_xx]

shurshur, выбивает ошибку:

AttributeError: 'TeleBot' object has no attribute 'cancel_step_handler'. Did you mean: 'clear_step_handler'?

Попробовал и bot.clear_step_handler(message) и bot.clear_step_handler_by_chat_id(message.chat.id) - тогда ошибок нет, но хандлер всё равно активен прошлый.

UPD:
Если это вставить сразу после хандлера - то он останавливается сразу ещё не выполнившись. А если в следующей функции - то он уже не останавливается

[xx_RuBiCoN_xx]

Кучу чего перепробовал, самое оригинальное было останавливать хандер, а потом вызывать некст степ, что прям точно должно было сработать - но нет. Всё равно вызывается тот первый хандлер :(

[xx_RuBiCoN_xx]

Методом проб и ошибок, понял, что после добавления юзера, когда хочу далить следующего то на моменте когда выбрал таблицу и тебя перекидывает в следующую функцию вот таким методом:
handle_table_selection2(bot, message, table_name)
На самом деле отправляет в handle_table_selection(bot, message, table_name) (без двойки, т.е. в функцию добавления юзера, а не удаления)

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

[shurshur]

xx_RuBiCoN_xx, да, вечно вместо clear_ вспоминаю cancel_.

Не должно быть никаких "методом проб и ошибок". Надо чётко понимать, что делает код.

Принцип очень простой. В обычном обработчике вешаем следующий (register_next_step_handler) и дальше по цепочке следующие. Как только надо "выкинуть в главное меню", делаем clear_step_handler (в теории можно было бы сделать register_next_step_handler на самый главный обработчик, но это не очень хорошо, так как при clear освобождается занятая под состояние память, а тут нет).

[xx_RuBiCoN_xx]

shurshur, ну вот если сразу после register_next_step_handler вписать clear_step_handler - то он сразу откидывается даже не получив ответа от пользователя.

А если clear_step_handler выполнить после def handle_number_input то он как бы выполняется, но при попытке вызвать def handle_number_input2 вызыается опять def handle_number_input

[shurshur]

xx_RuBiCoN_xx, не надо сразу, надо в следующем обработчике.

[xx_RuBiCoN_xx]

shurshur, я и в следующей функции пробовал, и непосредственно пере вызовом самого register_next_step_handler.

Тут проблема кажется даже не в хандлере а в том, что функция вызывается АДД, а не Делете, хотя прописана как АДД.

Потому как далее идёт всё по коду АДД. Ну при условии если адд выбирался первым

[shurshur]

xx_RuBiCoN_xx, ну вот в handle_number_input надо сделать clear. И в handle_number_input2 надо сделать clear. После того, как операция сделана, нужно сброситься до изначального состояния. Чтобы сделующее действие не вызывало handle_number_input или handle_number_input2, а вызывало основной обработчик.

А вообще, конечно, названия типа handle_number_input2 это плохо. Должно быть сразу понятно, что это обработчик, который удаляет. Типа handle_number_delete_from_table.

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

elif message.text == text42:
        handle_adm_btn1(bot, message)

[xx_RuBiCoN_xx]

shurshur, что имеется в виду под "сброситься до изначального состояния"? Разве это не clear после handle_number_input?

UPD: В админке всего две кнопки adm_btn1 и adm_btn2. Соответсвенно я и поделил функции таким же методом. А все текста я вывел в отдельный файл. Потому что этот бот, вероятно, может уйти в пользование человека далёкого от программирования. И если изменится название услуг или будет нужна коррекция названия кнопок - он сможет открыть один файл и поменять всё что ему нужно.

[shurshur]

xx_RuBiCoN_xx,
telebot (как и вообще большинство библиотек для написания чат-ботов) работает как обработчик поступающих событий. Приходит событие - вызывается его обработчик. И это очень хорошо работает, пока всё осмысленное взаимодействие с ботом локализовано одним сообщением, а предыдущие знать вообще не обязательно. Ну, например, бот-переводчик: мы кидаем ему фразу на одном языке, он вовзращает на другом языке.

Когда же поведение при обработке конкретного сообщения начинает находиться в зависимости от предыстории, то уже не так просто. Нужно для каждого пользователя (точнее, каждого chat_id) помнить некое состояние, которое определяет поведение на текущем шаге.

На самом деле хранить состояния своими силами тоже можно, и я бы посоветовал в учебных целях хотя бы один раз это попробовать. Заводим глобальную переменную-словарь user_statses = dict(), куда будем записывать состояния пользователей.

user_state = user_states.get(message.chat.id, None)
if user_state:
    if user_state["action"] = "button1": функция_обработчик_кнопки1(message)
    elif user_state["action"] = "button2": функция_обработчик_кнопк2(message)
    else: функция_которая_покажет_ошибку(message)
else:
    поведение по умолчанию, например, обработка нажатых кнопок
    в нужном месте выставляем состояние:
    user_state[message.chat.id] = {"action": "button1", "дополнительные": "данные"}

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

В состояние можно даже класть ссылку на функцию, которая будет вызвана:

user_state[message.chat.id]["func"] = функция_обработки_сообщения
#или даже так:
user_state[message.chat.id]["func"] = lambda msg: функция_обработки_сообщения(msg, "дополнительные", "параметры")

Тогда в нужном месте можно вызвать эту функцию:

if "func" in user_states.get(message.chat.id, {}):
    user_states[message.chat.id]["func"](message)

Но при таком подходе код может выглядеть достаточно громоздко. Поэтому библиотеки предлагают различные способы оптимизировать этот процесс. Вот telebot, например, позволяет использовать register_next_step_handler, который в какой-то свой внутренний словарь сохраняет следующую вызываемую функцию, примерно так же, как приведённом выше примере. Выбор обработчика при обработке очередного сообщения также делает библиотека, поэтому нам его писать не нужно.

Так что любой многошаговый бот просто содержит в каждом обработчике вызов register_next_step_handler, который приводит к изменению обработчика следующего сообщения от этого пользователя. Можно даже с ветвлениями: в зависимости от условий можно зарегистрировать в одной функции разные обработчики.

Если же надо, чтобы бот вернулся "в главное меню" (на самый первый обработчик), то делаем clear.

Answer 1

[xx_RuBiCoN_xx]

Для остановки handler подходит это:

bot.clear_step_handler(message)

или

bot.clear_step_handler_by_chat_id(message.chat.id)

Answer 2

[Medovochka]

Попробуйте так для файла:

# В userAdd.py
def handle_number_input(bot, message, table_name):
    number = message.text
    cursor = mydb.cursor()
    sql_query = f"INSERT INTO {table_name} (phone_number, is_try) VALUES (%s, %s)"
    values = (number, 1)
    cursor.execute(sql_query, values)
    mydb.commit()
    cursor.close()
    bot.send_message(message.chat.id, text41)

def handle_table_selection(bot, message, table_name):
    # Отправляем сообщение с запросом ввода номера
    bot.send_message(message.chat.id, text40)
    # Устанавливаем состояние пользователя в ожидание ввода номера
    bot.register_next_step_handler(message, lambda msg: handle_number_input(bot, msg, table_name))

# В userDelete.py
def handle_number_input2(bot, message, table_name):
    number = message.text
    cursor = mydb.cursor()
    sql_query = f"DELETE FROM {table_name} WHERE phone_number = %s"
    values = (number,)
    cursor.execute(sql_query, values)
    mydb.commit()
    cursor.close()
    bot.send_message(message.chat.id, text45)

def handle_table_selection2(bot, message, table_name):
    # Отправляем сообщение с запросом ввода номера
    bot.send_message(message.chat.id, text40)
    # Устанавливаем состояние пользователя в ожидание ввода номера
    bot.register_next_step_handler(message, lambda msg: handle_number_input2(bot, msg, table_name))

И соотвественно так для start.py :

# Всякий нужный код до этого момента
    elif message.text == text42:
        userAdd.handle_table_selection(bot, message, table_name)
    elif message.text == text43:
        userDelete.handle_table_selection2(bot, message, table_name)
# Всякий нужный код после этого момента

А вообще, вы скорее всего пользуетесь бездумно ChatGPT, который может выполнить что-то легкое, и то часто с ошибками, не говоря о чем-то более сложном

Comments

[xx_RuBiCoN_xx]

То есть вы сначала выполняете запрос к БД, а потом только просите ввести номер телефона который в эту БД нужно отправить? А лично вы не chatGPT?)

Да и при попытке добавить userDelete. или userAdd. - не находит для этого определения

UPD: Ааа, Вы код вверх ногами исполнили, тогда конечно виноват, зря быканул

[xx_RuBiCoN_xx]

Хотя знаешь, подход гениальный. Типа если хандлер останется ниже по коду, то он типа не активируется чтоли? Хз, протестил - перевернул код. Допёр что нужно import userAdd и userDelete сделать, а не только from userAdd и userDelete, в целом всё работает. Но проблема так и осталась.