source: monitoring/plugins/mpi/check_clarin_discojuice_json.py @ 2081

#!/usr/bin/python

import sys, getopt, httplib, subprocess, os, re, pdb, traceback, datetime
import simplejson as json
#import xml.etree.ElementTree


DESCRIPTION     = "Discojuice JSON"

nagios_codes    = {
                    'OK'        : 0,
                    'WARNING'   : 1,
                    'CRITICAL'  : 2,
                    'UNKNOWN'   : 3,
                    'DEPENDENT' : 4
                    }

def usage() :
    """ returns nagios status UNKNOWN with
        a one line usage description
        usage() calls nagios_return()
    """
    nagios_return('UNKNOWN',
                  "usage: %s -h host" % (sys.argv[0]))

def nagios_return_complex(results) :
    def deal_with_result(result) :
        print result['code'] + ": " + result['message']
        return result['code']

    # Scan all condition/status check results and create a list of appropriate exit codes.
    exit_code_keys          = map(lambda result     : deal_with_result(result), results)
    suggested_exit_codes    = list(map(lambda key   : nagios_codes[key], exit_code_keys))

    # Exit with the highest suggested exit code, because the higher the exit code the more problematic the status is and problems have priority over harmony.
    sys.exit(max(suggested_exit_codes))

def check_response_data_well_formedness(data, descriptive_string) :

    timestamp = datetime.datetime.today().isoformat()

    try :
        json.loads(data)
    except :
        traceback_string = traceback.format_exc()
        #pdb.set_trace()

        err_log_file_path = os.path.normpath("/tmp/err_unparseable." + descriptive_string.replace('/' , '__') + "_" + timestamp + ".log")

        with open(name = err_log_file_path, mode = "wt") as debugging_output_file :
            debugging_output_file.write(traceback_string)
        
        return False
    else :
        return True

def check_condition(host, UP_URL) :

    def handle_connection_failure(problem_description) :
        err_log_file_path = os.path.normpath("/tmp/err_connection_failure." + UP_URL.replace('/' , '__') + "_" + timestamp + ".log")

        with open(name = err_log_file_path, mode = "wt") as debugging_output_file :
           debugging_output_file.write(problem_description)
    
    timestamp                = datetime.datetime.today().isoformat()
    
    try :
        conn                 = httplib.HTTPConnection(host)   

        request              = conn.request("GET", UP_URL)
    except :
        traceback_string     = traceback.format_exc()

        handle_connection_failure(traceback_string + "\nThis problem originates from location 1 in 'check_clarin_discojuice_json.py'.")

        return {
                "code"      : "CRITICAL", 
                "message"   : '[%s] HTTP connection to host %s failed.' % (DESCRIPTION, host)
               }
    else :
        try :
            response         = conn.getresponse()

            data             = response.read()
            
            conn.close()
        except :
            traceback_string = traceback.format_exc()

            handle_connection_failure(traceback_string + "\nThis problem originates from location 2 in 'check_clarin_discojuice_json.py'.")
        else :
            if response.status == 200 :
                well_formed  = check_response_data_well_formedness(data, UP_URL)

                if well_formed :
                    return {
                            "code"      : "OK", 
                            "message"   : '[%s] Host %s is up and returns parseable JSON data at "%s".' % (DESCRIPTION, host, UP_URL)
                           }
                else :
                    return {
                            "code"      : "CRITICAL", 
                            "message"   : '[%s] Host %s is up but returns unparseable JSON data at "%s".' % (DESCRIPTION, host, UP_URL)
                            }
            else :

                handle_connection_failure("Unreachable URL: HTTP response code" + str(response.status) + "\nThis problem originates from location 3 in 'check_clarin_discojuice_json.py'.")
                return {
                        "code"      : "CRITICAL", 
                        "message"   : '[%s] Host %s has a problem with the URL path component "%s".' % (DESCRIPTION, host, UP_URL)
                       }

def main() :
    """ example options processing
        here we're expecting 1 option "-h"
        with a parameter
    """

    if len(sys.argv) < 2 :
        usage()

    try:
        opts, args  = getopt.getopt(sys.argv[1:], "h:")
    except getopt.GetoptError, err :
        usage()

    for o, value in opts :
        if o == "-h" :
            host    = value
        else :
            usage()

    UP_URLs          = ("/discojuice/metadata_clarin1.json", "/discojuice/metadata_clarin2.json",)
    
    # Check status for all UP_URLs.
    results          = map(lambda UP_URL : check_condition(host = host, UP_URL = UP_URL),
                           UP_URLs)
    
    nagios_return_complex(results)

if __name__ == "__main__" :
    main()