python 使用模板模式和工厂模式的混合设计开发各种邮件客户端发送邮件

# coding=utf8
"""
使用模板模式、工厂模式的混合设计,开发各种邮件客户端
"""
import sysreload(sys)
sys.setdefaultencoding('utf8')import re
import unittest
from abc import ABCMeta, abstractmethod
import smtplib
from email.mime.text import MIMETextclass MailClient(object):
"""一个通用的邮件客户端，作为模板模式的模板类。
该类为抽象类，不可以进行实例化。
_set_smtp方法为抽象方法，子类必须重写这个方法.
"""
__metaclass__ = ABCMeta def __init__(self, msg_from, passwd):
"""
:param msg_from: 发件人邮箱地址
:param passwd: 发件人邮箱密码,qq邮箱使用授权码是16个字母，而不是自己的邮箱密码。
"""
self._mail_name = None
self._msg_from = msg_from
self._passwd = passwd
self._smtp = None
self._set_smtp()
self.__login() @abstractmethod
def _set_smtp(self):
pass def __login(self):
try:
self._smtp.login(self._msg_from, self._passwd)
except smtplib.SMTPAuthenticationError as e:
raise smtplib.SMTPAuthenticationError(535, '{}邮箱登录失败\n'.format(self._mail_name) + e.message) def send_mail(self, msg_to, subject, content):
"""
发送邮件
:param msg_to: 所有收件人的邮箱地址列表,类型为字符串列表，单个接收人也可以用字符串
:param subject :邮件主题
:param content:邮件内容
:return :邮件是否发送成功
:type msg_to:list
:type subject:str
:type content:str
"""
msg = MIMEText(content, _charset='utf8')
msg['Subject'] = subject
msg['From'] = self._msg_from
msg['To'] = msg_to
self._smtp.sendmail(self._msg_from, msg_to, msg.as_string())
print '邮件发送成功，发送的邮件主题是： {} 。请去邮箱检查邮件'.format(subject)
return True def __str__(self):
return '登录帐号为： {0} 的{1}邮箱客户端'.format(self._msg_from, self._mail_name) def quit(self):
try: # 如果没登陆成功就关闭连接，会出错，try一下
self._smtp.quit()
except:
pass def __del__(self):
self.quit()class QQMailClient(MailClient):
def _set_smtp(self):
self._mail_name = 'qq'
self._smtp = smtplib.SMTP_SSL("smtp.qq.com", 465)class Wangyi163MailClient(MailClient):
def _set_smtp(self):
self._mail_name = ''
self._smtp = smtplib.SMTP_SSL("smtp.163.com", 465)class MailNameException(Exception):
"""不支持的邮箱异常类"""
def __init__(self, msg_from):
err = '你设置的邮箱账号是 {} ，不支持此邮箱'.format(msg_from)
super(MailNameException, self).__init__(err)class MailClientFactory(object):
@staticmethod
def get_mail_client(msg_from, passwd):
"""
:param msg_from:发件人邮箱地址
:param passwd:邮箱密码
:return: 具体的邮箱对象
:type msg_from:str
:type passwd:str
:rtype :MailClient
"""
if re.match('[0-9a-zA-Z_]+@qq\.com', msg_from):
return QQMailClient(msg_from, passwd)
elif re.match('[0-9a-zA-Z_]+@163\.com', msg_from):
return Wangyi163MailClient(msg_from, passwd)
else:
raise MailNameException(msg_from)class MailTest(unittest.TestCase):
def test_163(self):
"""测试163邮箱发送两个邮件"""
wangyi163_mail_client = MailClientFactory.get_mail_client('m13148804506@163.com', '')
print wangyi163_mail_client
wangyi163_mail_client.send_mail('909686716@qq.com', '测试网易163邮箱发送主题1', '测试网易163邮箱发送内容1')
result = wangyi163_mail_client.send_mail('909686716@qq.com', '测试网易163邮箱发送主题2', '测试网易163邮箱发送内容2')
self.assertEqual(result, True, msg='163邮件发送失败') def test_qq(self):
"""测试两个qq号发送邮件"""
qq_123456_client = MailClientFactory.get_mail_client('123456@qq.com', 'ssoodruxniyfxxxx')
print qq_123456_client
qq_789000_client = MailClientFactory.get_mail_client('789000@qq.com', 'ssoodruxniyfxxxx')
print qq_789000_client
result = qq_123456_client.send_mail('m13148804506@163.com', '测试qq邮箱发送主题3', '测试qq邮箱发送内容3')
self.assertEqual(result, True, msg='qq邮件发送失败')
result = qq_789000_client.send_mail('m13148804506@163.com', '测试qq邮箱发送主题4', '测试qq邮箱发送内容4')
self.assertEqual(result, True, msg='qq邮件发送失败') def test_126(self):
"""测试一个不支持的邮箱"""
self.assertRaises(MailNameException, MailClientFactory.get_mail_client, 'm13148804506@126.com', '')if __name__ == "__main__":
unittest.main()