码迷,mamicode.com
首页 > 编程语言 > 详细

python3 dns轮询业务检测

时间:2019-08-03 16:14:27      阅读:146      评论:0      收藏:0      [点我收藏+]

标签:cti   checkurl   类型   import   图片   ons   ima   socket   安装   

python3中必须使用dnspython3和httplib2!!

直接安装dnspython3和httplib2

技术图片

 

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

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

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

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

 

form 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):
    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)

 测试:

技术图片

 

python3 dns轮询业务检测

标签:cti   checkurl   类型   import   图片   ons   ima   socket   安装   

原文地址:https://www.cnblogs.com/abels0025/p/11295124.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!