#!/usr/bin/python
#coding: utf-8
#返回0:健康检查OK,权重保持
#返回1:健康检查失败,权重设为0
#返回2-255:健康检查OK,权重设置为:退出状态码-2
import commands
import sys, getopt
import os.path
def check_dns(server='', timeout='', name='', address=''):
status, output = commands.getstatusoutput("/usr/lib64/nagios/plugins/check_dns -s " + server + " -t " + timeout + " -H " + name + " -a " + address)
return status
if __name__ == '__main__':
server, timeout, name, address = '', '', '', ''
script = os.path.split(os.path.realpath(__file__))[1]
try:
opts, args = getopt.getopt(sys.argv[1:], "s:n:a:t:", ["server=", "name=", "address=", "timeout="])
except getopt.GetoptError, err:
#print str(err)
sys.exit(3)
for o, a in opts:
if o in ("-s", "--server"):
server = a
elif o in ("-n", "--name"):
name = a
elif o in ("-a", "--address"):
address = a
elif o in ("-t", "--timeout"):
timeout = a
else:
#print "usage: " + script + " -s server -n name -a address -t timeout"
sys.exit(3)
if server == '' or name == '' or address == '' or timeout == '':
#print "usage: " + script + " -s server -n name -a address -t timeout"
sys.exit(3)
else:
status = check_dns(server, timeout, name, address)
if status:
#print status
sys.exit(1)
else:
#print status
sys.exit(0)