Как упростить код?

Question

[dr4k]

Здравствуйте, подскажите как можно упростить данный код? И где я могу хранить информацию о товаре( к примеру мне нужно вывести марку, модель, цвет, цену)

@bot.message_handler(func=lambda mess: "Продукция1" == mess.text, content_types=['text'])
def handle_text(message):
    keyboard = types.ReplyKeyboardMarkup(row_width=2, resize_keyboard=True, one_time_keyboard=False)
    cid = message.chat.id
    keyboard.row("Товар1")
    keyboard.row("Товар2")
    keyboard.row("Товар3")
    keyboard.row("Товар4")
    keyboard.row("Товар5")
    keyboard.row("Товар6")
    keyboard.row("Товар7")
    keyboard.row("Товар8")
    keyboard.row("Товар9")
    keyboard.row("Товар10")
    keyboard.row("Товар11")
    keyboard.row("Товар12")
    bot.send_message(message.from_user.id, "Выберите товар", reply_markup=keyboard)

@bot.message_handler(func=lambda mess: "Товар1" == mess.text or "Товар2" == mess.text
	or "Товар3" == mess.text or "Товар4" == mess.text or "Товар5" == mess.text
	or "Товар6" == mess.text or "Товар7" == mess.text or "Товар8" == mess.text
	or "Товар9" == mess.text or "Товар10" == mess.text or "Товар11" == mess.text or "Товар12" == mess.text, content_types=['text'])
def handle_text(message):
	key = types.ReplyKeyboardMarkup(row_width=2, resize_keyboard=True, one_time_keyboard=False)
	if message.text == "Товар1":
		bot.send_message(message.chat.id, "", reply_markup=key)
	if message.text == "Товар2":
		bot.send_message(message.chat.id, "", reply_markup=key)
	if message.text == "Товар3":
		bot.send_message(message.chat.id, "", reply_markup=key)
	if message.text == "Товар4":
		bot.send_message(message.chat.id, "", reply_markup=key)
	if message.text == "Товар5":
		bot.send_message(message.chat.id, "", reply_markup=key)
	if message.text == "Товар6":
		bot.send_message(message.chat.id, "", reply_markup=key)
	if message.text == "Товар7":
		bot.send_message(message.chat.id, "", reply_markup=key)
	if message.text == "Товар8":
		bot.send_message(message.chat.id, "", reply_markup=key)
	if message.text == "Товар9":
		bot.send_message(message.chat.id, "", reply_markup=key)
	if message.text == "Товар10":
		bot.send_message(message.chat.id, "", reply_markup=key)
	if message.text == "Товар11":
		bot.send_message(message.chat.id, "", reply_markup=key)
	if message.text == "Товар12":
		bot.send_message(message.chat.id, "", reply_markup=key)

Comments

[Alexander]

О, Господи. Что это ? PHP ?

[spaceatmoon]

Alexander, на PHP явно так не пишут, думаю это Python.

[Alexander]

spaceatmoon, по синтаксису похож на питон, но по написанию кода PHP.

[spaceatmoon]

Alexander, PHP прогеры очень даже нормально пишут :) Вот битриксоиды да, другое дело.

Answer 1

[Neyury]

Упростить код можно как-то так

goods = ['Товар1', 'Товар2', 'Товар3', 'Товар4', 'Товар5', 'Товар6', 'Товар7', 'Товар8', 'Товар9', 'Товар10', 'Товар11', 'Товар12']

@bot.message_handler(func=lambda mess: "Продукция1" == mess.text, content_types=['text'])
def handle_text(message):
    keyboard = types.ReplyKeyboardMarkup(row_width=2, resize_keyboard=True, one_time_keyboard=False)

    for goods_item in goods:
        keyboard.row(goods_item)

    bot.send_message(message.from_user.id, "Выберите товар", reply_markup=keyboard)

@bot.message_handler(func=lambda mess: mess.text in goods, content_types=['text'])
def handle_text(message):
    keyboard = types.ReplyKeyboardMarkup(row_width=2, resize_keyboard=True, one_time_keyboard=False)

    if message.text in goods:
        bot.send_message(message.chat.id, "", reply_markup=keyboard )

Массив goods соответственно можете определять динамически как вам удобно, формировать другие списки и т.д.

Вариантов хранения огромное множество, какой выбрать зависит от способа размещения бота и способа получения первоначальных данных. Это может бытьфайл txt, csv, json, xml, или python объекты, сохраненные через модули shelve или pickle, или sqlite база данных, или полноценная база данных.

UPD:
Функции называются одинаково, надо дать им разные названия

Comments

[dr4k]

Благодарю за помощь, не могли бы вы подсказать( привести пример) как в приведенном вам примере выводить информацию о товаре марку, модель, цвет, цену если данные будут храниться в json или txt файле? Никак не могу с этим разобраться.

[Neyury]

В JSON проще. Как либо сохраняете данные в файл goods.json

И в коде используете так:

data = None
with open('goods.json', 'r', encoding='utf-8') as file: # открываем файл в режиме чтения
    data = json.load(file) # преобразуем данные из файла в формате json в python словарь

UPD:
К примеру в JSON файле у нас такая структура данных:

{
    "goods": [
        {
            "price": 100,
            "color": "red",
            "model": "model1"
        },
        {
            "price": 100,
            "color": "red",
            "model": "model1"
        },
        {
            "price": 100,
            "color": "red",
            "model": "model1"
        }
    ]
}

Доступ к списку товаров:
data['goods']

Атрибут товара с определенным индексом:
data['goods'][index]['price']

Итерация по списку товаров:

for goods_item in data['goods']:
    print(goods_item['price'])

Только если запросы к боту будут частые, стоит задуматься о другом способе хранения данных.

[dr4k]

Neyury, Спасибо я просто не пойму как в данном примере будет проверяться какой товар выбран, на примере моего json

@bot.message_handler(func=lambda mess: mess.text in goods, content_types=['text'])
def handle_text(message):
    keyboard = types.ReplyKeyboardMarkup(row_width=2, resize_keyboard=True, one_time_keyboard=False)

    if message.text in goods:
        bot.send_message(message.chat.id, "", reply_markup=keyboard )

[
{"Продукция1":{"name_good": "Товар1": {"price": "100", "color": "black", "name": "namegoods"},
  "Товар2": {"price": "100", "color": "black", "name": "namegoods"},
  "Товар3": {"price": "100", "color": "black", "name": "namegoods"},
  "Товар4": {"price": "100", "color": "black", "name": "namegoods"},
  "Товар5": {"price": "100", "color": "black", "name": "namegoods"},
  "Товар6": {"price": "100", "color": "black", "name": "namegoods"},
  "Товар7": {"price": "100", "color": "black", "name": "namegoods"}}},
]

[Neyury]

По хорошему JSON должен являться словарем, в котором уже могут быть массивы и другие данные, т.е преобразуйте ваши данные к такому виду:

{
    "data": [
        # ваш массив
    ]
}

Либо такому (рекомендую)

{
    "продукция1": {
        "товар1": {},
        "товар2": {}
    },
    "продукция2": {
        "товар3": {},
        "товар4": {}
    }
}

Я для примеру буду использовать второй вариант. Если мы знаем какая у нас продукция, то получить доступ к конкретному товару мы можем так. (Ниже пример тела функции)

goods_item = data['продукция1'].get(message.text)

if goods_item:
    bot.send_message(message.chat.id, goods_item['price'], reply_markup=keyboard)
else:
    bot.send_message(message.chat.id, "Товар отсутствует", reply_markup=keyboard)

Кстати, у вас две функции называются одинаково, это не правильно.

[Neyury]

Чтобы немного разобраться в питоне и в коде, который я привожу, рекомендую прочитать как минимум книгу "Укус питона": первая ссылка в гугле

В книге всего страниц 150, прочтете быстро. Думаю этого точно хватит, чтобы дальше разобраться с задачей.

[dr4k]

Огромное спасибо за помощь, буду разбираться. В программировании не силен, но задача стоит надо как то выкручиваться)

[Neyury]

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