Python 3 对DNS负载均衡环境下的WEB服务器的监控脚本

cuiyadll

2011-08-01

关注关注

很多大型网站为了减轻压力，避免单点故障，使用了多个服务器（IP）地址来为单一网站提供服务。

其中包括使用DNS轮训来实现负载。

但是对于管理员而言，如何可以监控这些服务器那些是ok那些是有问题的呢？

下面脚本可以帮你实现。

import socket,urllib.parse,configparser,re,sys,http.client,time
#coding:utf-8
#域名转换IP地址函数
def dom2ip(dname):
iplist=socket.gethostbyname_ex(dname)[2]
return iplist
#字典数据生成函数
def parse(url,key):
zhlist[key]={}
zhlist[key]['host']=a.hostname
zhlist[key]['path']=a.path
zhlist[key]['port']=a.port
zhlist[key]['query']=a.query
zhlist[key]['url']=a.geturl()
try:
zhlist[key]['ips']=dom2ip(a.hostname)
except Exception as err:
print('你的配置文件格式有错误,\n错误信息：',err,'\n程序已退出')
sys.exit(1)
#headers
headers={
'User-Agent':'Mozilla/5.0 (Windows NT 5.1) AppleWebKit/535.1 (KHTML, like Gecko)'
}
def getinfo(domain,ip,path='/',port=80):
headers['Host']=domain
h=http.client.HTTPConnection(ip,port,timeout=8)
h.request('GET',path,headers=headers)
r=h.getresponse()
print(r.getheaders())
print('服务器返回状态',r.status)
print('服务器返回结果',r.reason)
print('目标时间:',r.getheader('Date'))
print('目标Web端:',r.getheader('Server'))
print('目标网页上次修改时间:',r.getheader('Last-Modified'))
print('目标内容返回:',r.getheader('Content-Length'),'字节')
print('目标返回编码:',r.getheader('Content-Encoding'))
try:
#分析网页的字符集
a=r.getheader('Content-Type')
b=re.findall('charset=(.*)$',a,re.I)
for i in b:
charset=i
try:
charset
res=res.decode(charset,'ignore')
except:
res=r.read(2048)
strset=re.findall(b'charset=(.*?)"',res,re.I)
for i in strset:
charset=i.decode()
try:
charset
res=res.decode(charset,'ignore')
except:
strset=re.findall(b'charset="(.*?)',res,re.I)
for i in strset:
charset=i.decode()
try:
charset
res=res.decode(charset,'ignore')
except:
charset='utf8'#设置默认charset
res=res.decode(charset,'ignore')
print('目标脚本编码:',charset)
#显示Title
data=re.findall('<title>(.*)</title>',res,re.I)
for i in data:
print ('目标返回Title为',':',i.strip())
try:
i
except:
print ('目标网页有问题，未返回Title')
print('目标错误原因：',res)
#判断不同情况的返回
if r.status==302:
Location=r.getheader('Location')
print('注意：目标网页有转向,请修改配置文件从新定义监控页面,转向地址:',Location)
elif r.status==404:
print('注意：目标网页被删除,请修改配置文件从新定义监控页面')
else:
if r.status !=200:
print('网站其他问题，参考返回状态，适当修改配置文件后尝试')
except Exception as err:
print ('连接错误，详细信息：',err)
#创建配置实例
ini=configparser.ConfigParser()
#读配置文件
ini.read(sys.path[0]+'\yatere.ini')
#监控主机列表
host=ini.get('webserver','servers').split(';')
#删除对象
del ini
#对应列表
zhlist={}
#填充字典数据
print ('\n你设置的监控网址有',len(host),'个,分别是:\n')
for i in host:
a=urllib.parse.urlparse(i)
parse(i,a.hostname)
print (a.geturl())
print('\n','--'*20)
for i in zhlist:
print (i,'有',len(zhlist[i]['ips']),'个IP地址:')
for l in zhlist[i]['ips']:
print (l,end='\t')
print ('\n')
a=input('按回车键继续...')
#删除对象
del a,i,l,host
while True:
for domk in zhlist:
c=0
print('现在是',domk,'域的信息','---'*8,'\n\n')
for ipadd in zhlist[domk]['ips']:
c=c+1
print (domk,'的第',c,'台IP为',ipadd,'的服务器信息如下:')
getinfo(domk,ipadd,zhlist[domk]['path']+'?'+zhlist[domk]['query'])
print('\r')
time.sleep(5)
del domk,ipadd,c