+#! /usr/bin/env python
+
+###############################################################################
+# Nagios plugin template
+#
+# Notes
+# - The RHEL boxes I work on are currently limited to Python 2.6.6, hence the
+# use of (deprecated) optparse. If I can ever get them all updated to
+# Python 2.7 (or better yet, 3.3), I'll switch to argparse
+# - This template runs in 2.6-3.3. Any changes made will need to be appropriate
+# to the Python distro you want to use
+#
+###############################################################################
+
+__author__ = 'mr.it@cbs.dk - Michael Rasmussen'
+__version__= 0.1
+
+try: # RHEL8 RHEL9
+ from optparse import OptionParser, OptionGroup #
+ import logging as log
+ import sys
+ import urllib3 as urllib # python3-urllib3-1.24.2-5.el8.noarch.rpm python3-urllib3-1.26.5-3.el9.noarch.rpm
+ import json
+except:
+ sys.exit(2)
+
+## These will override any args passed to the script normally. Comment out after testing.
+#testargs = '--help'
+#testargs = '--version'
+#testargs = '-vvv'
+testing = True
+
+OK_RESPONSE = """
+{
+ "status": "UP",
+ "components": {
+ "db": {
+ "status": "UP",
+ "details": {
+ "database": "PostgreSQL",
+ "validationQuery": "isValid()"
+ }
+ },
+ "diskSpace": {
+ "status": "UP",
+ "details": {
+ "total": 2013582688256,
+ "free": 1635574571008,
+ "threshold": 10485760,
+ "path": "/home/mir/git/soasi-course-catalog-course-target-adapter/.",
+ "exists": true
+ }
+ },
+ "ping": {
+ "status": "UP"
+ }
+ }
+}"""
+DOWN_RESPONSE = """
+{
+ "status": "DOWN",
+ "components": {
+ "db": {
+ "status": "DOWN",
+ "components": {
+ "adapterDataSource": {
+ "status": "DOWN",
+ "details": {
+ "error": "org.springframework.jdbc.CannotGetJdbcConnectionException: Failed to obtain JDBC Connection"
+ }
+ },
+ "es3DataSource": {
+ "status": "UP",
+ "details": {
+ "database": "Microsoft SQL Server",
+ "validationQuery": "isValid()"
+ }
+ },
+ "providersDataSource": {
+ "status": "DOWN",
+ "details": {
+ "error": "org.springframework.jdbc.CannotGetJdbcConnectionException: Failed to obtain JDBC Connection"
+ }
+ }
+ }
+ },
+ "diskSpace": {
+ "status": "UP",
+ "details": {
+ "total": 1013309239296,
+ "free": 882010726400,
+ "threshold": 10485760,
+ "path": "C:\\Users\\hf.it\\Projects\\smart-integrations\\adapters\\target\\soasi-invigilation-report-target-adapter",
+ "exists": true
+ }
+ },
+ "ping": {
+ "status": "UP"
+ }
+ }
+}
+"""
+def main():
+ """ Main plugin logic goes here """
+
+ ## Parse command-line arguments
+ args, args2 = parse_args()
+
+ ## Uncomment to test logging levels against verbosity settings
+ # log.debug('debug message')
+ # log.info('info message')
+ # log.warning('warning message')
+ # log.error('error message')
+ # log.critical('critical message')
+ # log.fatal('fatal message')
+ options = vars(args)
+ keyword = options['keyword']
+ url = options['url']
+ agent = options['agent']
+ if keyword is None or url is None:
+ message = "Keywork: {0} url: {1}".format(keyword, url)
+ status = 3
+ log.fatal(message)
+ else:
+ try:
+ if not testing:
+ req_headers = {
+ 'User-Agent': agent
+ }
+ http = urllib.PoolManager()
+ response = http.request(
+ 'GET', url, headers = req_headers
+ )
+ data = response.data.decode('utf-8').replace("\\"," ")
+ else:
+ data = DOWN_RESPONSE.replace("\\"," ")
+ data = json.loads(data)
+ if 'status' in data and data['status'] == keyword:
+ message = "UP"
+ status = 2
+ else:
+ message = gather_message(data)
+ status = 0
+ except Exception as e:
+ print(e)
+ message = "DOWN"
+ status = 2
+
+ gtfo(status, message)
+
+def gather_message(json_data):
+ """ Assemble error messages """
+ msg = None
+ try:
+ if 'components' in json_data:
+ components = json_data['components']
+ for component in components:
+ items = components[component]
+ if 'components' in items:
+ for item in items['components']:
+ if 'status' in items['components'][item] and items['components'][item]['status'].upper() == 'DOWN':
+ if msg is not None:
+ msg += "\n{0}: {1}".format(item, items['components'][item]['details']['error'])
+ else:
+ msg = "{0}: {1}".format(item, items['components'][item]['details']['error'])
+ else:
+ if 'status' in items and items['status'].upper() == 'DOWN':
+ if msg is not None:
+ if 'details' in items and 'error' in items['details']:
+ error = items['details']['error']
+ else:
+ error = "No error message"
+ if msg is not None:
+ msg += "\n{0}: {1}".format(component, error)
+ else:
+ msg = "{0}: {1}".format(component, error)
+ else:
+ sys.exit(3)
+ except:
+ sys.exit(3)
+ return msg
+
+
+def parse_args():
+ """ Parse command-line arguments """
+
+ parser = OptionParser(usage='usage: %prog [-v|vv|vvv] [options]',
+ version='{0}: v.{1} by {2}'.format('%prog', __version__, __author__))
+
+ ## Verbosity (want this first, so it's right after --help and --version)
+ parser.add_option('-v', help='Set verbosity level',
+ action='count', default=0, dest='v')
+
+ ## CLI arguments specific to this script
+ group = OptionGroup(parser,'Plugin Options')
+ group.add_option('-a', '--agent', help="User agent for request. Default: Python-nagios",
+ default="Python-nagios", type='string')
+ group.add_option('-k', '--keyword', help="Keyword to search for in response", default=None)
+ group.add_option('-t', '--testing', help="Run in testing mode", default=False, action='store_true')
+ group.add_option('-u', '--url', help="URL to requested resource", default=None)
+
+ ## Common CLI arguments
+ #parser.add_option('-c', '--critical', help='Set the critical threshold. Default: %(default)s',
+ # default=97, type=float, dest='crit', metavar='##')
+ #parser.add_option('-w', '--warning', help='Set the warning threshold. Default: %(default)s',
+ # default=95, type=float, dest='warn', metavar='##')
+
+ parser.add_option_group(group)
+
+ ## Try to parse based on the testargs variable. If it doesn't exist, use args
+ try:
+ args, args2 = parser.parse_args(testargs.split())
+ except NameError:
+ args, args2 = parser.parse_args()
+
+ ## Set the logging level based on the -v arg
+ log.getLogger().setLevel([log.ERROR, log.WARN, log.INFO, log.DEBUG][args.v])
+
+ log.debug('Parsed arguments: {0}'.format(args))
+ log.debug('Other arguments: {0}'.format(args2))
+
+ return args, args2
+
+def gtfo(exitcode, message=''):
+ """ Exit gracefully with exitcode and (optional) message """
+
+ log.debug('Exiting with status {0}. Message: {1}'.format(exitcode, message))
+
+ if message:
+ print(message)
+ exit(exitcode)
+
+if __name__ == '__main__':
+ ## Initialize logging before hitting main, in case we need extra debuggability
+ log.basicConfig(level=log.DEBUG, format='%(asctime)s - %(funcName)s - %(levelname)s - %(message)s')
+ main()
projects / check_http.git / commitdiff