alerta-contrib/integrations/consul/consulalerta.py

#!/usr/bin/env python

import json
import os
import sys
import time

import consul
from alertaclient.api import Client

CONSUL_HOST = os.environ.get('CONSUL_HOST', '127.0.0.1')
CONSUL_PORT = int(os.environ.get('CONSUL_PORT', 8500))

client = consul.Consul(host=CONSUL_HOST, port=CONSUL_PORT, token=None,
                       scheme='http', consistency='default', dc=None, verify=True)

j = json.load(sys.stdin)
print('Request:')
print(j)

try:
    url = client.kv.get('alerta/apiurl')[1]['Value']
except Exception:
    print('No URL defined, exiting')
    sys.exit(1)

try:
    key = client.kv.get('alerta/apikey')[1]['Value']
except Exception:
    print('No key defined, exiting')
    sys.exit(1)


try:
    max_retries = int(client.kv.get('alerta/max_retries')[1]['Value'])
except TypeError:
    print('No value defined, using default')
    max_retries = 3

try:
    sleep = int(client.kv.get('alerta/sleep')[1]['Value'])
except TypeError:
    print('No value defined, using default')
    sleep = 2

try:
    timeout = int(client.kv.get('alerta/timeout')[1]['Value'])
except TypeError:
    print('No value defined, using default')
    timeout = 900

try:
    origin = client.kv.get('alerta/origin')[1]['Value']
except TypeError:
    print('No value defined, using default')
    origin = 'consul'

try:
    alerttype = client.kv.get('alerta/alerttype')[1]['Value']
except TypeError:
    print('No value defined, using default')
    alerttype = 'ConsulAlert'


api = Client(endpoint=url, key=key)

SEVERITY_MAP = {
    'critical': 'critical',
    'warning': 'warning',
    'passing': 'ok',
}


def createalert(data):
    try:
        environment = client.kv.get(
            'alerta/env/{}'.format(data['Node']))[1]['Value']
    except Exception:
        try:
            environment = client.kv.get('alerta/defaultenv')[1]['Value']
        except Exception:
            environment = 'Production'

    for _ in range(max_retries):
        try:
            print('Response:')
            response = api.send_alert(
                resource=data['Node'],
                event=data['CheckId'],
                value=data['Status'],
                correlate=SEVERITY_MAP.keys(),
                environment=environment,
                service=[data['CheckId']],
                severity=SEVERITY_MAP[data['Status']],
                text=data['Output'],
                timeout=timeout,
                origin=origin,
                type=alerttype
            )
            print(response)
        except Exception as e:
            print('HTTP Error: {}'.format(e))
            time.sleep(sleep)
            continue
        else:
            break
    else:
        print('api is down')


def main():
    for item in enumerate(j):
        i = item[0]
        createalert(j[i])


if __name__ == '__main__':
    main()