#-*- encoding: gb2312 -*-
import email
import string, sys, os, email
import time

class MailCreator:
    def __init__(self):
        # 创建邮件的message对象
        self.msg = email.Message.Message()
        self.mail = ""    
        
    def create(self, mailheader, maildata, mailattachlist=[]):
        # mailheader 是dict类型，maildata是list, 且里面第一项为纯文本类型，第二项为html.
        # mailattachlist 是list, 里面为附件文件名
        if not mailheader or not maildata:
            return
        
        for k in mailheader.keys():
            # 对subject要作特殊处理，中文要转换一下。
            # 比如 "我的一个测试邮件" 就要转换为 =?gb2312?b?ztK1xNK7uPay4srU08q8/g==?=
            if k == 'subject':
                self.msg[k] = email.Header.Header(mailheader[k], 'gb2312')           
            else:
                self.msg[k] = mailheader[k]
        # 创建纯文本部分
        body_plain = email.MIMEText.MIMEText(maildata[0], _subtype='plain', _charset='gb2312')
        body_html = None
        # 创建html部分，这个是可选的
        if maildata[1]:
            body_html = email.MIMEText.MIMEText(maildata[1], _subtype='html', _charset='gb2312')
        
        
        # 创建一个multipart, 然后把前面的文本部分和html部分都附加到上面，至于为什么，可以看看mime相关内容
        attach=email.MIMEMultipart.MIMEMultipart()
        attach.attach(body_plain)
        if body_html:
            attach.attach(body_html)
        # 处理每一个附件
        for fname in mailattachlist:
            attachment=email.MIMEText.MIMEText(email.Encoders._bencode(open(fname,'rb').read()))
            # 这里设置文件类型，全部都设置为Application.当然也可以是Image,Audio什么的，这里不管那么多
            attachment.replace_header('Content-type','Application/octet-stream;name="'+os.path.basename(fname)+'"')
            # 一定要把传输编码设置为base64,因为这里默认就是用的base64
            attachment.replace_header('Content-Transfer-Encoding', 'base64')
            attachment.add_header('Content-Disposition','attachment;filename="'+os.path.basename(fname)+'"')
            attach.attach(attachment)
        # 生成最终的邮件            
        self.mail = self.msg.as_string()[:-1] + attach.as_string()
        
        return self.mail

if __name__ == '__main__':
    mc = MailCreator()
    header = {'from': 'zhaowei@163.com', 'to':'weizhao@163.com', 'subject':'我的一个测试邮件'}
    data = ['plain text information', '<font color="red">html text information</font>']
    if sys.platform == 'win32':
        attach = ['c:\windows\clock.avi']
    else:
        attach = ['/bin/cp']
    
    mail = mc.create(header, data, attach)
    
    f = open("test.eml", "wb")
    f.write(mail)
    f.close()