TW-Requests/tgbot_app/kb/reply_kb.py
2024-07-15 10:15:51 +03:00

37 lines
1.2 KiB
Python

from aiogram.types import KeyboardButton
from aiogram.utils.keyboard import ReplyKeyboardBuilder
def get_kb(
*buttons: str,
placeholder: str | None = None,
request_contact: int | None = None,
request_location: int | None = None,
sizes: tuple[int] = (2,)
):
keyboard = ReplyKeyboardBuilder()
for idx, text in enumerate(buttons):
if request_contact and request_contact == idx + 1:
keyboard.add(KeyboardButton(text=text, request_contact=True))
elif request_location and request_location == idx + 1:
keyboard.add(KeyboardButton(text=text, request_location=True))
else:
keyboard.add(KeyboardButton(text=text))
return keyboard.adjust(*sizes).as_markup(
resize_keyboard=True, input_field_placeholder=placeholder
)
START_KB = get_kb(
'Зарегистрироваться.',
'Добавить комментарий.',
'Посмотреть всех пользователей.',
'Посмотреть все комментарии.',
placeholder='Что вас интересует?',
sizes=(1,)
)
ADD_KB = get_kb(
'Выйти без сохранения'
)