mailfile.py
功能：主要用来发送文件。for Python 2.x
#!/usr/bin/env python# -*- coding: utf-8 -*-class ConfigFile: def __init__(self): from os.path import expanduser home = expanduser('~') self.rcfile = '%s/.mailfilerc' % home self.loadConfigFile() def initConfigFile(self): c = '''from=xxx@@xxx.comto=xxx@@xxx.comhost=smtp.xxx.comport=25user=xxxpasswd=xxx''' f = open(self.rcfile, 'w') f.write(c) f.close() def loadConfigFile(self): import ConfigParser from os.path import exists if exists(self.rcfile): cf = ConfigParser.ConfigParser() cf.read(self.rcfile) self.host = cf.get('smtp', 'host') self.port = cf.getint('smtp', 'port') self.user = cf.get('smtp', 'user') self.passwd = cf.get('smtp', 'passwd') self.FROM = cf.get('mail','from') self.TO = cf.get('mail','to') else: self.initConfigFile() print ('** Please edit the config file %s !!!' % self.rcfile)def sendmail(host, port, user, passwd, FILE): import email, smtplib, base64, sys from os.path import getsize from os import remove from decimal import Decimal as dec fp1=open(FILE) msg = email.message_from_file(fp1) fp1.close() FROM = email.utils.parseaddr(msg.get('from')) TO = email.utils.parseaddr(msg.get('to')) h = email.Header.Header(msg.get('subject')) SUBJECT = email.Header.decode_header(h) print (('Connecting SMTP server %s ...') % host) smtp=smtplib.SMTP(host, port) #smtp.set_debuglevel(1) smtp.docmd('EHLO server') smtp.docmd('AUTH LOGIN') smtp.send(base64.b64encode(user)+'\r\n') smtp.getreply() smtp.send(base64.b64encode(passwd)+'\r\n') smtp.getreply() print('Logined') smtp.docmd('MAIL FROM:<%s>' % FROM) smtp.docmd('RCPT TO:<%s>' % TO) smtp.docmd('DATA') print('begin send data...') BUFSIZE=2048 f=open(FILE,'rb') fileLen=getsize(FILE) curper=0 bytes=f.read(BUFSIZE) while(bytes!=b'' and bytes!=''): curper+=len(bytes) sys.stdout.flush() sys.stdout.write('\r%6.2f %%' % (dec(100*curper)/dec(fileLen))) smtp.send(bytes.decode('utf-8')) bytes=f.read(BUFSIZE) f.close() smtp.send('\r\n.\r\n') smtp.getreply() smtp.quit() remove(FILE) print (('\nmail sended. ^_^'))class Mail: def __init__(self, FROM, TO, SUBJECT = None, CHARSET = 'utf-8'): from email.mime.multipart import MIMEMultipart self.FROM = FROM self.TO = TO self.SUBJECT = SUBJECT self.CHARSET = CHARSET self.text = '' self.fileinfo = '\n----\n' self.files = [] self.msg = MIMEMultipart() self.msg['From'] = '<%s>' % self.FROM self.msg['To'] = '<%s>' % self.TO def addtext(self, text = ''): self.text += text def addfile(self, FILE): import mimetypes from email.mime.image import MIMEImage from email.mime.text import MIMEText from email.mime.audio import MIMEAudio from email.mime.application import MIMEApplication from os.path import basename fileName = r'%s' % FILE baseName = basename(fileName) ctype, encoding = mimetypes.guess_type(fileName) if ctype is None or encoding is not None: ctype = 'application/octet-stream' maintype, subtype = ctype.split('/', 1) if maintype == 'text': att = MIMEText((lambda f: (f.read(), f.close()))(open(fileName, 'rb')), \ _subtype = subtype, _charset = self.CHARSET) att.add_header('Content-Disposition', 'attachment', \ filename = self.header(baseName)) att.set_charset(self.CHARSET) self.files.append(att) elif maintype == 'image': att = MIMEImage((lambda f: (f.read(), f.close()))(open(fileName, 'rb')), \ _subtype = subtype) att.add_header('Content-Disposition', 'attachment', \ filename = self.header(baseName)) att.set_charset(self.CHARSET) self.files.append(att) elif maintype == 'audio': att = MIMEAudio((lambda f: (f.read(), f.close()))(open(fileName, 'rb')), \ _subtype = subtype) att.add_header('Content-Disposition', 'attachment', \ filename = self.header(baseName)) att.set_charset(self.CHARSET) self.files.append(att) else: att = MIMEApplication((lambda f: (f.read(), f.close()))(open(fileName, 'rb')), \ _subtype = subtype) att.add_header('Content-Disposition', 'attachment', \ filename = self.header(baseName)) att.set_charset(self.CHARSET) self.files.append(att) if not self.SUBJECT: self.SUBJECT = baseName self.fileinfo += '%s\t%s\n' % (self.md5sum(fileName), baseName) def makemail(self): from email.mime.text import MIMEText import uuid self.msg['Subject'] = self.header(self.SUBJECT) txt = MIMEText(self.utf8(self.text + self.fileinfo), \ 'plain', self.CHARSET) self.msg.attach(txt) for x in self.files: self.msg.attach(x) fName = 'MAIL_' + str(uuid.uuid4()) fp = open(fName,'w') fp.write(self.msg.as_string()) fp.close() return fName def utf8(self, data): try: return data.decode(self.CHARSET).encode(self.CHARSET) except: return data.decode('gb18030').encode(self.CHARSET) def header(self, data): '''encode into base64.''' from email.Header import Header return Header(data, self.CHARSET).encode() def md5sum(self, FILE): '''return a file's md5sum.''' from hashlib import md5 BUFSIZE = 1024 m = md5() file = open(FILE, 'r') bytes = file.read(BUFSIZE) while(bytes != b''): m.update(bytes) bytes = file.read(BUFSIZE) file.close() return m.hexdigest()def main(): import sys from optparse import OptionParser from os.path import exists parser = OptionParser(usage = 'Usage: %prog <address>') parser.add_option('-a', dest = 'fileList', action = 'append', \ help = 'attach FILE', metavar = 'FILE') parser.add_option('-s', dest = 'subject', action = 'store', \ help = 'set SUBJECT', metavar = 'SUBJECT') parser.add_option('-v', dest = 'verbose', action = 'store_true', \ help = 'show only,not send') parser.add_option('-t', dest = 'isType', action = 'store_true', \ help = 'type text from stdin') parser.set_defaults(verbose = False) parser.set_defaults(isType = False) (opts, args) = parser.parse_args() if not opts.fileList and not opts.subject and not args: parser.print_help() sys.exit(1) cf = ConfigFile() From = cf.FROM if not args: mailTo = cf.TO else: mailTo = args mymail = Mail(From, mailTo, opts.subject) if opts.isType: text = sys.stdin.read() mymail.addtext(text) if opts.fileList: for file in opts.fileList: if exists(file): mymail.addfile(file) fileName = mymail.makemail() if opts.verbose: print(mymail.msg.as_string()) else: sendmail(cf.host, cf.port, cf.user, cf.passwd, fileName)if __name__ == '__main__' : main()
for Python 3.x
#!/usr/bin/env python3'''mailfile.py for python3'''class ConfigFile: def __init__(self): from os.path import expanduser home = expanduser('~') self.rcfile = '%s/.mailfilerc' % home self.loadConfigFile() def initConfigFile(self): c = '''from=xxx@@xxx.comto=xxx@@xxx.comhost=smtp.xxx.comport=25user=xxxpasswd=xxx''' f = open(self.rcfile, 'w') f.write(c) f.close() def loadConfigFile(self): import configparser from os.path import exists if exists(self.rcfile): cf = configparser.ConfigParser() cf.read(self.rcfile) self.host = cf.get('smtp', 'host') self.port = cf.getint('smtp', 'port') self.user = cf.get('smtp', 'user') self.passwd = cf.get('smtp', 'passwd') self.FROM = cf.get('mail','from') self.TO = cf.get('mail','to') else: self.initConfigFile() print ('** Please edit the config file %s !!!' % self.rcfile)def sendmail(host, port, user, passwd, FILE): import email, smtplib, base64, sys from os.path import getsize from os import remove from decimal import Decimal as dec fp=open(FILE) msg = email.message_from_file(fp) fp.close() FROM = email.utils.parseaddr(msg.get('from')) TO = email.utils.parseaddr(msg.get('to')) h = msg.get('subject') SUBJECT = email.header.decode_header(h) print(('Connecting SMTP server %s ...') % host) smtp = smtplib.SMTP(host, port) #smtp.set_debuglevel(1) smtp.docmd('EHLO server') smtp.docmd('AUTH LOGIN') smtp.send(base64.b64encode(user.encode('utf-8')).decode('utf-8')+'\r\n') smtp.getreply() smtp.send(base64.b64encode(passwd.encode('utf-8')).decode('utf-8')+'\r\n') smtp.getreply() print('Logined') smtp.docmd('MAIL FROM:<%s>' % FROM) smtp.docmd('RCPT TO:<%s>' % TO) smtp.docmd('DATA') print(('send mail %s to <%s> ...') % (SUBJECT.decode('utf-8'), TO)) print('begin send data...') BUFSIZE = 2048 f = open(FILE, 'rb') fileLen = getsize(FILE) curper = 0 bytes = f.read(BUFSIZE) while(bytes != b'' and bytes != ''): curper += len(bytes) sys.stdout.write('\r%6.2f %%' % (dec(100)*dec(curper)/dec(fileLen))) smtp.send(bytes.decode('utf-8')) bytes = f.read(BUFSIZE) f.close() smtp.send('\r\n.\r\n') smtp.getreply() smtp.quit() remove(FILE) print('\nmail sended. ^_^')class mail: def __init__(self, FROM, TO, SUBJECT = None, CHARSET = 'utf-8'): from email.mime.multipart import MIMEMultipart self.FROM = FROM self.TO = TO self.SUBJECT = SUBJECT self.CHARSET = CHARSET self.text = '' self.fileinfo = '\n----\n' self.files = [] self.msg = MIMEMultipart() self.msg['From'] = self.FROM self.msg['To'] = self.TO def addtext(self, text = ''): self.text += text def addfile(self, FILE): import mimetypes from email.mime.image import MIMEImage from os.path import basename fileName = r'%s' % FILE baseName = basename(fileName) ctype, encoding = mimetypes.guess_type(fileName) if ctype is None or encoding is not None: ctype = 'application/octet-stream' maintype, subtype = ctype.split('/', 1) att = MIMEImage((lambda f: (f.read(), f.close())) \ (open(fileName, 'rb')), _subtype = subtype) att.add_header('Content-Disposition', 'attachment', \ filename = self.header(baseName)) att.set_charset(self.CHARSET) att.set_payload(att.get_payload().decode()) self.files.append(att) if not self.SUBJECT: self.SUBJECT = baseName self.fileinfo += '%s\t%s\n' % (self.md5sum(fileName), baseName) def makemail(self): from email.mime.text import MIMEText import uuid self.msg['Subject'] = self.header(self.SUBJECT) txt = MIMEText(self.utf8(self.text + self.fileinfo), \ 'plain', self.CHARSET) self.msg.attach(txt) for x in self.files: self.msg.attach(x) fName = 'MAIL_' + str(uuid.uuid4()) fp = open(fName,'w') fp.write(self.msg.as_string()) fp.close() return fName def utf8(self, data): return data.encode(self.CHARSET) def header(self, data): '''encode into base64.''' from email.header import Header return Header(data, self.CHARSET).encode() def md5sum(self, FILE): '''return a file's md5sum.''' from hashlib import md5 BUFSIZE = 1024 m = md5() file = open(FILE, 'rb') bytes = file.read(BUFSIZE) while(bytes != b''): m.update(bytes) bytes = file.read(BUFSIZE) file.close() return m.hexdigest()def main(): import sys from optparse import OptionParser from os.path import exists parser = OptionParser(usage = 'Usage: %prog <address>') parser.add_option('-a', dest = 'fileList', action = 'append', \ help = 'attach FILE', metavar = 'FILE') parser.add_option('-s', dest = 'subject', action = 'store', \ help = 'set SUBJECT', metavar = 'SUBJECT') parser.add_option('-v', dest = 'verbose', action = 'store_true', \ help = 'show only,not send') parser.set_defaults(verbose = False) (opts, args) = parser.parse_args() if not opts.fileList and not opts.subject and not args: parser.print_help() sys.exit(1) cf = ConfigFile() From = cf.FROM if not args: mailTo = cf.TO else: mailTo = args mymail = mail(From, mailTo, opts.subject) if opts.fileList: for file in opts.fileList: if exists(file): mymail.addfile(file) fileName = mymail.makemail() if opts.verbose: print(mymail.msg.as_string()) else: sendmail(cf.host, cf.port, cf.user, cf.passwd, fileName)if __name__ == '__main__' : main()
页:
[1]