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

Python3 实现简易ping监控并发动报警邮件
1、实现原理通过ping命令结果获取丢包率和延时情况，当丢包率和延时达到预设值时，将结果作为邮件内容，并发送邮件
2、邮件是用smtplib和email实现发送的
3、目前脚本有个小问题就是：我打开文件，循环读，当读第一行时，我执行ping，获取结果，把结果写到一个文件里，同时判断是否达到预设值，如果达到，就把结果加到邮件内容，就这样一直循环，当把文件读完，然后关闭这两个文件，再发送邮件。总觉得有点问题，应该将结果存到一个列表，一次性写入。算了就先这样玩吧。

代码如下：

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

__author__   = 'babyshen'
__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]
threshold = line.split()[3] if len(line.split()) == 4 else 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.baidu.com',
'user':'xxxx',
'passwd':xxoo''}
fro = 'xxxxx'
to = ['ooooo']
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
升级版本已实现命令行参数和守护进程模式
github地址：https://github.com/babyshen/Python/blob/master/monitor_2.py