feat(repo): add processing for case when value is not found when upda…

…ting in the database
Desbordante · Oct 2, 2024 · 673611c · 673611c
1 parent b2ffbe3
commit 673611c
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/internal/dto/repository/exception.py b/internal/dto/repository/exception.py
@@ -0,0 +1,3 @@
+class ModelNotFoundException(Exception):
+    def __init__(self, message: str):
+        super().__init__(message)
diff --git a/internal/repository/relational/crud.py b/internal/repository/relational/crud.py
@@ -10,6 +10,7 @@
     BaseResponseSchema,
 )
 from internal.infrastructure.data_storage import Context
+from internal.dto.repository.exception import ModelNotFoundException
 
 
 class CRUD[
@@ -70,6 +71,11 @@ def update(
     ) -> ResponseSchema:
 
         db_model_instance = self._find(find_schema, context)
+        if not db_model_instance:
+            raise ModelNotFoundException(
+                "When updating data, the required row was not found in the database."
+            )
+
         update_schema_dict = update_schema.model_dump()
         fields_to_update_if_none = (
             fields_to_update_if_none if fields_to_update_if_none else set()