ilert-check-mk-v2.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# iLert Check_MK Native Plugin
#
# Copyright (c) 2013-2020, iLert GmbH. <support@ilert.com>
# All rights reserved.


import os
import json
import argparse
import io
import datetime
import sys

import requests

PLUGIN_VERSION = "1.0"


def log(level, message):
    sys.stdout.write("%s %s %s\n" %
                     (datetime.datetime.now().isoformat(), level, message))


def send(endpoint, port, apiKey, context):
    log("INFO", "Sending event to iLert...")
    headers = {"Content-type": "application/json",
               "Accept": "application/json",
               "Agent": "checkmk/extension/%s" % PLUGIN_VERSION}
    url = "%s:%s/api/v1/events/checkmk-ext/%s" % (endpoint, port, apiKey)

    try:
        response = requests.post(url,
                                 data=json.dumps(context),
                                 headers=headers,
                                 timeout=60)
    except requests.HTTPError as e:
        if e.code == 429:
            log("WARNING", "too many requests, will try later. Server response: %s" % e.read())
            exit(1)
        elif 400 <= e.code <= 499:
            log("WARNING", "event not accepted by iLert. Reason: %s" % e.read())
        else:
            log("ERROR", "could not send event to iLert. HTTP error code %s, reason: %s, %s" % (
                e.code, e.reason, e.read()))
            exit(1)
    except Exception as e:
        log("ERROR", "an unexpected error occurred. Please report a bug. Cause: %s %s" % (
            type(e), e.args))
        exit(1)
    else:
        log("INFO", "Event has been sent to iLert")


def main():
    parser = argparse.ArgumentParser(
        description='send events from CheckMK to iLert')
    parser.add_argument(
        '-a', '--apikey', help='API key for the alert source in iLert')
    parser.add_argument('-e', '--endpoint', default='https://api.ilert.com',
                        help='iLert API endpoint (default: %(default)s)')
    parser.add_argument('-p', '--port', type=int, default=443,
                        help='endpoint port (default: %(default)s)')
    parser.add_argument('--version', action='version', version=PLUGIN_VERSION)
    parser.add_argument('payload', nargs=argparse.REMAINDER,
                        help='event payload as key value pairs in the format key1=value1 key2=value2 ...')
    args = parser.parse_args()

    # TODO: use cmk utils after release
    # from cmk.notification_plugins import utils
    # ...
    # context = utils.collect_context()
    # ...

    # get all env vars to dict
    context = dict(PLUGIN_VERSION=PLUGIN_VERSION)
    for env in os.environ:
        if "NOTIFY_" in env:
            context[env] = os.environ[env]

    # ... and payload specified via command line
    for arg in args.payload:
        if arg:
            a = arg.split('=', 1)
            if a and a[0] and a[1]:
                context.update({a[0]: a[1]})

    if args.apikey is not None:
        apikey = args.apikey
    elif 'NOTIFY_PARAMETER_ILERT_API_KEY' in context:
        apikey = context.get['NOTIFY_PARAMETER_ILERT_API_KEY']
    else:
        apikey = None

    if apikey is None:
        log("ERROR", "parameter apikey is required in save mode and must be provided either via command line or in the pager field of the contact definition in CheckMK")
        exit(1)
    send(args.endpoint, args.port, apikey, context)

    exit(0)


if __name__ == '__main__':
    main()