fixup! sf_base: add decorator to gracefully handle exceptions

OCA · Oct 30, 2023 · 96812b3 · 96812b3
1 parent c0b26dc
commit 96812b3
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/shopfloor_base/utils.py b/shopfloor_base/utils.py
@@ -43,15 +43,18 @@ def catch_errors(response_error):
 
     def _catch_errors(fn):
         @wraps(fn)
-        def wrapper(*args, **kwargs):
+        def wrapper(self, *args, **kwargs):
+            savepoint = self._actions_for("savepoint").new()
             try:
-                res = fn(*args, **kwargs)
+                res = fn(self, *args, **kwargs)
             except ValidationError as ex:
+                savepoint.rollback()
                 message = {
                     "message_type": "error",
                     "body": ex.name,
                 }
-                return response_error(*args, **kwargs, message=message)
+                return response_error(self, *args, **kwargs, message=message)
+            savepoint.release()
             return res
 
         return wrapper