博客园 首页 私信博主 显示目录 隐藏目录 管理 动画

python3 dns轮询业务检测

python3中必须使用dnspython3和httplib2!!

直接安装dnspython3和httplib2

 

本脚本源自 刘天斯的《python 自动化运维技术与最佳实践》

该脚本功能为判断业务是否正常。过程为:首先脚本会向dns服务器提出解析请求,拿到请求返回结果后(一个域名返回多个A记录)

通过模仿浏览器访问业务ip,比对返回的html头是否与预期头相同,进而判断业务是否正常。

在文中脚本使用python2 缩写,在python3 下几乎不可运行。稍加改动脚本如下:

 

from dns import resolver#dnspython导入方法
import os
import httplib2
 
iplist=[]#定义域名ip列表变量
appdomain = 'www.baidu.com'#定义业务域名
 
def get_iplist(domain=""):  ##解析域名为函数,解析成功将追加到iplist
    try:
        A = resolver.query(domain,'A')#解析A记录类型
    except Exception as e:
        print('dns resolver error:' + str(e))
        return
    for i in A:
            iplist.append(i)#追加ip到iplist
    return True
 
def checkip(ip):#对iplist中IP进行可用检测
    checkurl = str(ip) + ":80"
    getcontent=""
    httplib2.socket.setdefaulttimeout(5)#定义http连接时间超时为5秒
    conn = httplib2.HTTPConnectionWithTimeout(checkurl)#创建http连接对象
 
    try:
        conn.request("GET","/",headers = {"HOST": appdomain}) #发起url请求,添加主机头 ##通过构造html头访问目标业务主机
        response = conn.getresponse()
        getcontent = response.read(15)#获取url前15个字符,做校验用
    finally:
        if getcontent == b"<!DOCTYPE html>" :     ##判断返回字符串是否与预期相同
#监控url一般事先定义好
print(str(ip)+'[ok]') else: print(str(ip)+'[error]')#此处可方警告、邮件、短信等 if __name__ =="__main__": if get_iplist(appdomain) and len(iplist) > 0:#域名解析正确至少返回一个ip for ip in iplist: checkip(ip) else: print('dns resolve error')

 测试:

 

posted @ 2019-08-03 16:00  【Abel】  阅读(594)  评论(0编辑  收藏  举报