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

Question

[Vadimganin]

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

Код:

@bot.message_handler(commands=["put"])
def get_put(message):
    sent_put = bot.send_message(message.chat.id, "Сколько денег(в рублях) ты хочешь положить в копилку?")
    bot.register_next_step_handler(sent_put, put)

def put(message):
        global get_put, balance
        get_put = message.text
        try:
            balance += int(get_put)
            spent_message = "Ты положил " + str(get_put) + "₽" + "\n" + "Теперь в копилке " + str(balance) + "₽"
            bot.send_message(message.chat.id, spent_message, parse_mode="html")
        except ValueError:
            bot.send_message(message.chat.id, "Ошибка! Попробуй написать только число(без букв и символов)")

Answer 1

[Иван Гребнев]

Используя isdigit(), как сказали до этого, пишешь условие, если встречается ошибка, то возвращается на ту же позицию:

balance = 0
@bot.message_handler(commands=["put"])
def get_put(message):
    sent_put = bot.send_message(message.chat.id, "Сколько денег(в рублях) ты хочешь положить в копилку?")
    bot.register_next_step_handler(sent_put, put)

def put(message):
    global get_put
    get_put = message.text
    if (message.text).isdigit() == False:
        error_message = bot.send_message(message.chat.id, "Ошибка! Попробуй написать только число(без букв и символов)")
        bot.register_next_step_handler(error_message, put)
    else:
        balance += int(get_put)
        spent_message = "Ты положил " + str(get_put) + "₽" + "\n" + "Теперь в копилке " + str(balance) + "₽"
        bot.send_message(message.chat.id, spent_message, parse_mode="html")

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

Comments

[Vadimganin]

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

Вот так:
- Сколько денег(в рублях) ты хочешь положить в копилку?
- fff
- Ошибка! Попробуй написать только число(без букв и символов)
- 123
- Ты положил 123₽
Теперь в копилке 123₽

[Иван Гребнев]

Так это и будет таким образом работать.

[Vadimganin]

Иван Гребнев, а, точно. Все работает, спасибо!

Answer 2

[AVKor]

Не так надо делать. Валидация на ввод:

$ python3
Python 3.10.2 (main, Mar  8 2022, 23:56:15) [GCC 10.2.1 20210110] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> '123'.isdigit()
True
>>> '123+'.isdigit()
False
>>> '123a'.isdigit()
False

Comments

[Vadimganin]

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

[AVKor]

Vadimganin, Что-то в таком духе

def check_is_int(get_put):
    return get_put.strip().isdigit()

balance = 0
get_put = input("Введите число ")

while not check_is_int(get_put):
   print('Введено не число. Повторите попытку.')
   get_put = input("Введите число ")

balance += int(get_put)