41 lines
1.6 KiB
Python
41 lines
1.6 KiB
Python
from telegram import Update, InlineKeyboardButton, InlineKeyboardMarkup
|
|
from telegram.ext import CommandHandler, CallbackQueryHandler, ConversationHandler, ContextTypes
|
|
from db import SessionLocal
|
|
from models import Group, Button
|
|
|
|
SELECT_GROUP, MANAGE_BUTTONS = range(2)
|
|
|
|
async def group_buttons_start(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
|
session = SessionLocal()
|
|
groups = session.query(Group).all()
|
|
session.close()
|
|
keyboard = [[InlineKeyboardButton(g.name, callback_data=str(g.id))] for g in groups]
|
|
await update.message.reply_text(
|
|
"Выберите группу для настройки клавиатуры:",
|
|
reply_markup=InlineKeyboardMarkup(keyboard)
|
|
)
|
|
return SELECT_GROUP
|
|
|
|
async def select_group(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
|
query = update.callback_query
|
|
await query.answer()
|
|
group_id = int(query.data)
|
|
context.user_data['group_id'] = group_id
|
|
session = SessionLocal()
|
|
buttons = session.query(Button).filter_by(group_id=group_id).all()
|
|
session.close()
|
|
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)
|
|
return ConversationHandler.END
|
|
|
|
group_buttons_conv = ConversationHandler(
|
|
entry_points=[CommandHandler('group_buttons', group_buttons_start)],
|
|
states={
|
|
SELECT_GROUP: [CallbackQueryHandler(select_group)],
|
|
},
|
|
fallbacks=[]
|
|
)
|