Как устранить ошибку с генерацией случайных чисел в боте?

Question

[capybara123]

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

spoiler

import telebot
from telebot import types
from random import randint
bot = telebot.TeleBot('токен')

@bot.message_handler(commands=['start'])
def start(message):
    markup = types.ReplyKeyboardMarkup(resize_keyboard = True)
    game = types.KeyboardButton('game')
    markup.add(game)
    bot.send_message(message.chat.id, ', {0.first_name}. Клавиатура снизу'.format(message.from_user), reply_markup = markup) 

random_number = randint(1,5)

@bot.message_handler(content_types = ['text'])
def bot_message(message): 
    if message.chat.type == 'private':
        if message.text == 'game':
            if random_number == 1:
                game = open('game/1.jpg', 'rb')
                bot.send_photo(message.chat.id, game)
                markup = types.ReplyKeyboardMarkup(resize_keyboard = True)
                variant1 = types.KeyboardButton(f'{random_number + 1} кота')
                variant2 = types.KeyboardButton(f'{random_number + 2} кота')
                variant3 = types.KeyboardButton('1 кот')
                markup.add(variant1, variant2, variant3)
                bot.send_message(message.chat.id, 'Выбери правильный вариант ответа', reply_markup = markup)
            elif random_number == 2:
                game = open('game/2.jpg', 'rb')
                bot.send_photo(message.chat.id, game)
                markup = types.ReplyKeyboardMarkup(resize_keyboard = True)
                variant1 = types.KeyboardButton(f'{random_number + 2} кота')
                variant2 = types.KeyboardButton(f'{random_number - 1} кот')
                variant3 = types.KeyboardButton('2 кота')
                markup.add(variant1, variant2, variant3)
                bot.send_message(message.chat.id, 'Выбери правильный вариант ответа', reply_markup = markup)
            elif random_number == 3:
                game = open('game/3.jpg', 'rb')
                bot.send_photo(message.chat.id, game)
                markup = types.ReplyKeyboardMarkup(resize_keyboard = True)
                variant1 = types.KeyboardButton(f'{random_number + 2} котов')
                variant2 = types.KeyboardButton('3 кота')
                variant3 = types.KeyboardButton(f'{random_number - 1} кота')
                markup.add(variant1, variant2, variant3)
                bot.send_message(message.chat.id, 'Выбери правильный вариант ответа', reply_markup = markup)
            elif random_number == 4:
                game = open('game/4.jpg', 'rb')
                bot.send_photo(message.chat.id, game)
                markup = types.ReplyKeyboardMarkup(resize_keyboard = True)
                variant1 = types.KeyboardButton('4 кота')
                variant2 = types.KeyboardButton(f'{random_number - 2} кота')
                variant3 = types.KeyboardButton(f'{random_number + 1} котов')
                markup.add(variant1, variant2, variant3)
                bot.send_message(message.chat.id, 'Выбери правильный вариант ответа', reply_markup = markup)
            elif random_number == 5:
                game = open('game/5.jpg', 'rb')
                bot.send_photo(message.chat.id, game)
                markup = types.ReplyKeyboardMarkup(resize_keyboard = True)
                variant1 = types.KeyboardButton(f'{random_number - 2} кота')
                variant2 = types.KeyboardButton(f'{random_number - 4} кот')
                variant3 = types.KeyboardButton('5 котов')
                markup.add(variant1, variant2, variant3)
                bot.send_message(message.chat.id, 'Выбери правильный вариант ответа', reply_markup = markup)

        # Проверка на правильные ответы
        elif message.text == '1 кот' and random_number == 1:
            markup = types.ReplyKeyboardMarkup(resize_keyboard = True)
            chillbutton0 = types.KeyboardButton('game')
            markup.add(chillbutton0)
            bot.send_message(message.chat.id, 'Правильно!', reply_markup = markup)
        elif message.text == '2 кота' and random_number == 2:
            markup = types.ReplyKeyboardMarkup(resize_keyboard = True)
            chillbutton0 = types.KeyboardButton('game')
            markup.add(chillbutton0)
            bot.send_message(message.chat.id, 'Правильно!', reply_markup = markup)
        elif message.text == '3 кота' and random_number == 3:
            markup = types.ReplyKeyboardMarkup(resize_keyboard = True)
            chillbutton0 = types.KeyboardButton('game')
            markup.add(chillbutton0)
            bot.send_message(message.chat.id, 'Правильно!', reply_markup = markup)
        elif message.text == '4 кота' and random_number == 4:
            markup = types.ReplyKeyboardMarkup(resize_keyboard = True)
            chillbutton0 = types.KeyboardButton('game')
            markup.add(chillbutton0)
            bot.send_message(message.chat.id, 'Правильно!', reply_markup = markup)
        elif message.text == '5 котов' and random_number == 5:
            markup = types.ReplyKeyboardMarkup(resize_keyboard = True)
            chillbutton0 = types.KeyboardButton('game')
            markup.add(chillbutton0)
            bot.send_message(message.chat.id, 'Правильно!', reply_markup = markup)

        # Проверка на НЕ правильные ответы 
        elif message.text == '1 кот' and random_number != 1:
            markup = types.ReplyKeyboardMarkup(resize_keyboard = True)
            chillbutton0 = types.KeyboardButton('game')
            markup.add(chillbutton0)
            bot.send_message(message.chat.id, 'Прости, ты не угадал :(', reply_markup = markup)
        elif message.text == '2 кота' and random_number != 2:
            markup = types.ReplyKeyboardMarkup(resize_keyboard = True)
            chillbutton0 = types.KeyboardButton('game')
            markup.add(chillbutton0)
            bot.send_message(message.chat.id, 'Прости, ты не угадал :(', reply_markup = markup)
        elif message.text == '3 кота' and random_number != 3:
            markup = types.ReplyKeyboardMarkup(resize_keyboard = True)
            chillbutton0 = types.KeyboardButton('game')
            markup.add(chillbutton0)
            bot.send_message(message.chat.id, 'Прости, ты не угадал :(', reply_markup = markup)
        elif message.text == '4 кота' and random_number != 4:
            markup = types.ReplyKeyboardMarkup(resize_keyboard = True)
            chillbutton0 = types.KeyboardButton('game')
            markup.add(chillbutton0)
            bot.send_message(message.chat.id, 'Прости, ты не угадал :(', reply_markup = markup)
        elif message.text == '5 котов' and random_number != 5:
            markup = types.ReplyKeyboardMarkup(resize_keyboard = True)
            chillbutton0 = types.KeyboardButton('game')
            markup.add(chillbutton0)
            bot.send_message(message.chat.id, 'Прости, ты не угадал :(', reply_markup = markup)
bot.polling()

Comments

[Модератор]

Когда вы регистрировались, вы обещали выполнять Регламент. Ваше сообщение нарушило сразу несколько параграфов Регламента:
3.1, 3.4, 3.6.

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

Будьте благоразумны. Еще раз ознакомьтесь с Регламентом и соблюдайте его.

[Модератор]

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

Answer 1

[Elvis]

Потому что нужно random_number = randint(1,5) перенести в функцию начала новой игры. У вас сейчас генерация, как вы правильно заметили, при старте скрипта, в основном теле программы.

Comments

[capybara123]

Но если я перенесу переменную под if message.text == 'game: то там будет возникать ошибка, вроде будет ругаться на то, что типо "в этом теле нет этой переменной"

[shurshur]

capybara123, можно создать эту переменную вне всех функций, а в нужных функциях объявить через global и заполнять случайным значением именно там. Использование global общепринято считается плохой практикой в python, но работать будет.

Однако при этом возникнет другая проблема: эта переменная будет общей для всех пользователей бота. Если два игрока играют одновременно, то будет конфликт. Но я рекомендую заняться этой проблемой на следующей стадии. А сначала навести в этом боте порядок и переписать его хоть как-то по-человечески.

Дело в том, что в нём куча повторяющегося кода. Так делать крайне плохо. Тем более что улучшения прям напрашиваются.

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

# варианты выбора для котов
cat_choices = {
    1: (2, 3, 1),
    2: (4, 1, 2),
    ...
}

# функция, которая склоняет котов: по их количеству
# возвращает человекочитабельную строку
def get_cats_str(count:int):
    if count == 1: return "1 кот"
    elif count < 5: return f"{count} кота"
    else: return f"{count} котов"

После этого код станет заметно короче и поубавится дублирование. Станет проще делать в нём доработки и меньше риск где-то ошибиться при этом.

if message.text == 'game':
    random_number = randint(1,5)
    # конструкция with используется для того, чтобы файл
    # автоматически был закрыт при выходе из блока with
    with open(f"game/{random_number}.jpg", "rb") as cat_image_file:
        bot.send_photo(message.chat.id, cat_image_file)
    markup = types.ReplyKeyboardMarkup(resize_keyboard = True)
    cats = cat_choices[random_number]:
    variant1 = types.KeyboardButton(get_cats_str(cats[0])
    variant2 = types.KeyboardButton(get_cats_str(cats[1])
    variant3 = types.KeyboardButton(get_cats_str(cats[2])
    markup.add(variant1, variant2, variant3)
    bot.send_message(message.chat.id, 'Выбери правильный вариант ответа', reply_markup = markup)
else:
    # берём первое число перед первым пробелом, для простоты не проверяем, что там дальше "коты"
    cats_count = message.text.split(" ")[0]
    if cats_count.isdigit():
        cats_count = int(cats_count)
        if cats_count == random_number:
            # это верный ответ
        else:
            # это неверный ответ
     else: # не число:
         bot.send_message(message.chat.id, 'Так сколько котов-то?")

Также советую начать выбирать более осмысленные названия переменным. Например, назвать файл с картинкой переменной game это мягко говоря странно. Чем осмысленнее названия переменных, тем понятнее код для восприятия. В том числе и самому себе (особенно если вернуться к нему через некоторое время позже). Кстати, random_number тоже плохое название, лучше что-нибудь про котов или игру, например, selected_game_cats (выбранные для игры коты).

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

Словарь с котами можно держать не в исходном коде, а в отдельном файле (json или yaml). Это тоже гораздо удобнее, заодно можно изучить работу с этими форматами.

А затем можно заняться той самой проблемой, что несколько игроков одновременно играть не могут. Для этого надо хранить не просто общий random_number, а отдельный random_number для каждого пользователя. Для этого заведём словарь, в котором ключом будет chat.id:

user_random_number = {}

...
if message.text == 'game':
    user_random_number[message.chat.id] = randint(1,5)

Но ещё лучше будет освоить FSM (Finite State Machine - конечный автомат состояний). В telebot есть довольно простая реализация FSM на основе вызова bot.register_next_step_handler. В данном случае можно после вывода задачи выставлять следующий обработчик ("next step handler") в другую функцию, которая проверяет ответ, а в ней после ответа сбрасывать обработчик на используемый по умолчанию. При вызове register_next_step_handler можно также передавать дополнительные аргументы, и вот можно передать тот самый random_number. Заодно мы окончательно избавимся от глобальной переменной.

Даже на таком несложном боте можно очень многому научиться. Полезные темы для чтения: словари (dict), работа с файлами и открытие файла через контекстный менеджер с помощью with, глобальные переменные, работа с файлами в формате json, использование FSM (register_next_step_handler) для создания многошаговых взаимодействий пользователя с ботом.

[RimMirK]

shurshur, достойный ответ!