1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
| #! /usr/bin/env python
import os
import smtplib
from email.MIMEMultipart import MIMEMultipart
from email.MIMEText import MIMEText
from email.MIMEImage import MIMEImage
# 定义函数 _sendmail
def _sendmail(smtp_server,port,account,password,str_from,list_to,msg):
smtp = smtplib.SMTP(smtp_server,port)
smtp.ehlo()
smtp.starttls()
smtp.ehlo()
# smtp.login(account, password)
smtp.sendmail(str_from, list_to,msg)
smtp.close()
# 定义函数 _get_pictures
def _get_pictures(image_dir):
#建立空列表Pictures
pictures = []
#os.listdir 返回指定目录下的所有文件和目录名
for f in os.listdir(image_dir):
# append 是在列表最后增加内容，这里是将list出来的信息f增加到Picture列表中
pictures.append(f)
return pictures
# 定义函数 _create_msg
def _create_msg(screen_name,screens,image_dir,str_from,list_to):
msgRoot = MIMEMultipart('related')
# 定义邮件主题
msgRoot['Subject'] = 'Zabbix Screen Report: %s' % screen_name
#定义发件人、收件人
msgRoot['From'] = str_from
msgRoot['To'] = ",".join(list_to)
msgRoot.preamble = 'This is a multi-part message in MIME format.'
# Encapsulate the plain and HTML versions of the message body in an
# 'alternative' part, so message agents can decide which they want to display.
msgAlternative = MIMEMultipart('alternative')
msgRoot.attach(msgAlternative)
# 定义邮件正文内容
msgText = MIMEText('This is the alternative plain text message.')
msgAlternative.attach(msgText)
contents = ""
# tuple 元组 定义hsize,vsize变量
contents += "<h1>Screen %s</h1><br>" % screen_name
_,hsize,vsize,_,_,_,_,= tuple(screens[0].split('.'))
contents +="<table>"
# sorted()是产生一个新的列表并排序
screens = sorted(screens)
y= -1
for f in screens:
# 将文件名以为.分割点
items = f.split('.')
# 原文件名格式:$path/$name.$hsize.$vsize.$y.$x.$id.png
# 经过切片后取 image_Y,Image_x,image_ID
# tuple 元组
_,_,_,image_y,image_x,image_id,_ = tuple(items)
#定义邮件中的图片文件名：image-screen_name-image_id
image_name = "image-%s-%s" % (screen_name, image_id)
#fp = open("test.txt",w) 直接打开一个文件，如果文件不存在则创建文件
#关于open 模式：
#w 以写方式打开，
#a 以追加模式打开 (从 EOF 开始, 必要时创建新文件)
#r+ 以读写模式打开
#w+ 以读写模式打开 (参见 w )
#a+ 以读写模式打开 (参见 a )
#rb 以二进制读模式打开
#wb 以二进制写模式打开 (参见 w )
#ab 以二进制追加模式打开 (参见 a )
#rb+ 以二进制读写模式打开 (参见 r+ )
#wb+ 以二进制读写模式打开 (参见 w+ )
#ab+ 以二进制读写模式打开 (参见 a+ )
fp = open('%s/%s' % (image_dir,f), 'rb')
#fp.read([size]) #size为读取的长度，以byte为单位
msgImage = MIMEImage(fp.read())
#fp.close() #关闭文件。python会在一个文件不用后自动关闭文件，不过这一功能没有保证，最好还是养成自己关闭的习惯。
#如果一个文件在关闭后还对其进行操作会产生ValueError
fp.close()
msgImage.add_header('Content-ID', "<%s>" % image_name)
msgRoot.attach(msgImage)
#定义循环
if y != image_y:
if y!= -1:
contents +="</tr>"
y = image_y
contents +="<tr>"
# 定义td中嵌套图片文件
contents +="<td><img src='cid:%s'></td>" % image_name
# table结束
contents += "</table>"
# 定义邮件格式为HTML
msgText = MIMEText(contents, 'html')
msgAlternative.attach(msgText)
# msgRoot.attach(msgAlternative)
return msgRoot
# Create the root message and fill in the from, to, and subject headers
def main(str_from,list_to,image_dir):
pictures = _get_pictures(image_dir)
# 定义screen name,如: Citrix_APP_Server_Loading
for screen_name in list(set([x.split('.')[0] for x in pictures ])):
# startswith 判断字串开始
# screens 是一组 相同前缀的文件,如: screens ['Zabbix_server.2.2.1.1.525.png', 'Zabbix_server.2.2.1.0.524.png']
screens = [x for x in pictures if x.startswith(str(screen_name) + '.') ]
msgRoot = _create_msg(screen_name,screens,image_dir,str_from,list_to)
#定义邮件服务器地址、端口
_sendmail(smtp_server,port,'','',str_from,list_to,msgRoot.as_string())
# 定义发送邮件信息，发件人，收件人，图片目录
if __name__ == '__main__':
smtp_server = '10.210.1.16'
port = 25
str_from = 'adm_zbx@corp.com.cn'
list_to = [
"roger.ling@corp.com.cn"
]
image_dir = '/data/graph/maps'
main(str_from,list_to,image_dir)
|
