Python测试环境巡检工具

Python测试环境巡检

测试过程中,由于测试环境机器较多且全都是在虚拟机上做的,经常会有某些机器挂掉导致站点访问出错的问题,手动去一个一个Ping实在是麻烦,于是用Python写了一个小脚本,来批量Ping测试。
实现思路:

一、设计要求:

1.目标机器的IP可配置
2.可以一次性Ping多台机器
3.Ping测试的结果可查询
4.如果有Ping不通的机器,可以主动邮件通知

二、设计思路:

1.将需要测试的IP放在一个Txt文本文件中,所有目标机器的IP手动维护在这个文件中
2.将Ping测试的结果放在一个Txt文件中,包括IP,测试结果等
3.主程序要实现,Ping测试和邮件发送两个主要方法

三、代码实现:

1.新建一个Ping文件夹,该文件目录下创建一个host.txt文件,用来维护待测IP
2.新建ping.py,代码如下:
import os
import sendMail

def getIpStatus():
    hasFail = False
    try:
        Host = open("host.txt", "r")        
        Result = open("result.txt", "w")
        Result.write("测试环境巡检结果:\n")

        for line in Host.readlines():
            line = line.strip("\n")
            if "#" in line:
                Result.write(line +"\n")
            else:
                res = os.system("ping " + line)
                if res == 0:
                    print(line +" 可用...")
                    Result.write(line + " 可用...\n")
                else:
                    print(line +" 不可用...")
                    Result.write(line + " 不可用...\n")
                    hasFail = True
        Host.close()
        Result.close()
    except:
        print("Host文件读取失败...")

    if hasFail:
        print("有机器无法ping通!")
        return (True)

if __name__ == '__main__':
    if getIpStatus():
        sendMail.sendMail()
        
3.新建sendMail.py实现邮件发送功能
import smtplib
from email.mime.text import MIMEText
from email.mime.multipart import MIMEMultipart
from email.header import Header

def sendMail():
    mail_host="******"  #设置邮件服务器
    mail_user="******"    #用户名
    mail_pass="******"   #口令 

    sender = '******'   #发送邮件的邮箱
    receivers = [******]  # 接收邮件的邮箱,可设置为你的QQ邮箱或者其他邮箱,多个邮箱用,分隔开来

#创建一个带附件的实例
    message = MIMEMultipart()
    message['From'] = "******"  #邮件发送人
    message['To'] =  "******"   #邮件接收人
    subject = '测试环境巡检结果'  #邮件主题
    message['Subject'] = Header(subject, 'utf-8')   

#邮件正文内容
    message.attach(MIMEText('测试环境巡检有机器未能ping通,详细结果见附件……', 'plain', 'utf-8'))

# 构造附件1,传送附件文件
    att1 = MIMEText(open("result.txt", 'rb').read(), 'base64', 'utf-8')
    att1["Content-Type"] = 'application/octet-stream'
# 这里的filename可以任意写,写什么名字,邮件中显示什么名字
    att1["Content-Disposition"] = 'attachment; filename="Result.txt"'
    message.attach(att1)

    try:
        smtpObj = smtplib.SMTP() 
        smtpObj.connect(mail_host, 25)    # 25 为 SMTP 端口号
        smtpObj.login(mail_user,mail_pass)  
        smtpObj.sendmail(sender, receivers, message.as_string())
        print ("邮件发送成功")
    except smtplib.SMTPException:
        print ("Error: 无法发送邮件")
4.运行结果截图:
5.注意事项:
5.1Windows系统和Linux系统对于换行符的规定不一样,按照换行符进行切割的时候,需要特殊处理
5.2 Linux中中文编码需要特殊处理,具体方法是在中文字符前加“U”
5.3 os.system-ping方法返回的是Ping命令是否有运行结果,对于本来就不存在的IP,也会有返回值,所以结果可能也是通的,需要注意。如果要兼容这个问题,可以使用正则匹配命令返回内容的方法进行处理,我这里是知道具体的机器IP的,所以处理相对简单
5.4 邮件服务器的配置,可上对应邮件提供商网站查询
Done!


没有更多推荐了,返回首页