Fixed type hints, added cart item counter on main

services/orders.py

@@ -93,9 +93,15 @@ class OrderService:
         order_item = await self.order_items_repo.get_item_by_order_and_product(
             session, order_id=order.id, product_id=product_id
         )
+
+        if not order_item:
+            return 0
+
         return order_item.quantity
 
-    async def clear_cart_item(self, session, *, customer: int, product_id: int) -> None:
+    async def clear_cart_item(
+        self, session: AsyncSession, *, customer: int, product_id: int
+    ) -> None:
         order = await self.order_repo.get_draft_order_by_user(session, customer)
 
         if not order:
@@ -115,3 +121,13 @@ class OrderService:
         await self.order_items_repo.update_order_item_quantity(
             session, order_item_id=order_item.id, quantity=0
         )
+
+    async def get_cart_items_count(self, session: AsyncSession, customer: int) -> int:
+        order = await self.order_repo.get_draft_order_by_user(session, customer)
+        if not order:
+            return 0
+
+        order_item = await self.order_items_repo.get_items_by_order(
+            session, order_id=order.id
+        )
+        return len(order_item)