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

Python3 实现简易ping监控并发送报警邮件

时间:2016-12-31 07:42:20      阅读:210      评论:0      收藏:0      [点我收藏+]

标签:监控   邮件   python   

Python3 实现简易ping监控并发动报警邮件

1、实现原理通过ping命令结果获取丢包率和延时情况,当丢包率和延时达到预设值时,将结果作为邮件内容,并发送邮件

2、邮件是用smtplib和email实现发送的

3、目前脚本有个小问题就是:我打开文件,循环读,当读第一行时,我执行ping,获取结果,把结果写到一个文件里,同时判断是否达到预设值,如果达到,就把结果加到邮件内容,就这样一直循环,当把文件读完,然后关闭这两个文件,再发送邮件。总学得有点问题,应该将结果存到一个列表,一次性写入。算了就先这样玩吧。


代码如下:

# -*- coding:utf-8 -*-

__author__   = ‘GuoZhaoHui‘
__version__  = ‘0.0.3‘
__datetime__ = ‘2016-12-30 03:30:00‘

import os
import subprocess
import smtplib
import sys
from email.mime.multipart import MIMEMultipart
from email.mime.base import MIMEBase
from email.mime.text import MIMEText
from email.utils import COMMASPACE,formatdate
from email import encoders

# server[‘name‘], server[‘user‘], server[‘passwd‘]
def send_mail(server, fro, to, subject="", text="", files=[]):
    assert type(server) == dict
    assert type(to) == list
    assert type(files) == list

    msg = MIMEMultipart()
    msg[‘From‘] = fro                # 邮件的发件人
    msg[‘Subject‘] = subject         # 邮件的主题
    msg[‘To‘] = COMMASPACE.join(to)  # COMMASPACE==‘, ‘ 收件人可以是多个,to是一个列表
    msg[‘Date‘] = formatdate(localtime=True) # 发送时间
    msg.attach(MIMEText(text,‘plain‘,‘utf-8‘))

    for file in files:          # 添加附件可以是多个,files是一个列表,可以为空
        part = MIMEBase(‘application‘, ‘octet-stream‘) #‘octet-stream‘: binary data
        with open(file,‘rb‘) as f:
            part.set_payload(f.read())
        encoders.encode_base64(part)
        part.add_header(‘Content-Disposition‘, ‘attachment; filename="%s"‘ % os.path.basename(file))
        msg.attach(part)

    smtp = smtplib.SMTP()
    smtp.connect(server[‘name‘]) # connect有两个参数,第一个为邮件服务器,第二个为端口,默认是25
    smtp.login(server[‘user‘], server[‘passwd‘]) # 用户名,密码
    smtp.sendmail(fro, to, msg.as_string()) # 发件人,收件人,发送信息
    smtp.close()  # 关闭连接

def ping_test(src,dest):
    ping = subprocess.Popen(‘/bin/ping -i 0.2 -c 4 -q -I ‘ + src + ‘ ‘ + dest,
                            shell=True,
                            stderr=subprocess.PIPE,
                            stdout=subprocess.PIPE) # 执行命令
    res,err = ping.communicate()
    if err: sys.exit(err.decode().strip(‘\n‘))
    pres = list(res.decode().split(‘\n‘))
    loss = pres[3].split()[5]  # 获取丢包率
    try:
        rtt = pres[4].split(‘/‘)[4] # 获取rtt avg值
    except IndexError:
        rtt = ""
    return loss,rtt

def ping_res(file,revfile):
    with open(file, ‘r‘, encoding=‘utf-8‘) as f:
        with open(revfile, ‘w‘, encoding=‘utf-8‘) as f1:
            textmail = text = ‘%-15s %-15s %-15s %-5s %-5s \n‘ % (‘host‘,‘src‘,‘dest‘,‘loss‘,‘rtt‘)
            f1.write(text)
            for line in f:
                host = line.split()[0]
                src = line.split()[1]
                dest = line.split()[2]
                try:
                    threshold = line.split()[3]
                except IndexError: # 默认延时报警阀值
                    threshold = 100
                loss, rtt = ping_test(src, dest)
                text = ‘%-15s %-15s %-15s %-5s %-5s \n‘ % (host, src, dest, loss, rtt)
                f1.write(text)
                # 判断丢包率和延时是不是达到预设值
                if float(loss.strip(‘%‘)) > 10 or float(rtt) > float(threshold):
                    textmail += text
    return textmail

if __name__ == ‘__main__‘:
    try:
        file = ‘/root/ping_test/ping_test.cfg‘ # 配置文件,格式(空格隔开):主机名,源地址,目标地址,延时阀值(默认是100)
        revfile = ‘/root/ping_test/ping_test.rev‘ # 结果文件
        textmail = ping_res(file,revfile) # 邮件内容
        # print(textmail)
        server = {‘name‘:‘mail.netpas.co‘,
                  ‘user‘:‘guozhaohui‘,
                  ‘passwd‘:‘‘}
        fro = ‘guozhaohui@netpas.co‘
        to = [‘455416841@qq.com‘]
        subject = ‘ping_test‘
        if len(textmail.split(‘\n‘)) == 2 : sys.exit() # 当只有一行时退出程序不发送邮件
        send_mail(server,fro,to,subject,textmail)
    except smtplib.SMTPAuthenticationError:
        sys.exit("Mail Authentication Failed")
    except KeyboardInterrupt:
        sys.exit(249)
    except Exception as e:
        sys.exit(e)

github地址:https://github.com/babyshen/Python/blob/master/monitor.py

本文出自 “baby神” 博客,请务必保留此出处http://babyshen.blog.51cto.com/8405584/1887791

Python3 实现简易ping监控并发送报警邮件

标签:监控   邮件   python   

原文地址:http://babyshen.blog.51cto.com/8405584/1887791

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