52 lines
2.1 KiB
Python
52 lines
2.1 KiB
Python
from telegram import Update, InlineKeyboardButton, InlineKeyboardMarkup
|
|
from telegram.ext import CommandHandler, CallbackQueryHandler, ConversationHandler, ContextTypes
|
|
from db import AsyncSessionLocal
|
|
from models import Group, Button
|
|
|
|
SELECT_GROUP, MANAGE_BUTTONS = range(2)
|
|
|
|
async def group_buttons_start(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
|
session = AsyncSessionLocal()
|
|
try:
|
|
groups_result = await session.execute(Group.__table__.select())
|
|
groups = groups_result.scalars().all()
|
|
keyboard = [[InlineKeyboardButton(f'{getattr(g, "name", str(g.name))}', callback_data=str(getattr(g, "id", str(g.id))))] for g in groups]
|
|
if update.message:
|
|
await update.message.reply_text(
|
|
"Выберите группу для настройки клавиатуры:",
|
|
reply_markup=InlineKeyboardMarkup(keyboard)
|
|
)
|
|
return SELECT_GROUP
|
|
return ConversationHandler.END
|
|
finally:
|
|
await session.close()
|
|
|
|
async def select_group(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
|
query = update.callback_query
|
|
if query and query.data:
|
|
await query.answer()
|
|
group_id = int(query.data)
|
|
if context.user_data is None:
|
|
context.user_data = {}
|
|
context.user_data['group_id'] = group_id
|
|
session = AsyncSessionLocal()
|
|
try:
|
|
buttons_result = await session.execute(Button.__table__.select().where(Button.group_id == group_id))
|
|
buttons = buttons_result.scalars().all()
|
|
text = "Кнопки этой группы:\n"
|
|
for b in buttons:
|
|
text += f"- {b.name}: {b.url}\n"
|
|
text += "\nДобавить новую кнопку: /add_button\nУдалить: /del_button <название>"
|
|
await query.edit_message_text(text)
|
|
finally:
|
|
await session.close()
|
|
return ConversationHandler.END
|
|
|
|
group_buttons_conv = ConversationHandler(
|
|
entry_points=[CommandHandler('group_buttons', group_buttons_start)],
|
|
states={
|
|
SELECT_GROUP: [CallbackQueryHandler(select_group)],
|
|
},
|
|
fallbacks=[]
|
|
)
|