将邮件内容转为企业微信预警通知脚本

 #!/usr/bin/env python
# _*_ encoding:utf-8 _*_
import poplib
import os, subprocess, sys, time
import email
from email.parser import Parser
from email.parser import Parser
from email.header import decode_header
from email.utils import parseaddr
import wechat_work_webhook
#indent用于演示缩进
def print_info(msg, indent=0):
    """
    打印邮件内容信息
    :param msg:
    :param indent:
    :return:
    """
    if indent == 0:
        # 邮件的From, To, Subject存在于根对象上:
        for header in ['From', 'To', 'Subject']:
            value = msg.get(header, '')
            if value:
                if header == 'Subject':
                    # 需要解码Subject字符串:
                    value = decode_str(value)
                else:
                    # 需要解码Email地址:
                    hdr, addr = parseaddr(value)
                    name = decode_str(hdr)
                    value = u'%s <%s>' % (name, addr)
                    # print('%s%s: %s' % (' ' * indent, header, value))
        value = msg.get("From", '')
        hdr, addr = parseaddr(value)
        name = decode_str(addr)
        value = msg.get(header, '')
        subject = decode_str(value)
        content_type = msg.get_content_type() # 获取内容类型
        if content_type == 'text/plain' or content_type == 'text/html':
            content = msg.get_payload(decode=True) # 提取内容
            charset = guess_charset(msg) # 提取字符集
        if charset:
            content = content.decode(charset) # 对内容进行解码
        #过滤指定内容的邮件告警
        if name == "svosalram@baiwutong.com":
            #print("发件人：" + name,)
            #print(" 邮件主题：" + subject,)
            #print(" 邮件主题：" + content)
            wechat = wechat_work_webhook.connect("机器人地址")
            #wechat.markdown(content)
            content = content.replace("<br/>","\n\n")
            wechat.markdown(content)
    if (msg.is_multipart()):
        # 如果邮件对象是一个MIMEMultipart,
        # get_payload()返回list，包含所有的子对象:
        parts = msg.get_payload()
        for n, part in enumerate(parts):
            # print('%spart %s' % (' ' * indent, n))
            # print('%s--------------------' % (' ' * indent))
            # 递归打印每一个子对象:
            print_info(part, indent + 1)

    else:
        # 邮件对象不是一个MIMEMultipart,
        # 就根据content_type判断:
        content_type = msg.get_content_type()
        if content_type == 'text/plain' or content_type == 'text/html':
            # 纯文本或HTML内容:
            content = msg.get_payload(decode=True)
            # 要检测文本编码:
            charset = guess_charset(msg)
            if charset == "utf-8":
                pass
            elif charset == "gbk":
                content = content.decode(charset)

        else:
            # 不是文本,作为附件处理:
            #print('%sAttachment: %s' % (' ' * indent, content_type))
            pass


def decode_str(s):
    value, charset = decode_header(s)[0]
    if charset:
        value = value.decode(charset)
    return value


def guess_charset(msg):
    # 先从msg对象获取编码:
    charset = msg.get_charset()
    if charset is None:
        # 如果获取不到，再从Content-Type字段获取:
        content_type = msg.get('Content-Type', '').lower()
        pos = content_type.find('charset=')
        if pos >= 0:
            charset = content_type[pos + 8:].strip()
    return charset

# 邮箱登录信息
emailServer = poplib.POP3('ip:port')
emailServer.user('username')
emailServer.pass_('password')

# 设置为1，可查看向pop3服务器提交了什么命令
emailServer.set_debuglevel(0)

# 获取欢迎消息
serverWelcome = emailServer.getwelcome()
# print(serverWelcome)
# 获取一些统计信息：邮件数量，邮箱大小
emailMsgNum, emailSize = emailServer.stat()
# list()返回所有邮件的编号：查看返回的列表类似['1 82923', '2 2184', ...]
resp, mails, octets = emailServer.list()
# 存放上一次邮箱数量文本文件路径
#email_num_path = r'C:\Users\cai\Desktop\email_num_path.txt'
# 存放上一次邮箱数量文本文件路径
email_num_path = r'/hskj/script/email_num_path.txt'
#获取文本中存储的邮件数量
with open(email_num_path, 'r') as f:
    old_email_numbers = int(f.read())

cur_email_numbers = emailMsgNum

# 当邮件数量大于上一次脚本执行时邮件数量时，执行循环
if cur_email_numbers > old_email_numbers:
    pass
# 当邮件数量等于上一次脚本执行时邮件数量时，退出程序
elif cur_email_numbers == old_email_numbers:
    print("no-new-email")
    sys.exit()
elif cur_email_numbers < old_email_numbers:
    # 将最新的邮件数量写入到文本中，并退出程序
    print("someEmailDeleted")
    with open(email_num_path, 'w') as f:
        f.write(str(cur_email_numbers))
    sys.exit()


for index in range(cur_email_numbers, old_email_numbers, -1):
    resp, lines, octet = emailServer.retr(index)
    #print(mails)
    # lines存储了邮件的原始文本的每一行,可以获得整个邮件的原始文本:
    def decode_function(x):
        try:
            x = x.decode()
            return x
        except:
            x = x.decode(encoding='gbk')
            return x

    msg_content = '\r\n'.join(map(decode_function, lines))
    # 稍后解析出邮件:
    msg = Parser().parsestr(msg_content)
    #打印邮件内容
    try:
        print_info(msg)
    except:
        print("pythonErrorPrintInfoFunction")
        continue
# 退出邮箱服务器
emailServer.quit()
# 将最新的邮件数量写入到文本中
with open(email_num_path, 'w') as f:
    f.write(str(cur_email_numbers))

该脚本可用于部分厂商服务只支持邮件预警的情况。