Как оптимизировать моего телеграм-бота, написанного на Python?

Question

[belugasinister]

Я написал довольно простого бота, который предоставляет информацию или дополнительный выбор в зависимости от запроса пользователя. Он работает хорошо и выполняет свою работу, однако мне было интересно, как я могу оптимизировать свой код и лучше использовать телебота? В настоящее время основное действие происходит в callback query handler, и что-то мне подсказывает, что мне не хватает эффективности (например, когда я попытаюсь создать больше разветвлений выбора юзера). Ниже приведен код, упрощенный для демонстрационных целей:

import telebot
from telebot import types

token = bottoken

bot = telebot.TeleBot(token)

@bot.message_handler(commands=['start'])
def welcome(message):

	# here i give the main choice that is always visible to the user

	markup = types.ReplyKeyboardMarkup(resize_keyboard=False)
	item1 = types.KeyboardButton("Button1")
	item2 = types.KeyboardButton("Button2")

	markup.add(item1, item2)

	bot.send_message(message.chat.id, "Welcome message".format(message.from_user, bot.get_me()),
		parse_mode='html', reply_markup=markup)


@bot.message_handler(content_types=['text'])
def main_choice(message):

	# here i develop each of the main choices and redirect to the callback handler

	if message.text == 'Button1':

		# Forward to callback handler

			markup = types.InlineKeyboardMarkup()
			item1 = types.InlineKeyboardButton("Yes", callback_data = 'demo_yes')
			item2 = types.InlineKeyboardButton("No", callback_data = 'demo_no')

			markup.add(item1, item2)

			bot.send_message(message.chat.id, 'Some text', reply_markup=markup)


	elif message.text == 'Button2':

		# some logic

	else :
		bot.send_message(message.chat.id, 'Sorry, i dont understand. You can use /help to check how i work')


@bot.callback_query_handler(func=lambda call: True)
def tempo(call):

	# here a handle the queries launched both by main buttons and the inline keyboard 


	if call.data == "demo_yes":

		keyboard = types.InlineKeyboardMarkup()
		item1 = types.InlineKeyboardButton("Text1", callback_data = 'call1')
		item2 = types.InlineKeyboardButton("Text2", callback_data = 'call2')
		keyboard.add(item1,item2)

		bot.send_message(call.message.chat.id, 'message', reply_markup = keyboard)

	elif call.data == "demo_no":

		kkeyboard = types.InlineKeyboardMarkup()
		item1 = types.InlineKeyboardButton("Text3", callback_data = 'call3')
		item2 = types.InlineKeyboardButton("Text4", callback_data = 'call4')
		keyboard.add(item1,item2)

		bot.send_message(call.message.chat.id, 'message', reply_markup = keyboard)

	if call.data == "call1":

		keyboard = types.InlineKeyboardMarkup()
		item1 = types.InlineKeyboardButton("Text5", callback_data = 'another_call1')
		item2 = types.InlineKeyboardButton("Text6", callback_data = 'another_call2')
		keyboard.add(item1,item2)

		bot.send_message(call.message.chat.id, 'message', reply_markup = keyboard)

	elif call.data == "call2":

		kkeyboard = types.InlineKeyboardMarkup()
		item1 = types.InlineKeyboardButton("Text7", callback_data = 'another_call3')
		item2 = types.InlineKeyboardButton("Text8", callback_data = 'another_call4')
		keyboard.add(item1,item2)

		bot.send_message(call.message.chat.id, 'message', reply_markup = keyboard)

Comments

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

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

Answer 1

[Alexa2007]

Ну на первый взгляд можно просто раскидать минимум на три файла:
1.py

import telebot
from telebot import types
token = bottoken
bot = telebot.TeleBot(token)
bot.polling()

2.py

#Все хендлеры

@bot.message_handler(commands=['start'])
def welcome(message):

2.py

#Все коллбеки без elif а нормально читаемые

@bot.callback_query_handler(func=lambda call: call.data == '7')
def seven(call):
    bot.send_message(call.message.chat.id,text = 'seven')

@bot.callback_query_handler(func=lambda call: call.data == '1')
def one(call):
    bot.send_message(call.message.chat.id,text = 'one')

Comments

[belugasinister]

Спасибо! Как связать потом эти файлы? Т.е. как прописать обращение к этим файлам в зависимости от логики?

[Alexa2007]

Ну вот.. наваял по быстрому:
!!!!!!!!Запускать надо handlers.py !!!!!!
Если запустишь app.py ничего не произойдет

app.py

# -*- coding: utf-8 -*-
import telebot
API_TOKEN = '***'
bot = telebot.TeleBot(API_TOKEN, threaded=False)
import handlers, callbacks
bot.polling()

callbacks.py

from app import bot as bot


@bot.callback_query_handler(func=lambda call: call.data == 'call1')
def seven(call):
    bot.send_message(call.message.chat.id,text = 'one')

@bot.callback_query_handler(func=lambda call: call.data == 'call2')
def seven(call):
    bot.send_message(call.message.chat.id,text = 'two')

@bot.callback_query_handler(func=lambda call: call.data == 'call3')
def seven(call):
    bot.send_message(call.message.chat.id,text = 'three')

@bot.callback_query_handler(func=lambda call: call.data == 'call4')
def seven(call):
    bot.send_message(call.message.chat.id,text = 'four')

@bot.callback_query_handler(func=lambda call: call.data == 'call5')
def seven(call):
    bot.send_message(call.message.chat.id,text = 'five')

@bot.callback_query_handler(func=lambda call: call.data == 'call6')
def seven(call):
    bot.send_message(call.message.chat.id,text = 'six')

@bot.callback_query_handler(func=lambda call: call.data == 'call7')
def seven(call):
    bot.send_message(call.message.chat.id,text = 'seven')

handlers.py

from app import bot as bot
from utils import start_btns, day_btns # Или from utils import *

@bot.message_handler(commands=['start'])
@bot.message_handler(func=lambda message: message.text == "start")
def wellcome(message):
    if message.chat.type == 'private':
        bot.send_message(message.chat.id,'Hello',parse_mode='markdown',reply_markup=start_btns())


@bot.message_handler(func=lambda message: message.text == "help")
@bot.message_handler(func=lambda message: message.text == "Кнопка три")
def do_some(message):
    if message.chat.type == 'private':
        bot.send_message(message.chat.id,'Days of week',reply_markup=day_btns())

utils.py

from telebot import types




def start_btns():
    buttons = types.ReplyKeyboardMarkup(resize_keyboard=True, row_width=2)
    buttons.add(*['start','help','Кнопка три'])
    return buttons

def day_btns():
	days = types.InlineKeyboardMarkup(row_width=7)
	days.add(*[types.InlineKeyboardButton(text='Day '+str(i),callback_data='call'+str(i)) for i in range(1,8)])
	return days

[belugasinister]

Alexa2007, большое спасибо! Пойду тестить