利用python批量检查网站的可用性
前言
随着站点的增多，管理复杂性也上来了，俗话说：人多了不好带，我发现站点多了也不好管，因为这些站点里有重要的也有不重要的，重要核心的站点当然就管理的多一些，像一些万年都不出一次问题的，慢慢就被自己都淡忘了，冷不丁那天出个问题，还的手忙脚乱的去紧急处理，所以规范的去管理这些站点是很有必要的，今天我们就做第一步，不管大站小站，先统一把监控做起来，先不说业务情况，最起码那个站点不能访问了，要第一时间报出来，别等着业务方给你反馈，就显得我们不够专业了，那接下来我们看看如果用python实现多网站的可用性监控，脚本如下：
#!/usr/bin/envpython
importpickle,os,sys,logging
fromhttplibimportHTTPConnection,socket
fromsmtplibimportSMTP
defemail_alert(message,status):
fromaddr='xxx@163.com'
toaddrs='xxxx@qq.com'
server=SMTP('smtp.163.com:25')
server.starttls()
server.login('xxxxx','xxxx')
server.sendmail(fromaddr,toaddrs,'Subject:%s\r\n%s'%(status,message))
server.quit()
defget_site_status(url):
response=get_response(url)
try:
ifgetattr(response,'status')==200:
return'up'
exceptAttributeError:
pass
return'down'
defget_response(url):
try:
conn=HTTPConnection(url)
conn.request('HEAD','/')
returnconn.getresponse()
exceptsocket.error:
returnNone
except:
logging.error('BadURL:',url)
exit(1)
defget_headers(url):
response=get_response(url)
try:
returngetattr(response,'getheaders')()
exceptAttributeError:
return'Headersunavailable'
defcompare_site_status(prev_results):
defis_status_changed(url):
status=get_site_status(url)
friendly_status='%sis%s'%(url,status)
printfriendly_status
ifurlinprev_resultsandprev_results[url]!=status:
logging.warning(status)
email_alert(str(get_headers(url)),friendly_status)
prev_results[url]=status
returnis_status_changed
defis_internet_reachable():
ifget_site_status('www.baidu.com')=='down'andget_site_status('www.sohu.com')=='down':
returnFalse
returnTrue
defload_old_results(file_path):
pickledata={}
ifos.path.isfile(file_path):
picklefile=open(file_path,'rb')
pickledata=pickle.load(picklefile)
picklefile.close()
returnpickledata
defstore_results(file_path,data):
output=open(file_path,'wb')
pickle.dump(data,output)
output.close()
defmain(urls):
logging.basicConfig(level=logging.WARNING,filename='checksites.log',
format='%(asctime)s%(levelname)s:%(message)s',
datefmt='%Y-%m-%d%H:%M:%S')
pickle_file='data.pkl'
pickledata=load_old_results(pickle_file)
printpickledata
ifis_internet_reachable():
status_checker=compare_site_status(pickledata)
map(status_checker,urls)
else:
logging.error('Eithertheworldendedorwearenotconnectedtothenet.')
store_results(pickle_file,pickledata)
if__name__=='__main__':
main(sys.argv[1:])
脚本核心点解释：
1、getattr（）是python的内置函数，接收一个对象，可以根据对象属性返回对象的值。
2、compare_site_status（）函数是返回的是一个内部定义的函数。
3、map()，需要2个参数，一个是函数，一个是序列，功能就是将序列中的每个元素应用函数方法。
总结
以上就是这篇文章的全部内容，有需要的朋友们可以参考借鉴。