keyboards/inline_kb.py - master - AlexF88/FFB

keyboards/inline_kb.py
63 строки2 KB
Alex
Add func delete product by id from db
16 июл 2026, 17:50
16 июл 2026, 17:50fe5ec15
from aiogram.utils.keyboard import InlineKeyboardBuilder, InlineKeyboardMarkup, InlineKeyboardButton

from database.db_utils import (db_get_all_category, db_get_products,
                               db_get_finally_price, db_get_product_for_delete)


def generate_category_menu(chat_id: int) -> InlineKeyboardMarkup:
    '''Кнопки категории'''
    categoies = db_get_all_category()
    total_price = db_get_finally_price(chat_id)
    builder = InlineKeyboardBuilder()
    builder.button(text=f'Ваша корзина ({total_price if total_price else 0})',
                   callback_data='Ваша корзина')
    [builder.button(text=category.category_name,
                    callback_data=f'category_{category.id}') for category in categoies]

    builder.adjust(1, 2)

    return builder.as_markup()


def show_product_by_catrgory(category_id: int):
    '''Кнопки продуктов'''
    products = db_get_products(category_id)
    builder = InlineKeyboardBuilder()
    [builder. button(text=product.product_name,
                     callback_data=f'product_{product.ia}') for product in products]
    builder.adjust(2)
    builder.row(
        InlineKeyboardButton(text='Назад',
                             callback_data='return_to_category')
    )
    return builder.as_markup()


def generate_constructor_button(quantity:int=1) -> InlineKeyboardMarkup:
    '''Кнопка выбора количества продуктов'''
    builder = InlineKeyboardBuilder()
    builder.button(text='-',
                   callback_data='action -')
    builder.button(text=str(quantity),
                   callback_data=str(quantity))
    builder.button(text='+',
                   callback_data='action +')
    builder.button(text='Положить в корзину',
                   callback_data='put into cart')
    builder.adjust(3, 1)

    return builder.as_markup()


def generate_delete_product(chat_id: int) -> InlineKeyboardMarkup:
    '''Удаление продукта'''
    builder = InlineKeyboardBuilder()
    cart_products = db_get_product_for_delete(chat_id)
    builder.button(text='Оформить заказ',
                   callback_data='order_pay')
    for finally_cart_id, product_name in cart_products:
        builder.button(text=f'{product_name}',
                       callback_data=f'delete_{finally_cart_id}')

    builder.adjust(1)
    return builder.as_markup()