check_hadoop_datanode.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:50070/dfshealth.jsp")
html = urlItem.read()
urlItem.close()
return float(re.findall('.+Live Nodes</a> <td id="col2"> :<td id="col3">\\s+(\d+)<tr class="rowAlt">.+', 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 Nodes %d' %(value)
sys.exit(2)
elif(value <= options.w):
print 'WARNING - Live Nodes %d' %(value)
sys.exit(1)
else:
print 'OK - Live Nodes %d' %(value)
sys.exit(0)