def process_action(alert: Alert, action: str, text: str, timeout: int = None) -> Tuple[Alert, str, str, Optional[int]]:
wanted_plugins, wanted_config = plugins.routing(alert)
updated = None
for plugin in wanted_plugins:
if alert.is_suppressed:
break
try:
updated = plugin.take_action(alert, action, text, timeout=timeout, config=wanted_config)
except NotImplementedError:
pass # plugin does not support action() method
except (RejectException, ForwardingLoop, InvalidAction, AlertaException):
raise
except Exception as e:
if current_app.config['PLUGINS_RAISE_ON_ERROR']:
raise ApiError("Error while running action plugin '{}': {}".format(plugin.name, str(e)))
else:
logging.error("Error while running action plugin '{}': {}".format(plugin.name, str(e)))
if isinstance(updated, Alert):
updated = updated, action, text, timeout
if isinstance(updated, tuple):
if len(updated) == 4:
alert, action, text, timeout = updated
elif len(updated) == 3:
alert, action, text = updated
if updated:
alert.update_tags(alert.tags)
alert.attributes = alert.update_attributes(alert.attributes)
return alert, action, text, timeout
def process_note(alert: Alert, text: str) -> Tuple[Alert, str]:
wanted_plugins, wanted_config = plugins.routing(alert)
updated = None
for plugin in wanted_plugins:
try:
updated = plugin.take_note(alert, text, config=wanted_config)
except NotImplementedError:
pass # plugin does not support take_note() method
except (RejectException, ForwardingLoop, AlertaException):
raise
except Exception as e:
if current_app.config['PLUGINS_RAISE_ON_ERROR']:
raise ApiError("Error while running note plugin '{}': {}".format(plugin.name, str(e)))
else:
logging.error("Error while running note plugin '{}': {}".format(plugin.name, str(e)))
if isinstance(updated, Alert):
updated = updated, text
if isinstance(updated, tuple) and len(updated) == 2:
alert, text = updated
if updated:
alert.update_tags(alert.tags)
alert.update_attributes(alert.attributes)
return alert, text
def process_status(alert: Alert, status: str, text: str) -> Tuple[Alert, str, str]:
wanted_plugins, wanted_config = plugins.routing(alert)
updated = None
for plugin in wanted_plugins:
if alert.is_suppressed:
break
try:
updated = plugin.status_change(alert, status, text, config=wanted_config)
except TypeError:
updated = plugin.status_change(alert, status, text) # for backward compatibility
except (RejectException, AlertaException):
raise
except Exception as e:
if current_app.config['PLUGINS_RAISE_ON_ERROR']:
raise ApiError("Error while running status plugin '{}': {}".format(plugin.name, str(e)))
else:
logging.error("Error while running status plugin '{}': {}".format(plugin.name, str(e)))
if updated:
try:
alert, status, text = updated
except Exception:
alert = updated
if updated:
alert.update_tags(alert.tags)
alert.attributes = alert.update_attributes(alert.attributes)
return alert, status, text
def process_alert(alert: Alert) -> Alert:
wanted_plugins, wanted_config = plugins.routing(alert)
skip_plugins = False
for plugin in wanted_plugins:
if alert.is_suppressed:
skip_plugins = True
break
try:
alert = plugin.pre_receive(alert, config=wanted_config)
except TypeError:
alert = plugin.pre_receive(alert) # for backward compatibility
except (RejectException, HeartbeatReceived, BlackoutPeriod, RateLimit, ForwardingLoop):
raise
except Exception as e:
if current_app.config['PLUGINS_RAISE_ON_ERROR']:
raise RuntimeError("Error while running pre-receive plugin '{}': {}".format(plugin.name, str(e)))
else:
logging.error("Error while running pre-receive plugin '{}': {}".format(plugin.name, str(e)))
if not alert:
raise SyntaxError("Plugin '%s' pre-receive hook did not return modified alert" % plugin.name)
try:
is_duplicate = alert.is_duplicate()
if is_duplicate:
alert = alert.deduplicate(is_duplicate)
else:
is_correlated = alert.is_correlated()
if is_correlated:
alert = alert.update(is_correlated)
else:
alert = alert.create()
except Exception as e:
raise ApiError(str(e))
updated = None
for plugin in wanted_plugins:
if skip_plugins:
break
try:
updated = plugin.post_receive(alert, config=wanted_config)
except TypeError:
updated = plugin.post_receive(alert) # for backward compatibility
except Exception as e:
if current_app.config['PLUGINS_RAISE_ON_ERROR']:
raise ApiError("Error while running post-receive plugin '{}': {}".format(plugin.name, str(e)))
else:
logging.error("Error while running post-receive plugin '{}': {}".format(plugin.name, str(e)))
if updated:
alert = updated
if updated:
alert.update_tags(alert.tags)
alert.attributes = alert.update_attributes(alert.attributes)
return alert
