alerta_alerta-contrib/plugins/cachet/alerta_cachet.py

65 lines
1.6 KiB
Python

import json
import logging
import os
import cachetclient.cachet as cachet
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.cachet')
CACHET_API_URL = os.environ.get(
'CACHET_API_URL') or app.config['CACHET_API_URL']
CACHET_API_TOKEN = os.environ.get(
'CACHET_API_TOKEN') or app.config['CACHET_API_TOKEN']
CACHET_SSL_VERIFY = True if (os.environ.get(
'CACHET_SSL_VERIFY') == 'True' or app.config.get('CACHET_SSL_VERIFY', False)) else False
STATUS_MAP = {
'open': 1, # Investigating
'ack': 2, # Identified
'assigned': 3, # Watching
'closed': 4 # Fixed
}
class CachetIncident(PluginBase):
def __init__(self, name=None):
self.incidents = cachet.Incidents(
endpoint=CACHET_API_URL, api_token=CACHET_API_TOKEN, verify=CACHET_SSL_VERIFY)
super().__init__(name)
def pre_receive(self, alert):
return alert
def post_receive(self, alert):
name = alert.event
status = STATUS_MAP[alert.status]
message = alert.text
r = json.loads(self.incidents.get(
name=name, message=message, status=status))
if r['meta']['pagination']['count']:
return
try:
r = json.loads(self.incidents.post(
name=name, message=message, status=status, visible=True))
except Exception as e:
raise RuntimeError('Cachet: ERROR - %s', e)
LOG.debug('Cachet: %s', r)
def status_change(self, alert, status, text):
return