36 lines
1.3 KiB
Python
36 lines
1.3 KiB
Python
import logging
|
||
from telebot import types
|
||
from database.models import Categories
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
|
||
def register(bot):
|
||
@bot.message_handler(func=lambda message: message.text == "Show Categories")
|
||
def get_categories(message):
|
||
try:
|
||
data = Categories.getAllCategories()
|
||
if data:
|
||
markup = types.InlineKeyboardMarkup(row_width=2)
|
||
|
||
for row in data:
|
||
btn = types.InlineKeyboardButton(
|
||
f"{row[1]} | id#{row[0]}", callback_data=f"category_{row[0]}")
|
||
markup.add(btn)
|
||
|
||
# add button for creating a new category
|
||
markup.add(types.InlineKeyboardButton(
|
||
"➕ Create New Category", callback_data="create_category"))
|
||
|
||
bot.send_message(
|
||
message.chat.id, "Here is the data:", reply_markup=markup)
|
||
else:
|
||
markup = types.InlineKeyboardMarkup(row_width=1)
|
||
markup.add(types.InlineKeyboardButton(
|
||
"➕ Create New Category", callback_data="create_category"))
|
||
|
||
bot.reply_to(message, "No data found.", reply_markup=markup)
|
||
except Exception as e:
|
||
logger.error(f"Error in 'Show Categories' handler: {e}")
|
||
bot.reply_to(message, "Sorry, an error occurred.")
|