Commit ffc15fb9: added check_tachyon.py, check_tachyon_master.py, check_tachyon_worker.py

check_tachyon_master.py

+1

/check_tachyon_master.py

View

check_tachyon.py

check_tachyon_worker.py

+1

/check_tachyon_worker.py

View

check_tachyon.py

+113

/check_tachyon.py

View

#!/usr/bin/env python

# vim:ts=4:sts=4:sw=4:et

#

# Author: Hari Sekhon

# Date: 2016-02-02 17:46:18 +0000 (Tue, 02 Feb 2016)

#

# https://github.com/harisekhon/nagios-plugins

#

# License: see accompanying Hari Sekhon LICENSE file

#

# If you're using my code you're welcome to connect with me on LinkedIn and optionally send me feedback

# to help improve or steer this or other code I publish

#

# http://www.linkedin.com/in/harisekhon

#

"""

Nagios Plugin to check a Tachyon Master/Worker is online

Queries the WebUI and displays the uptime

"""

from __future__ import absolute_import

from __future__ import division

from __future__ import print_function

#from __future__ import unicode_literals

import os

import re

import sys

try:

from bs4 import BeautifulSoup

import requests

except ImportError as _:

print(_)

sys.exit(3)

libdir = os.path.abspath(os.path.join(os.path.dirname(__file__), 'pylib'))

sys.path.append(libdir)

try:

# pylint: disable=wrong-import-position

from harisekhon.utils import log, qquit, prog

from harisekhon.utils import validate_host, validate_port, isStr, space_prefix

from harisekhon import NagiosPlugin

except ImportError as _:

print('module import failed: %s' % _, file=sys.stderr)

print("Did you remember to build the project by running 'make'?", file=sys.stderr)

print("Alternatively perhaps you tried to copy this program out without it's adjacent libraries?", file=sys.stderr)

sys.exit(4)

__author__ = 'Hari Sekhon'

__version__ = '0.1'

class CheckTachyon(NagiosPlugin):

def __init__(self):

# Python 2.x

super(CheckTachyon, self).__init__()

# Python 3.x

# super().__init__()

name = ''

default_port = None

if re.search('master', prog, re.I):

name = 'Master'

default_port = 19999

elif re.search('worker|slave', prog, re.I):

name = 'Worker'

default_port = 30000

self.name = space_prefix(name)

self.default_port = default_port

def add_options(self):

self.add_hostoption(name='Tachyon%(name)s' % self.__dict__,

default_host='localhost',

default_port=self.default_port)

def run(self):

self.no_args()

host = self.get_opt('host')

port = self.get_opt('port')

validate_host(host)

validate_port(port)

log.info('querying Tachyon%(name)s' % self.__dict__)

url = 'http://%(host)s:%(port)s/home' % locals()

log.debug('GET %s' % url)

try:

req = requests.get(url)

except requests.exceptions.RequestException as _:

qquit('CRITICAL', _)

log.debug("response: %s %s" % (req.status_code, req.reason))

log.debug("content:\n{0}\n{1}\n{2}".format('='*80, req.content.strip(), '='*80))

if req.status_code != 200:

qquit('CRITICAL', "Non-200 response! %s %s" % (req.status_code, req.reason))

soup = BeautifulSoup(req.content, 'html.parser')

link = soup.find('th', text=re.compile('Uptime:?', re.I))

if link is None:

qquit('UNKNOWN', 'failed to find Tachyon%(name)s uptime' % self.__dict__)

link = link.find_next_sibling()

if link is None:

qquit('UNKNOWN', 'failed to find Tachyon%(name)s uptime (next sibling tag not found)' % self.__dict__)

uptime = link.get_text()

if not uptime or not isStr(uptime) or not re.search(r'\d+\s+second', uptime):

qquit('UNKNOWN', 'Tachyon{0} uptime format not recognized: {1}'.format(self.name, uptime))

self.msg = 'Tachyon{0} uptime: {1}'.format(self.name, uptime) # pylint: disable=attribute-defined-outside-init

self.ok()

if __name__ == '__main__':

CheckTachyon().main()



	check_tachyon_master.py +
	check_tachyon_worker.py +
	check_tachyon.py +