59 lines
1.6 KiB
Python
59 lines
1.6 KiB
Python
from typing import Optional
|
|
|
|
from aiogram.types import InlineKeyboardButton, InlineKeyboardMarkup
|
|
from aiogram.utils.keyboard import InlineKeyboardBuilder
|
|
|
|
from dto.categories import CategoryDTO
|
|
|
|
|
|
def cart_btn(user_id: int) -> InlineKeyboardButton:
|
|
return InlineKeyboardButton(
|
|
text="🛒 Корзина ()", callback_data="cart"
|
|
) # TODO: Database integration, get cart items count
|
|
|
|
|
|
def get_back_btn(
|
|
data: str = "menu:main", text: Optional[str] = "⬅️ Назад"
|
|
) -> InlineKeyboardButton:
|
|
return InlineKeyboardButton(text=text, callback_data=data)
|
|
|
|
|
|
async def main_menu_kb(user_id: int) -> InlineKeyboardMarkup:
|
|
return InlineKeyboardBuilder(
|
|
[
|
|
[
|
|
InlineKeyboardButton(text="📦 Каталог", callback_data="cat:root"),
|
|
InlineKeyboardButton(text="🔍", callback_data="find"),
|
|
],
|
|
[cart_btn(user_id)],
|
|
]
|
|
).as_markup()
|
|
|
|
|
|
async def render_category(
|
|
children: list[CategoryDTO],
|
|
parent_id: Optional[int] = None,
|
|
show_menu: bool = False,
|
|
) -> InlineKeyboardMarkup:
|
|
btns = [
|
|
InlineKeyboardButton(
|
|
text=category.name,
|
|
callback_data=f"cat:{category.id}",
|
|
)
|
|
for category in children
|
|
]
|
|
|
|
nav = []
|
|
|
|
if parent_id is not None:
|
|
nav.append(
|
|
InlineKeyboardButton(text="⬅️ Назад", callback_data=f"cat:{parent_id}")
|
|
)
|
|
|
|
if show_menu:
|
|
nav.append(InlineKeyboardButton(text="⬅️ Назад", callback_data="menu:main"))
|
|
|
|
return InlineKeyboardBuilder(
|
|
[btns[i : i + 2] for i in range(0, len(children), 2)] + [nav],
|
|
).as_markup()
|