Decorator that throws traps a twisted error and rethrows your app specific error.

twisted_wrap_exception.py

def wrap_exception(handle, error_type, error_message, log=None):
  def attach_check(method):
    @functools.wraps(method)
    def wrapper(self, *args, **kwargs):
        d = method(self, *args, **kwargs)
        def on_error(failure):
          failure.trap(handle)
          logger = logging if not log else logging.getLogger(log)
          logger.error("%s, %s" % (error_message, failure.getErrorMessage()))
          raise error_type(error_message)
        d.addErrback(on_error)
        return d
    return wrapper
  return attach_check


# Example

@wrap_exception(ConnectError, ExternalStorageException, 'error indexing block in elastic', log='SbrDao')
def elastic_index(self, block):
  pass