check_rubrik_node_status.py

#!/usr/bin/env python

# Copyright 2018 Rubrik, Inc.
#
#  Permission is hereby granted, free of charge, to any person obtaining a copy
#  of this software and associated documentation files (the "Software"), to
#  deal in the Software without restriction, including without limitation the
#  rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
#  sell copies of the Software, and to permit persons to whom the Software is
#  furnished to do so, subject to the following conditions:
#
#  The above copyright notice and this permission notice shall be included in
#  all copies or substantial portions of the Software.
#
#  THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
#  IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
#  FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
#  AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
#  LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
#  FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
#  DEALINGS IN THE SOFTWARE.

import argparse
import logging
import urllib3

import nagiosplugin
import rubrik_cdm


_log = logging.getLogger('nagiosplugin')


class RubrikNodeStatus(nagiosplugin.Resource):
    """Rubrik CDM Node Status
    Returns the status of all nodes in the Rubrik CDM cluster
    """

    def __init__(self, rubrik_ip, rubrik_user, rubrik_pass):
        self.rubrik_ip = rubrik_ip
        self.rubrik_user = rubrik_user
        self.rubrik_pass = rubrik_pass

    @property
    def name(self):
        return 'RUBRIK_NODE_STATUS'

    def get_nodes(self):
        """Gets node statuses from Rubrik CDM"""

        urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)

        rk = rubrik_cdm.Connect(node_ip=self.rubrik_ip, username=self.rubrik_user, password=self.rubrik_pass)
        nodes = rk.get('internal', '/node')

        return nodes

    def probe(self):
        nodes = self.get_nodes()

        bad_nodes = 0
        for node in nodes['data']:
            if node['status'] != 'OK':
                bad_nodes += 1
        
        if not bad_nodes:
            _log.debug('All {0} nodes are in an OK status'.format(nodes['total']))
        else:
            _log.debug('{0} nodes returned an unhealthy status'.format(bad_nodes))
        
        metric = nagiosplugin.Metric('Unhealthy nodes', bad_nodes, min=0, context='rk_bad_nodes')

        return metric


def parse_args():
    argp = argparse.ArgumentParser()
    
    argp.add_argument('-s', '--rubrik_ip', help="Rubrik hostname or ip address")
    argp.add_argument('-u', '--rubrik_user', help="Rubrik username")
    argp.add_argument('-p', '--rubrik_pass', help="Rubrik password")
    argp.add_argument('-w', '--warning', metavar='RANGE', default='~:0', help='return warning if occupancy is outside RANGE. Value is expressed in number of unhealthy nodes')
    argp.add_argument('-c', '--critical', metavar='RANGE', default='~:0', help='return critical if occupancy is outside RANGE. Value is expressed in number of unhealthy nodes')
    argp.add_argument('-v', '--verbose', action='count', default=0, help='increase output verbosity (use up to 3 times)')
    argp.add_argument('-t', '--timeout', default=30, help='abort execution after TIMEOUT seconds')
    
    return argp.parse_args()


@nagiosplugin.guarded
def main():
    args = parse_args()
    check = nagiosplugin.Check(RubrikNodeStatus(args.rubrik_ip, args.rubrik_user, args.rubrik_pass))
    check.add(nagiosplugin.ScalarContext('rk_bad_nodes', args.warning, args.critical))
    check.main(args.verbose, args.timeout)


if __name__ == '__main__':
    main()