Python邮件处理(收取、解码及附件下载)

之前在工作中需要用到自动查收邮件和下载附件，于是用python做了一些自动化的尝试。以下代码主要实现了两个功能，一个是下载并解码邮件正文（print_info函数），另一个是下载邮件附件（get_att函数）。所有代码都已在python3.6下运行通过，不过需要修改自己邮箱的账号密码、邮箱服务器、附件保存地址。

from email.parser import Parser
from email.header import decode_header
from email.utils import parseaddr
import email
import poplib

#python邮件读取
def guess_charset(msg):
    charset = msg.get_charset()         #获取msg编码方式
    if charset is None:
        content_type = msg.get('Content-Type', '').lower()   #获取内容类型字符串
        pos = content_type.find('charset=')           #内容类型中查找“charset=”字符串的位置
        if pos >= 0:                         
            charset = content_type[pos + 8:].strip()  #若存在上述字符串，则返回内容类型
    return charset

def decode_str(s):
    value, charset = decode_header(s)[0]      #Returns a list of (string, charset) pairs
    if charset:
        value = value.decode(charset)
    return value

def get_att(msg, fpath):
    attachment_files = []

    for part in msg.walk():
        file_name = part.get_filename()#获取附件名称类型
        contType = part.get_content_type()
        
        if file_name: 
            h = email.header.Header(file_name)
            dh = email.header.decode_header(h)     #对附件名称进行解码，
            #Returns a list of (string, charset) pairs containing each of the decoded parts of the header
            filename = dh[0][0]
            if dh[0][1]:
                filename = decode_str(str(filename,dh[0][1]))#将附件名称可读化
                print(filename)
                #filename = filename.encode("utf-8")
            data = part.get_payload(decode=True)#下载附件
            attachment_files.append(filename)
            with open(fpath + filename, 'wb') as att_file:  #在指定目录下创建文件，注意二进制文件需要用wb模式打开
                att_file.write(data)           #保存文件
    return attachment_files

def print_info(msg, indent=0):
    if indent == 0:
        for header in ['From', 'To', 'Subject']:
            value = msg.get(header, '')
            if value:
                if header=='Subject':
                    value = decode_str(value)
                else:
                    hdr, addr = parseaddr(value)   #根据地址返回一个realname和email address组成的元组
                    name = decode_str(hdr)
                    value = u'%s <%s>' % (name, addr)  #u指后面字符串以Unicode格式进行编码。该代码指按照前边格式组织后边两个数据
            print('%s%s: %s' % ('  ' * indent, header, value))
    if (msg.is_multipart()):
        parts = msg.get_payload()
        for n, part in enumerate(parts):
            print('%spart %s' % ('  ' * indent, n))
            print('%s--------------------' % ('  ' * indent))
            print_info(part, indent + 1)
    else:
        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)
            print('%sText: %s' % ('  ' * indent, content + '...'))
        else:
            print('%sAttachment: %s' % ('  ' * indent, content_type))
    return content


email = '12345678@qq.com'
password = 'abcdefghigk'               #该步骤输入密码
pop3_server = 'pop.qq.com'           #修改对应的邮箱服务器
server = poplib.POP3_SSL(pop3_server)
server.set_debuglevel(1)
print(server.getwelcome().decode('utf-8'))
server.user(email)
server.pass_(password)
print('Messages: %s. Size: %s' % server.stat())
resp, mails, octets = server.list()
index = len(mails)
for i in range(index,index-30,-1):
    #倒序遍历邮件
    resp, lines, octets = server.retr(i)        #retr返回一个三元素列表，第一个元素是response，即返回状态；第二个元素是一个列表，包括每一行的数据；第三个数据是octets
    msg_content = b'\r\n'.join(lines).decode('utf-8')   #返回line1\r\nline2\r\n 连接lines列表中的所有行。
    #解析邮件:
    msg = Parser().parsestr(msg_content)
    #获取邮件时间
    date1 = time.strptime(msg.get("Date")[0:24],'%a, %d %b %Y %H:%M:%S') #格式化收件时间
    date2 = time.strftime("%Y%m%d", date1)#邮件时间格式转换
    print(msg.get("from"))               #发件人
    print_info(msg)        #获取邮件内容  
server.quit()