check_by_saltpeer.py

#!/usr/bin/env python
# -*- coding: UTF-8 -*-
#
# NOTE!!! For this to work, you will have to set up peers on your salt-master
# http://docs.saltstack.com/ref/peer.html
#
# Copyright (C) 2014 Morten Bekkelund
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# See: http://www.gnu.org/copyleft/gpl.html
#
# The intention of this script is to replace NRPE, ssh etc for remote execution
# of nagiosplugins. The script runs as root on the salt-master.
#

import sys
import argparse
import salt.client

# exitcodes for nagios .. pretty much just using unknown for now, but still :P
NAGIOS_EXIT = {"OK": 0, "WARNING": 1, "CRITICAL": 2, "UNKNOWN": 3}

client = salt.client.Caller()

def execute_plugin(host,plugin,timeout):
    """
        executing nagiosplugin via salt's cmd.run_all function
    """

    ret    = client.sminion.functions['publish.publish'](host,'cmd.run_all', plugin)
    exit   = ret[host]["retcode"]
    output = ret[host]["stdout"]

    return exit, output

if __name__ == '__main__':
    parser = argparse.ArgumentParser(description='Options:')
    parser.add_argument('-H', metavar='host', type=str, help='remote host to be queried', required=True)
    parser.add_argument('-p', metavar='plugin', type=str, help='full path to remote nagiosplugin', required=True)
    parser.add_argument('-t', metavar='timeout', type=str, help='timout in seconds', required=False)

    try:
        arguments = parser.parse_args()
    except Exception as e:
        print "Error in argumentparsing: {0}".format(e)
        sys.exit(NAGIOS_EXIT["UNKNOWN"])

    (exit, output) = execute_plugin(arguments.H,arguments.p,arguments.t)
    if not output:
        print "Remote execution of {0} unsuccessful.".format(arguments.p)
        print "Check if the plugin exists on the remote host and that the permissions " \
                "are correct."

print(output)
sys.exit(exit)