40 lines
1.6 KiB
Python
40 lines
1.6 KiB
Python
import logging
|
||
from telebot import types, TeleBot
|
||
from database.models import Students, Admins
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
|
||
def register(bot: TeleBot):
|
||
@bot.message_handler(func=lambda message: message.text == "Show Students")
|
||
def get_students(message):
|
||
if not Admins.is_authenticated(message.from_user.id):
|
||
bot.send_message(
|
||
message.chat.id, "⛔ Unauthorized access!\nPlease /login first.")
|
||
return
|
||
|
||
try:
|
||
data = Students.getAllStudents()
|
||
if data:
|
||
markup = types.InlineKeyboardMarkup(row_width=2)
|
||
|
||
for row in data:
|
||
btn = types.InlineKeyboardButton(
|
||
f"@{row[1]} | id#{row[0]}", callback_data=f"student_{row[0]}")
|
||
markup.add(btn)
|
||
|
||
# add button for creating a new student
|
||
markup.add(types.InlineKeyboardButton(
|
||
"➕ Create New Student 🔒", callback_data="create_student"))
|
||
|
||
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 Student 🔒", callback_data="create_student"))
|
||
bot.reply_to(message, "No data found.", reply_markup=markup)
|
||
except Exception as e:
|
||
logger.error(f"Error in get_students handler: {e}")
|
||
bot.reply_to(message, "Sorry, an error occurred.")
|