invoices, order handling, db updates

handlers/client/menu.py

@@ -5,6 +5,7 @@ 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()
@@ -12,11 +13,16 @@ router = Router()
 
 @router.message(Command("start"))
 async def main_menu(
-    msg: Message, state: FSMContext, session: AsyncSession, order_service: OrderService
+    msg: Message,
+    state: FSMContext,
+    session: AsyncSession,
+    order_items_repo: OrderItemRepository,
 ):
     await state.clear()
 
-    cart_items = await order_service.get_cart_items_count(session, msg.from_user.id)
+    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
@@ -27,11 +33,13 @@ async def main_menu_cb(
     cb: CallbackQuery,
     state: FSMContext,
     session: AsyncSession,
-    order_service: OrderService,
+    order_items_repo: OrderService,
 ):
     await state.clear()
 
-    cart_items = await order_service.get_cart_items_count(session, cb.from_user.id)
+    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