gistfile1.py

import collections
import sys
import traceback

S_OK = 0
S_UP = 0
S_WARNING = S_WARN = 1
S_CRITICAL = S_CRIT = 2
S_DOWN = 2
S_UNKNOWN = 3


ExitCode = collections.namedtuple('ExitCode', ['code_name', 'code_value'])


OK = ExitCode('OK', S_OK)
WARNING = ExitCode('WARNING', S_WARNING)
CRITICAL = ExitCode('CRITICAL', S_CRITICAL)
UNKNOWN = ExitCode('UNKNOWN', S_UNKNOWN)


def print_and_exit(exit_code, message, exc_info=False):
    print '%s - %s' % (exit_code.code_name, message)
    if exc_info:
        traceback.print_exc()
    sys.exit(exit_code.code_value)


def safe_nagios_wrapper(func):
    """
    A decorator useful for wrapping the 'main' part of a python based nagios
    monitor.  This allows it to catch unhandled exceptions and print them in
    a nagios friendly format.
    """
    def new_function(*args, **kwargs):
        try:
            func(*args, **kwargs)
        except SystemExit:
            raise
        except Exception, e:
            print_and_exit(
                UNKNOWN, 'Unknown exception: %s' % e.message, exc_info=True)
    return new_function