refactor: entity not found exceptions handler

modules/fulfillment_base/repositories/deal_product.py

@@ -12,6 +12,7 @@ from modules.fulfillment_base.schemas.deal_product import (
 )
 from repositories.base import BaseRepository
 from repositories.mixins import RepGetAllMixin, RepUpdateMixin
+from utils.exceptions import ObjectNotFoundException
 
 
 class DealProductRepository(
@@ -34,7 +35,9 @@ class DealProductRepository(
             .order_by(DealProduct.product_id)
         )
 
-    async def get_by_id(self, deal_id: int, product_id: int) -> Optional[DealProduct]:
+    async def get_by_id(
+        self, deal_id: int, product_id: int, raise_if_not_found: Optional[bool] = True
+    ) -> Optional[DealProduct]:
         stmt = (
             select(DealProduct)
             .options(
@@ -45,8 +48,10 @@ class DealProductRepository(
             )
             .where(DealProduct.deal_id == deal_id, DealProduct.product_id == product_id)
         )
-        result = await self.session.execute(stmt)
-        return result.scalar_one_or_none()
+        result = (await self.session.execute(stmt)).scalar_one_or_none()
+        if result is None and raise_if_not_found:
+            raise ObjectNotFoundException("Связь сделки с товаром не найдена")
+        return result
 
     async def create(self, data: CreateDealProductSchema):
         deal_product = DealProduct(**data.model_dump())