mirror of
https://github.com/alerta/alerta-contrib.git
synced 2025-03-16 21:43:30 +00:00

* plugin/alerta-prometheus: fix 500 error when response is not as expected Alertmanager Api may respond with empty set of silenceIds, which may cause exception "alerta.exceptions.ApiError: Alertmanager: ERROR - list indices must be integers or slices, not str" This error is seen from the client, and the Alert can not be ACK-ed * plugin/alerta-prometheus: minor code improvement
123 lines
5 KiB
Python
123 lines
5 KiB
Python
|
|
import datetime
|
|
import logging
|
|
import os
|
|
import requests
|
|
|
|
try:
|
|
from alerta.plugins import app # alerta >= 5.0
|
|
except ImportError:
|
|
from alerta.app import app # alerta < 5.0
|
|
from alerta.plugins import PluginBase
|
|
|
|
LOG = logging.getLogger('alerta.plugins.prometheus')
|
|
|
|
DEFAULT_ALERTMANAGER_API_URL = 'http://localhost:9093'
|
|
|
|
ALERTMANAGER_API_URL = os.environ.get('ALERTMANAGER_API_URL') or app.config.get('ALERTMANAGER_API_URL', None)
|
|
ALERTMANAGER_USERNAME = os.environ.get('ALERTMANAGER_USERNAME') or app.config.get('ALERTMANAGER_USERNAME', None)
|
|
ALERTMANAGER_PASSWORD = os.environ.get('ALERTMANAGER_PASSWORD') or app.config.get('ALERTMANAGER_PASSWORD', None)
|
|
ALERTMANAGER_SILENCE_DAYS = os.environ.get('ALERTMANAGER_SILENCE_DAYS') or app.config.get('ALERTMANAGER_SILENCE_DAYS', 1)
|
|
ALERTMANAGER_SILENCE_FROM_ACK = os.environ.get('ALERTMANAGER_SILENCE_FROM_ACK') or app.config.get('ALERTMANAGER_SILENCE_FROM_ACK', False)
|
|
|
|
class AlertmanagerSilence(PluginBase):
|
|
|
|
def __init__(self, name=None):
|
|
|
|
self.auth = (ALERTMANAGER_USERNAME, ALERTMANAGER_PASSWORD) if ALERTMANAGER_USERNAME else None
|
|
|
|
super(AlertmanagerSilence, self).__init__(name)
|
|
|
|
def pre_receive(self, alert):
|
|
return alert
|
|
|
|
def post_receive(self, alert):
|
|
return
|
|
|
|
def status_change(self, alert, status, text):
|
|
|
|
if alert.event_type != 'prometheusAlert':
|
|
return
|
|
|
|
if alert.status == status:
|
|
return
|
|
|
|
if status == 'ack':
|
|
|
|
if ALERTMANAGER_SILENCE_FROM_ACK:
|
|
silence_seconds = int(alert.timeout)
|
|
else:
|
|
try:
|
|
silence_days = int(ALERTMANAGER_SILENCE_DAYS)
|
|
except Exception as e:
|
|
LOG.error(
|
|
"Alertmanager: Could not parse 'ALERTMANAGER_SILENCE_DAYS': %s", e)
|
|
raise RuntimeError(
|
|
"Could not parse 'ALERTMANAGER_SILENCE_DAYS': %s" % e)
|
|
silence_seconds = silence_days * 86400
|
|
|
|
LOG.debug('Alertmanager: Add silence for alertname=%s instance=%s timeout=%s',
|
|
alert.event, alert.resource, str(silence_seconds))
|
|
data = {
|
|
"matchers": [
|
|
{
|
|
"name": "alertname",
|
|
"value": alert.event
|
|
},
|
|
{
|
|
"name": "instance",
|
|
"value": alert.resource
|
|
}
|
|
],
|
|
"startsAt": datetime.datetime.utcnow().replace(microsecond=0).isoformat() + ".000Z",
|
|
"endsAt": (datetime.datetime.utcnow() + datetime.timedelta(seconds=silence_seconds))
|
|
.replace(microsecond=0).isoformat() + ".000Z",
|
|
"createdBy": "alerta",
|
|
"comment": text if text != '' else "silenced by alerta"
|
|
}
|
|
|
|
base_url = ALERTMANAGER_API_URL or alert.attributes.get('externalUrl', DEFAULT_ALERTMANAGER_API_URL)
|
|
url = base_url + '/api/v1/silences'
|
|
|
|
LOG.debug('Alertmanager: URL=%s', url)
|
|
LOG.debug('Alertmanager: data=%s', data)
|
|
|
|
try:
|
|
r = requests.post(url, json=data, auth=self.auth, timeout=2)
|
|
except Exception as e:
|
|
raise RuntimeError("Alertmanager: ERROR - %s" % e)
|
|
LOG.debug('Alertmanager: %s - %s', r.status_code, r.text)
|
|
|
|
# example r={"status":"success","data":{"silenceId":8}}
|
|
try:
|
|
data = r.json().get('data', [])
|
|
if data:
|
|
silenceId = data['silenceId']
|
|
alert.attributes['silenceId'] = silenceId
|
|
else:
|
|
silenceId = alert.attributes.get('silenceId', "unknown")
|
|
text = text + ' (silenced in Alertmanager)'
|
|
except Exception as e:
|
|
raise RuntimeError("Alertmanager: ERROR - %s" % e)
|
|
LOG.debug('Alertmanager: Added silenceId %s to attributes', silenceId)
|
|
|
|
elif status == 'open':
|
|
LOG.debug('Alertmanager: Remove silence for alertname=%s instance=%s', alert.event, alert.resource)
|
|
|
|
silenceId = alert.attributes.get('silenceId', None)
|
|
if silenceId:
|
|
base_url = ALERTMANAGER_API_URL or alert.attributes.get('externalUrl', DEFAULT_ALERTMANAGER_API_URL)
|
|
url = base_url + '/api/v1/silence/%s' % silenceId
|
|
try:
|
|
r = requests.delete(url, auth=self.auth, timeout=2)
|
|
except Exception as e:
|
|
raise RuntimeError("Alertmanager: ERROR - %s" % e)
|
|
LOG.debug('Alertmanager: %s - %s', r.status_code, r.text)
|
|
|
|
try:
|
|
alert.attributes['silenceId'] = None
|
|
except Exception as e:
|
|
raise RuntimeError("Alertmanager: ERROR - %s" % e)
|
|
LOG.debug('Alertmanager: Removed silenceId %s from attributes', silenceId)
|
|
|
|
return alert, status, text
|