Allow commit-events to abort by returning False

machinable-org · May 24, 2024 · 299ea20 · 299ea20
1 parent 26939cb
commit 299ea20
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/machinable/interface.py b/src/machinable/interface.py
@@ -303,7 +303,9 @@ def commit(self) -> Self:
         if index.find_by_id(self.uuid) is not None:
             return self
 
-        self.on_before_commit()
+        if self.on_before_commit() is False:
+            # allow on_before_commit to abort
+            return self
 
         self.__model__.context = context = self.compute_context()
         self.__model__.uuid = update_uuid_payload(self.__model__.uuid, context)
@@ -312,7 +314,9 @@ def commit(self) -> Self:
         assert self.config is not None
         self.__model__.predicate = self.compute_predicate()
 
-        self.on_commit()
+        if self.on_commit() is False:
+            # allow on_commit to abort
+            return self
 
         # commit to index
         for k, v in self.__related__.items():