refactor: entity not found exceptions handler

modules/fulfillment_base/repositories/product_service.py

@@ -8,6 +8,7 @@ from modules.fulfillment_base.models import DealProductService
 from modules.fulfillment_base.schemas.product_service import *
 from repositories.base import BaseRepository
 from repositories.mixins import RepUpdateMixin
+from utils.exceptions import ObjectNotFoundException
 
 
 class ProductServiceRepository(
@@ -15,10 +16,15 @@ class ProductServiceRepository(
     RepUpdateMixin[DealProductService, UpdateProductServiceSchema],
 ):
     entity_class = DealProductService
+    entity_not_found_msg = "Связь услуги с товаром не найдена"
     session: AsyncSession
 
     async def get_by_id(
-        self, deal_id: int, product_id: int, service_id: int
+        self,
+        deal_id: int,
+        product_id: int,
+        service_id: int,
+        raise_if_not_found: Optional[bool] = True,
     ) -> Optional[DealProductService]:
         stmt = (
             select(DealProductService)
@@ -31,8 +37,10 @@ class ProductServiceRepository(
                 DealProductService.service_id == service_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: CreateProductServiceSchema):
         deal_product_service = DealProductService(**data.model_dump())