# coding=utf8

"""

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

"""

import sys

reload(sys)

sys.setdefaultencoding('utf8')

import re

import unittest

from abc import ABCMeta, abstractmethod

import smtplib

from email.mime.text import MIMEText

class 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()