def execute_action(self, action, params): if action == 'start': try: self.start(params['job_id'], params['current_tag']) except freezer_api_exc.BadDataFormat: raise except Exception as e: raise freezer_api_exc.FreezerAPIException(e) elif action == 'end': try: self.end(params['job_id'], params['result']) except freezer_api_exc.BadDataFormat: raise except Exception as e: raise freezer_api_exc.FreezerAPIException(e) else: raise freezer_api_exc.MethodNotImplemented("Bad Action Method")
def execute_event(self, event, params): handler = self.event_handlers.get(event, None) if not handler: raise freezer_api_exc.BadDataFormat("Bad Action Method") try: self.event_result = handler(params) except freezer_api_exc.BadDataFormat: raise except Exception as e: raise freezer_api_exc.FreezerAPIException(e) return self.event_result
def as_before_hook(self): """Extract process_request method as "before" hook :return: before hook function """ # Need to wrap this up in a closure because the parameter counts # differ def before_hook(req, resp, params=None): return self.process_request(req, resp) try: return before_hook except AttributeError as ex: # No such method, we presume. message_template = ("Failed to get before hook from middleware " "{0} - {1}") message = message_template.format(self.__name__, ex.message) LOG.error(message) LOG.exception(ex) raise freezer_api_exc.FreezerAPIException(message)
def test_FreezerAPIException(self): e = exceptions.FreezerAPIException(message='testing') self.assertRaises(falcon.HTTPError, e.handle, self.ex, self.mock_req, self.mock_req, None)