X-Git-Url: https://git.stderr.nl/gitweb?p=matthijs%2Fprojects%2Fbackupninja.git;a=blobdiff_plain;f=src%2Flib%2Fbackupninja%2Fhandlers%2F__init__.py;h=29dc75631823ed79d5ecda18b99698d6dcacb782;hp=f6559f9b692294b401244f7dc65eb47c203a8779;hb=96f1913c89c0d934c9b75ff9fae2c0eda44e4823;hpb=101bff5ef5fec6382ab572a286c4e79d0312c6cd

diff --git a/src/lib/backupninja/handlers/__init__.py b/src/lib/backupninja/handlers/__init__.py
index f6559f9..29dc756 100644
--- a/src/lib/backupninja/handlers/__init__.py
+++ b/src/lib/backupninja/handlers/__init__.py
@@ -24,6 +24,20 @@ import logging as log
 
 from backupninja import config
 
+def fail_on_exception(f):
+    """
+    This is a decorator meant for methods on the Action class. It
+    catches any exceptions thrown, sets the failed attribute to True and
+    rethrows the exception.
+    """
+    def inner(self, *args, **kwargs):
+        try:
+            f(self, *args, **kwargs)
+        except:
+            self.failed = True
+            raise
+    return inner
+
 class Action(object):
     """
     Subclasses of Action represent handlers for various action types.
@@ -38,6 +52,9 @@ class Action(object):
         # See backupninja.config.load_config for the structure of this
         # value.
         self.default_config = {}
+        # Assume we'll run succesfully. If anything fails in the
+        # meanwhile, set this to True.
+        self.failed = False
 
     def run(self, **kwargs):
         """