Функция для телеграм бота на python?

Question

[tokyodead]

Добрый день, подскажите, пишу бота для телеграм, я новичок в этом и не могу понять, как сделать так что если это справка#1 отправлять ему одну справку, а если это справка#2 отправлять ему другую справку?
Вот код

bot = telebot.TeleBot(TOKEN)

@bot.message_handler(commands=['start'])
def start(message):
    markup = types.ReplyKeyboardMarkup(resize_keyboard = True)
    item1 = types.KeyboardButton('Заказать справку')
    markup.add(item1)
    bot.send_message(message.chat.id, 'Вас приветствует бот myboy', reply_markup=markup)

@bot.message_handler(content_types=['text'])
def bot_message(message):
    if message.chat.type == 'private':
        if message.text == 'Заказать справку':
            markup = types.ReplyKeyboardMarkup(resize_keyboard=True)
            item1 = types.KeyboardButton('Справка#1')
            back = types.KeyboardButton('Назад')
            markup.add(item1, back)
            bot.send_message(message.chat.id, 'Какая справка Вам нужна?', reply_markup=markup)

        elif message.text == 'Справка#1':
            bot.send_message(message.chat.id, 'Введите ИИН')

        elif message.text.isdigit and len(message.text) == 12:
            with con:
                cur = con.cursor()
                sql = "SELECT `fullname` FROM `bot_db` WHERE `iin`=%s and status=1 and user_type=1 limit 1 "
                cur.execute(sql, (message.text,))
                result = cur.fetchone()
                name = result
                age = '1997-01-08'
                page_name = str(name) + '.pdf'
                get_pdf(name,age, page_name)
                bot.send_message(message.chat.id, result)
                r = open('C:\\Users\\Alkon PC\\PycharmProjects\\mainpy\\{}'.format(page_name), 'rb')
                bot.send_document(message.chat.id, r)

Answer 1

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

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

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

Comments

[tokyodead]

Спасибо Вам большое, добрый человек)
Да, я сейчас знакомлюсь с register_next_step_handler, попробую написать все по новой.

[tokyodead]

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

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

tokyodead, можно сделать структуру каталогов в проекте /<инн>/<вид справки>/<дата, если нужна>/<имя файла>.
Потом просто по нужному пути обращаться.

[tokyodead]

Михаил, Я на правильном пути?) Составил для себя какую то мини структуру чтобы понять как писать дальше.

bot = telebot.TeleBot(TOKEN)

@bot.message_handler(commands=['start'])
def start(message):
    markup = types.ReplyKeyboardMarkup(resize_keyboard = True)
    item1 = types.KeyboardButton('Заказать справку')
    markup.add(item1)
    bot.send_message(message.chat.id, 'Вас приветствует бот myboy', reply_markup=markup)

@bot.message_handler(content_types=['text'])
def bot_message(message):
    if message.text == 'Заказать справку':
        markup = types.ReplyKeyboardMarkup(resize_keyboard=True)
        item1 = types.KeyboardButton('Справка#1')
        item2 = types.KeyboardButton('Справка#2')
        item3 = types.KeyboardButton('Справка#3')
        back = types.KeyboardButton('Назад')
        markup.add(item1, item2, item3, back)
        bot.send_message(message.chat.id, 'Какая справка Вам нужна?', reply_markup=markup)
    elif message.text == 'Справка#1':
        bot.send_message(message.chat.id, 'Введите ИИН')
        bot.register_next_step_handler(message, process_first_step)
    elif message.text == 'Справка#2':
        bot.send_message(message.chat.id, 'Введите ИИН')
        bot.register_next_step_handler(message, process_second_step)


def process_first_step(message):
    bot.send_message(message.chat.id, 'First Process')

def process_second_step(message):
    bot.send_message(message.chat.id, 'Second Process')

bot.polling(none_stop=True)

[tokyodead]

Михаил, Пока что сохраняю файл с помощью random
Возникла проблема, все работает, отправляю ИИН, и мне приходит справка,
но если я еще раз ввожу ИИН, соединение с БД закрывает, выходит ошибка и бот останавливается, можете подсказать как это решить?
Вот код

bot = telebot.TeleBot(TOKEN)

@bot.message_handler(commands=['start'])
def start(message):
    markup = types.ReplyKeyboardMarkup(resize_keyboard = True)
    item1 = types.KeyboardButton('Заказать справку')
    markup.add(item1)
    bot.send_message(message.chat.id, 'Вас приветствует бот myboy', reply_markup=markup)

@bot.message_handler(content_types=['text'])
def bot_message(message):
    if message.text == 'Заказать справку':
        markup = types.ReplyKeyboardMarkup(resize_keyboard=True)
        item1 = types.KeyboardButton('Справка#1')
        item2 = types.KeyboardButton('Справка#2')
        item3 = types.KeyboardButton('Справка#3')
        back = types.KeyboardButton('Назад')
        markup.add(item1, item2, item3, back)
        bot.send_message(message.chat.id, 'Какая справка Вам нужна?', reply_markup=markup)
    elif message.text == 'Справка#1':
        bot.send_message(message.chat.id, 'Введите ИИН')
        bot.register_next_step_handler(message, process_first_step)
    elif message.text == 'Справка#2':
        bot.send_message(message.chat.id, 'Введите ИИН')
        bot.register_next_step_handler(message, process_second_step)


def process_first_step(message):
     if len(message.text) == 12:
        with con:
            cur = con.cursor()
            sql = "SELECT `fullname` FROM `bot_db` WHERE `iin`=%s and status=1 and user_type=1 limit 1 "
            cur.execute(sql, (message.text,))
            result = cur.fetchone()
        rand = random.randint(30,60)
        pdf = FPDF()
        pdf.add_page()
        pdf.set_font('Arial', 'B', 16)
        pdf.cell(40, 10, 'Hello World!')
        pdf.output('{}.pdf'.format(rand), 'F')
        r = open('C:\\Users\\Alkon PC\\PycharmProjects\\main_bot\\{}.pdf'.format(rand), 'rb')
        bot.send_message(message.chat.id, 'Name: {}'.format(result))
        bot.send_document(message.chat.id, r)


def process_second_step(message):
    if len(message.text) == 12:
        with con:
            cur = con.cursor()
            sql = "SELECT `fullname` FROM `bot_db` WHERE `iin`=%s and status=1 and user_type=1 limit 1 "
            cur.execute(sql, (message.text,))
            result = cur.fetchone()
        rand = random.randint(30, 60)
        pdf = FPDF()
        pdf.add_page()
        pdf.set_font('Arial', 'B', 16)
        pdf.cell(40, 10, 'Hello World!')
        pdf.output('{}.pdf'.format(rand), 'F')
        r = open('C:\\Users\\Alkon PC\\PycharmProjects\\main_bot\\{}.pdf'.format(rand), 'rb')
        bot.send_message(message.chat.id, 'Name: {}'.format(result))
        bot.send_document(message.chat.id, r)
        

bot.polling(none_stop=True)

С БД работаю с помощь библиотеки pymysql

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

tokyodead, В принципе да. Чтобы не плодить различные функции process_first_step, process_second_step и пр, можно сделать одну функцию, которая будет принимать аргумент, показывающий какой кнопкой эта функция была вызвана.
register_next_step_handler первым параметром принимает сообщение, вторым - объект функции, которую стоит выполнить, а дальше можно передавать параметры, которые вызываемая функция будет получать при вызове (помимо message):

...
    elif message.text == 'Справка#1':
        bot.send_message(message.chat.id, 'Введите ИИН')
        bot.register_next_step_handler(message, process, 'First Process')
    elif message.text == 'Справка#2':
        bot.send_message(message.chat.id, 'Введите ИИН')
        bot.register_next_step_handler(message,  process, 'SecondProcess')

def process(message, process_number):
    bot.send_message(message.chat.id, process_number)

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

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