check_hadoop_tasktracker.py

'''
Created on 2011-11-25

@author: yue.zhang
'''
#!/usr/bin/env python

import commands
import sys
from optparse import OptionParser
import urllib
import re

def get_value():
    urlItem = urllib.urlopen("http://182.61.128.18:50030/jobtracker.jsp")
    html = urlItem.read()
    urlItem.close()
    return float(re.findall('.+<a href="machines.jsp\?type=active">(\d+)</a>.+', html)[0])

if __name__ == '__main__':

    parser = OptionParser(usage="%prog [-w] [-c]", version="%prog 1.0")
    parser.add_option("-w", "--warning", type="int", dest="w", default=18)
    parser.add_option("-c", "--critical", type="int", dest="c", default=15)
    (options, args) = parser.parse_args()

    if(options.c >= options.w):
        print '-w must greater then -c'
        sys.exit(1)

    value = get_value()

    if(value <= options.c ) :
	print 'CRITICAL - Live Tasktracker %d' %(value)
        sys.exit(2)
    elif(value <= options.w):
        print 'WARNING - Live Tasktracker %d' %(value)
	sys.exit(1)
    else:   
        print 'OK - Live Tasktracker %d' %(value) 
	sys.exit(0)