51 lines
1.3 KiB
Python
51 lines
1.3 KiB
Python
from aiogram import Router, F
|
|
from aiogram.types import CallbackQuery, Message
|
|
from aiogram.filters import Command
|
|
from aiogram.fsm.context import FSMContext
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from misc.kb import main_menu_kb
|
|
from repositories.order_items import OrderItemRepository
|
|
from services.orders import OrderService
|
|
|
|
router = Router()
|
|
|
|
|
|
@router.message(Command("start"))
|
|
async def main_menu(
|
|
msg: Message,
|
|
state: FSMContext,
|
|
session: AsyncSession,
|
|
order_items_repo: OrderItemRepository,
|
|
):
|
|
await state.clear()
|
|
|
|
cart_items = await order_items_repo.get_items_count_by_customer(
|
|
session, msg.from_user.id
|
|
)
|
|
await msg.answer(
|
|
"hii!", reply_markup=main_menu_kb(cart_items)
|
|
) # TODO: Write a welcome message
|
|
|
|
|
|
@router.callback_query(F.data.startswith("menu:"))
|
|
async def main_menu_cb(
|
|
cb: CallbackQuery,
|
|
state: FSMContext,
|
|
session: AsyncSession,
|
|
order_items_repo: OrderService,
|
|
):
|
|
await state.clear()
|
|
|
|
try:
|
|
await cb.message.reply_to_message.delete()
|
|
except Exception:
|
|
...
|
|
|
|
cart_items = await order_items_repo.get_items_count_by_customer(
|
|
session, cb.from_user.id
|
|
)
|
|
await cb.message.edit_text(
|
|
"hii!", reply_markup=main_menu_kb(cart_items)
|
|
) # TODO: Write a welcome message
|