75 lines
2.5 KiB
Python
75 lines
2.5 KiB
Python
import logging
|
|
import os
|
|
|
|
import requests
|
|
# from alerta.exceptions import RejectException
|
|
from alerta.plugins import PluginBase
|
|
|
|
try:
|
|
from alerta.plugins import app # alerta >= 5.0
|
|
except ImportError:
|
|
from alerta.app import app # alerta < 5.0
|
|
|
|
LOG = logging.getLogger('alerta.plugins')
|
|
|
|
ALERTOPS_URL = os.environ.get('ALERTOPS_URL') or app.config['ALERTOPS_URL']
|
|
DASHBOARD_URL = os.environ.get('DASHBOARD_URL') or app.config['DASHBOARD_URL']
|
|
|
|
|
|
class TriggerEvent(PluginBase):
|
|
|
|
def pre_receive(self, alert, **kwargs):
|
|
return alert
|
|
|
|
@staticmethod
|
|
def _event_type(severity):
|
|
if severity in ['cleared', 'normal', 'ok']:
|
|
return 'close'
|
|
else:
|
|
return 'open'
|
|
|
|
def post_receive(self, alert, **kwargs):
|
|
if alert.repeat:
|
|
return
|
|
|
|
message = '{}: {} alert for {} - {}'.format(
|
|
alert.environment, alert.severity.capitalize(), ','.join(alert.service), alert.resource)
|
|
|
|
payload = {
|
|
'source_id': alert.id,
|
|
'source_status': TriggerEvent._event_type(alert.severity),
|
|
'description': message,
|
|
'resource': alert.resource,
|
|
'source': 'alerta',
|
|
'source_url': '{}/#/alert/{}'.format(DASHBOARD_URL, alert.id),
|
|
'details': alert.get_body(history=False)}
|
|
LOG.debug('AlertOps Payload: %s', payload)
|
|
|
|
try:
|
|
r = requests.post(ALERTOPS_URL, json=payload, timeout=2)
|
|
except Exception as e:
|
|
raise RuntimeError('AlertOps connection error: %s' % e)
|
|
LOG.debug('AlertOps response: {} - {}'.format(r.status_code, r.text))
|
|
return
|
|
|
|
def status_change(self, alert, status, text, **kwargs):
|
|
if status not in ['ack', 'assign']:
|
|
return
|
|
|
|
message = '{}: {} alert for {} - {}'.format(
|
|
alert.environment, alert.severity.capitalize(), ','.join(alert.service), alert.resource)
|
|
payload = {
|
|
'source_id': alert.id,
|
|
'source_status': TriggerEvent._event_type(alert.severity),
|
|
'description': message,
|
|
'resource': alert.resource,
|
|
'source': 'alerta',
|
|
'source_url': '{}/#/alert/{}'.format(DASHBOARD_URL, alert.id),
|
|
'details': alert.get_body(history=False)}
|
|
|
|
try:
|
|
r = requests.post(ALERTOPS_URL, json=payload, timeout=2)
|
|
except Exception as e:
|
|
raise RuntimeError('AlertOps connection error: %s' % e)
|
|
LOG.debug('AlertOps response: {} - {}'.format(r.status_code, r.text))
|