83 lines
3.6 KiB
Python
83 lines
3.6 KiB
Python
from aiogram.filters.callback_data import CallbackData
|
||
from aiogram.types import InlineKeyboardButton
|
||
from aiogram.utils.keyboard import InlineKeyboardBuilder
|
||
|
||
from bot.handlers.user.model_processing import get_model_page, ChooseCallBack
|
||
|
||
|
||
class PageCallBack(CallbackData, prefix="page"):
|
||
page: int
|
||
page_name: str
|
||
|
||
def get_user_main_btns(*, page: int): # page 0
|
||
btns = [
|
||
InlineKeyboardButton(text="Бесплатные уроки", callback_data=PageCallBack(page=1, page_name="lessons").pack()),#1
|
||
InlineKeyboardButton(text="Выбрать тариф", callback_data=PageCallBack(page=2, page_name="tarif").pack()),#2
|
||
InlineKeyboardButton(text="Больше о моделях", callback_data=PageCallBack(page=3, page_name="more_about_models").pack()),#3
|
||
]
|
||
|
||
return get_kb_buttons(btns=btns)
|
||
|
||
|
||
def get_user_lessons_page(*, page): # page 1
|
||
btns = [
|
||
InlineKeyboardButton(text="Выбрать модель", callback_data=PageCallBack(page=2, page_name="tarif").pack()), # 2
|
||
InlineKeyboardButton(text="⬅️Назад", callback_data=PageCallBack(page=0, page_name="main").pack()), # 2
|
||
]
|
||
|
||
return get_kb_buttons(btns=btns)
|
||
|
||
def get_user_tarif_page(*, page): # page 2
|
||
btns = [
|
||
InlineKeyboardButton(text="1o", callback_data=PageCallBack(page=4, page_name='1o').pack()), # 4
|
||
InlineKeyboardButton(text="1o mini", callback_data=PageCallBack(page=5, page_name='1o mini').pack()), # 5
|
||
InlineKeyboardButton(text="4o", callback_data=PageCallBack(page=6, page_name='4o').pack()), # 6
|
||
InlineKeyboardButton(text="4o mini", callback_data=PageCallBack(page=7, page_name='4o mini').pack()), # 7
|
||
InlineKeyboardButton(text="⬅️Назад", callback_data=PageCallBack(page=0, page_name="main").pack()), # 1
|
||
]
|
||
|
||
return get_kb_buttons(btns=btns)
|
||
|
||
async def get_user_model_description_page(*, page: int, name: str, user_id):
|
||
text, kbd = await get_model_page(page=page, name=name, user_id=user_id)
|
||
|
||
kbd.add(InlineKeyboardButton(text="⬅️Назад", callback_data=PageCallBack(page=2, page_name="tarif").pack()),)
|
||
|
||
return text, kbd.adjust(*(1,)).as_markup()
|
||
|
||
def get_more_about_modules_page(*, page: int): # page 3
|
||
btns = [
|
||
# InlineKeyboardButton(text="Какая модель мне подходит", callback_data=PageCallBack(page=8, page_name='match_model').pack()),
|
||
InlineKeyboardButton(text="Как узнать, какая модель мне подходит", callback_data='choose'),
|
||
InlineKeyboardButton(text="Узнать стоимость моделей", callback_data=PageCallBack(page=9, page_name='price_list').pack()),
|
||
InlineKeyboardButton(text="Выбрать модель", callback_data=PageCallBack(page=2, page_name="tarif").pack()), # 2
|
||
InlineKeyboardButton(text="⬅️Назад", callback_data=PageCallBack(page=0, page_name="main").pack()),
|
||
]
|
||
|
||
return get_kb_buttons(btns=btns)
|
||
|
||
def get_price_list_page(*, page: int):
|
||
btns = [
|
||
InlineKeyboardButton(text="⬅️Назад", callback_data=PageCallBack(page=3, page_name="more_about_models").pack()),
|
||
]
|
||
|
||
return get_kb_buttons(btns=btns)
|
||
|
||
|
||
def get_kb_buttons(*, btns: list[InlineKeyboardButton], sizes: tuple[int] = (1,)):
|
||
kb_temp = InlineKeyboardBuilder()
|
||
|
||
for btn in btns:
|
||
kb_temp = kb_temp.add(btn)
|
||
|
||
return kb_temp.adjust(*sizes).as_markup()
|
||
|
||
def get_callback_btns(*, btns: dict[str, str]):
|
||
keyboard = InlineKeyboardBuilder
|
||
|
||
for text, data in btns.items():
|
||
keyboard.add(InlineKeyboardButton(text=text, callback_data=data).pack())
|
||
|
||
return keyboard.as_markup()
|
||
|