Как сделать input на python (telegram bot)?

Question

[Ярик Ярик]

import telebot
import config
import random
 
from telebot import types
 
bot = telebot.TeleBot(config.TOKEN)
 
@bot.message_handler(commands=['start'])
def welcome(message):

    markup = types.ReplyKeyboardMarkup(resize_keyboard=True)
    item1 = types.KeyboardButton("Cлучайное число(от 1 до 10)")
    item2 = types.KeyboardButton("Геометрическая Прогрессия")
 
    markup.add(item1, item2)
 
    bot.send_message(message.chat.id, "Добро пожаловать, {0.first_name}!\nЯ - <b>{1.first_name}</b>\nБот, который способен на все.".format(message.from_user, bot.get_me()),
        parse_mode='html', reply_markup=markup)
 
@bot.message_handler(content_types=['text'])
def main(message):
    if message.chat.type == 'private':
        if message.text == 'Cлучайное число(от 1 до 10)':
            bot.send_message(message.chat.id, str(random.randint(0,10)))
        elif message.text == 'Геометрическая Прогрессия':
            b1 = int(input('Первый член геометрической прогрессии(b1) = '))
            q = int(input('Знаменатель прогрессии(q) = '))
            n = int(input('Номер последнего члена(n) = '))
            bn = b1 * q**(n - 1)
            sn = (b1*(q**n - 1))/(q-1)
            bot.send_message(message.chat.id, str('Последний член геометрической прогрессии(bn) = '+str(bn)+'\nСумма геометрической прогрессии(SUMn) = '+str(sn)))         
        else:
            bot.send_message(message.chat.id, 'Я не знаю что ответить :(')

if name == '__main__':
    bot.skip_pending = True
    bot.polling(none_stop=True)

Делаю калькулятор геометрической прогрессии для практики.
b1 = int(input('Первый член геометрической прогрессии(b1) = '))
q = int(input('Знаменатель прогрессии(q) = '))
n = int(input('Номер последнего члена(n) = '))
Не знаю как заменить вот эти строчки... В консоли все пишет естественно, а как сделать, чтобы в телеграмме бот запрашивал, не знаю. Любой подсказке буду очень рад

Comments

[Александр Катунин]

input здесь вообще никак....
машина состояний надо)

[Ярик Ярик]

Александр Катунин, register_next_step_handler - через эту штучку :??? Я пытался чет сам сделать, но оно в конце ошибки выдало..

Answer 1

[Владислав Ликов]

import telebot
import config
import random
 
from telebot import types
 
bot = telebot.TeleBot(config.TOKEN)
 
@bot.message_handler(commands=['start'])
def welcome(message):
    markup = types.ReplyKeyboardMarkup(resize_keyboard=True)
    item1 = types.KeyboardButton("Cлучайное число(от 1 до 10)")
    item2 = types.KeyboardButton("Геометрическая Прогрессия")
    markup.add(item1, item2)
    msg = bot.send_message(message.chat.id, "Добро пожаловать, {0.first_name}!\nЯ - <b>{1.first_name}</b>\nБот, который способен на все.".format(message.from_user, bot.get_me()),
        parse_mode='html', reply_markup=markup)
    bot.register_next_step_handler(msg, main)
 
@bot.message_handler(content_types=['text'])
def main(message):
    if message.text == 'Cлучайное число(от 1 до 10)':
     bot.send_message(message.chat.id, str(random.randint(0,10)))
    elif message.text == 'Геометрическая Прогрессия':
     msg = bot.send_message(message.chat.id, "Введите первый член геометрической прогрессии, затем знаменатель прогрессии и номер последнего члена (писать через пробел)")
     bot.register_next_step_handler(msg, calc)   
    else:
     bot.send_message(message.chat.id, 'Я не знаю что ответить :(')

def calc(message):
    a, b, c = message.text.split()
    b1 = int(a)
    q = int(b)
    n = int(c)
    bn = b1 * q**(n - 1)
    sn = (b1*(q**n - 1)) // (q-1)
    bot.send_message(message.chat.id, f'Последний член геометрической прогрессии(bn) = {bn}\nСумма геометрической прогрессии(SUMn) = {sn}')



bot.polling(none_stop=True)

Вот код не идеал, но работает

Comments

[Ярик Ярик]

Вау, кое-что нужно было подправить, но заработало ! Спасибо огромное