You've already forked Travel-Agent
86 lines
2.1 KiB
Python
86 lines
2.1 KiB
Python
__all__ = ("sex_keyboard", "travels_keyboard")
|
|
|
|
from aiogram.types import InlineKeyboardButton
|
|
from aiogram.utils.keyboard import InlineKeyboardBuilder, ReplyKeyboardBuilder
|
|
|
|
from app.config import Config
|
|
|
|
|
|
def sex_keyboard(choices: str | list):
|
|
builder = ReplyKeyboardBuilder()
|
|
|
|
if isinstance(choices, str):
|
|
choices = [choices]
|
|
|
|
[builder.button(text=choice) for choice in choices]
|
|
return builder.as_markup(resize_keyboard=True)
|
|
|
|
|
|
def travels_keyboard(travels: list, page: int, pages: int, user_id: int):
|
|
builder = InlineKeyboardBuilder()
|
|
rows = []
|
|
|
|
start_index = page * Config.PAGE_SIZE
|
|
end_index = min((page + 1) * Config.PAGE_SIZE, len(travels))
|
|
|
|
for travel in travels[start_index:end_index]:
|
|
button_text = travel.title
|
|
|
|
if travel.author_id == user_id:
|
|
button_text += " 👑"
|
|
|
|
rows.append(
|
|
InlineKeyboardButton(
|
|
text=button_text,
|
|
callback_data=f"travel_detail_{travel.id}",
|
|
),
|
|
)
|
|
|
|
for _ in range(0, Config.PAGE_SIZE - len(rows)):
|
|
rows.append(InlineKeyboardButton(text=" ", callback_data="pass"))
|
|
|
|
builder.row(*rows, width=2)
|
|
|
|
navigation_row = []
|
|
|
|
if page > 0:
|
|
navigation_row.append(
|
|
InlineKeyboardButton(
|
|
text="⬅️",
|
|
callback_data=f"travels_page_{page - 1}",
|
|
),
|
|
)
|
|
else:
|
|
navigation_row.append(
|
|
InlineKeyboardButton(
|
|
text=" ",
|
|
callback_data="pass",
|
|
),
|
|
)
|
|
|
|
navigation_row.append(
|
|
InlineKeyboardButton(
|
|
text=f"{page + 1}/{pages}",
|
|
callback_data="pass",
|
|
),
|
|
)
|
|
|
|
if page < pages - 1:
|
|
navigation_row.append(
|
|
InlineKeyboardButton(
|
|
text="➡️",
|
|
callback_data=f"travels_page_{page + 1}",
|
|
),
|
|
)
|
|
else:
|
|
navigation_row.append(
|
|
InlineKeyboardButton(
|
|
text=" ",
|
|
callback_data="pass",
|
|
),
|
|
)
|
|
|
|
builder.row(*navigation_row)
|
|
|
|
return builder.as_markup()
|