""" Action superclass with common functionality """
-import sys
+import sys, ConfigParser
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.
# 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):
"""
"""
self.conf = config.load_config(filename, self.default_config)
-
+ def get_config_optional(self, section, option):
+ """
+ Returns the value of the given option. If the option was not set
+ (and no default was set in self.default_config), return None.
+
+ This is a convenience wrapper for ConfigParser.get(), since that
+ throws an exception on unset options.
+ """
+ try:
+ return self.conf.get(section, option)
+ except ConfigParser.NoOptionError:
+ return None
+
+ def get_config_mandatory(self, section, option):
+ """
+ Returns the value of the given option. If the option was not set
+ (and no default was set in self.default_config), raises a
+ backupninja.config.ConfigError.
+
+ This is a convenience wrapper for ConfigParser.get(), since that
+ has a very generic exception message on unknown options.
+ """
+ try:
+ return self.conf.get(section, option)
+ except ConfigParser.NoOptionError:
+ raise config.ConfigError("Option '%s' in section '%s' is mandatory, please configure it" % (option, section))
def create_action(ty):
"""