基于smtplib包制作而成，但在實(shí)踐中發(fā)現(xiàn)一個(gè)不知道算不算是smtplib留的一個(gè)坑，在網(wǎng)絡(luò)斷開(kāi)的情況下發(fā)送郵件時(shí)會(huì)拋出一個(gè)socket.gaierror的異常，但是smtplib中并沒(méi)有捕獲這個(gè)異常，導(dǎo)致程序會(huì)因這個(gè)異常終止，因此代碼中針對(duì)這部分的異常進(jìn)行處理，確保不會(huì)異常終止。
#!/usr/bin/env python# -*- coding: utf-8 -*-__author__ = 'Zoa Chou'# see http://www.mudoom.com/Article/show/id/29.html for detailimport loggingimport smtplibimport mimetypesimport socketfrom email import encodersfrom email.header import Headerfrom email.mime.text import MIMEText, MIMENonMultipartfrom email.mime.base import MIMEBasefrom email.utils import parseaddr, formataddrclass Mailer(object): def __init__(self): pass def send_mail(self, smtp_server, from_address, to_address, subject, body, files=None): """ 發(fā)送郵件主程序 :param smtp_server: dict 郵件服務(wù)器設(shè)置 :keyword host: string smtp服務(wù)器地址 :keyword port: int smtp服務(wù)器端口號(hào) :keyword user: string 用戶(hù)名 :keyword passwd: string 密碼 :keyword ssl: bool 是否啟用ssl,默認(rèn)False :keyword timeout: int 超時(shí)時(shí)間,默認(rèn)10s :param from_address: 發(fā)件人郵箱 :param to_address: 收件人郵箱 :param subject: 郵件標(biāo)題 :param body: 郵件內(nèi)容 :param files: 附件 :raise: NetworkError/MailerException """ # 格式化郵件內(nèi)容 body = self._encode_utf8(body) # 郵件類(lèi)型 content_type = 'html' if body.startswith('<html>') else 'plain' msg = MIMENonMultipart() if files else MIMEText(body, content_type, 'utf-8') # 格式化郵件數(shù)據(jù) msg['From'] = self._format_address(from_address) msg['To'] = ', '.join(self._format_list(to_address)) msg['subject'] = self._encode_utf8(subject) # 構(gòu)造附件數(shù)據(jù) if files: msg.attach(MIMEText(body, content_type, 'utf-8')) cid = 0 for file_name, payload in files: file_name = self._encode_utf8(file_name) main_type, sub_type = self._get_file_type(file_name) if hasattr(payload, 'read'): payload = payload.read() f_name = self._encode_header(file_name) mime = MIMEBase(main_type, sub_type, filename=f_name) mime.add_header('Content-Disposition', 'attachment', filename=f_name) mime.add_header('Content-ID', '<%s>' % cid) mime.add_header('X-Attachment-Id', '%s' % cid) mime.set_payload(payload) encoders.encode_base64(mime) msg.attach(mime) cid += 1 host = smtp_server.get('host') port = smtp_server.get('port') user = smtp_server.get('user') passwd = smtp_server.get('passwd') ssl = smtp_server.get('ssl', False) time_out = smtp_server.get('timeout', 10) # 沒(méi)有輸入端口則使用默認(rèn)端口 if port is None or port == 0: if ssl: port = 465 else: port = 25 logging.debug('Send mail form %s to %s' % (msg['From'], msg['To'])) try: if ssl: # 開(kāi)啟ssl連接模式 server = smtplib.SMTP_SSL('%s:%d' % (host, port), timeout=time_out) else: server = smtplib.SMTP('%s:%d' % (host, port), timeout=time_out) # 開(kāi)啟調(diào)試模式 # server.set_debuglevel(1) # 如果存在用戶(hù)名密碼則嘗試登錄 if user and passwd: server.login(user, passwd) # 發(fā)送郵件 server.sendmail(from_address, to_address, msg.as_string()) logging.debug('Mail sent success.') # 關(guān)閉stmp連接 server.quit() except socket.gaierror, e: """ 網(wǎng)絡(luò)無(wú)法連接 """ logging.exception(e) raise NetworkError(e) except smtplib.SMTPServerDisconnected, e: """ 網(wǎng)絡(luò)連接異常 """ logging.exception(e) raise NetworkError(e) except smtplib.SMTPException, e: """ 郵件發(fā)送異常 """ logging.exception(e) raise MailerException(e) def _format_address(self, s): """ 格式化郵件地址 :param s:string 郵件地址 :return: string 格式化后的郵件地址 """ name, address = parseaddr(s) return formataddr((self._encode_header(name), self._encode_utf8(address))) def _encode_header(self, s): """ 格式化符合MIME的頭部數(shù)據(jù) :param s: string 待格式化數(shù)據(jù) :return: 格式化后的數(shù)據(jù) """ return Header(s, 'utf-8').encode() def _encode_utf8(self, s): """ 格式化成utf-8編碼 :param s: string 待格式化數(shù)據(jù) :return: string 格式化后的數(shù)據(jù) """ if isinstance(s, unicode): return s.encode('utf-8') else: return s def _get_file_type(self, file_name): """ 獲取附件類(lèi)型 :param file_name: 附件文件名 :return: dict 附件MIME """ s = file_name.lower() pos = s.rfind('.') if pos == -1: return 'application', 'octet-stream' ext = s[pos:] mime = mimetypes.types_map.get(ext, 'application/octet-stream') pos = mime.find('/') if pos == (-1): return mime, '' return mime[:pos], mime[pos+1:] def _format_list(self, address): """ 將收件人地址格式化成list :param address: string/list 收件人郵箱 :return: list 收件人郵箱list """ l = address if isinstance(l, basestring): l = [l] return [self._format_address(s) for s in l]class MailerException(Exception): """ 郵件發(fā)送異常類(lèi) """ passclass NetworkError(MailerException): """ 網(wǎng)絡(luò)異常類(lèi) """ pass# test for @qq.comif __name__ == '__main__': import sys def prompt(prompt): """ 接收終端輸入的數(shù)據(jù) """ sys.stdout.write(prompt + ": ") return sys.stdin.readline().strip() from_address = prompt("From(Only @qq.com)") passwd = prompt("Password") to_address = prompt("To").split(',') subject = prompt("Subject") print "Enter message, end with ^D:" msg = '' while 1: line = sys.stdin.readline() if not line: break msg = msg + line print "Message length is %d" % len(msg) # QQ郵箱默認(rèn)設(shè)置 smtp_server = {'host': 'smtp.qq.com', 'port': None, 'user': from_address, 'passwd': passwd, 'ssl': True} mailer = Mailer() try: mailer.send_mail(smtp_server, from_address, to_address, subject, msg) except MailerException, e: print(e)以上所述就是本文的全部?jī)?nèi)容了，希望大家能夠喜歡。
新聞熱點(diǎn)
疑難解答
圖片精選
網(wǎng)友關(guān)注
