import smtplib
from email.mime.text import MIMEText
from email.mime.image import MIMEImage
from email.mime.application import MIMEApplication
from email.mime.multipart import MIMEMultipart


class SendEMail(object):
  """封裝發(fā)送郵件類"""

  def __init__(self, host, port, msg_from, pwd):

    self.msg_from = msg_from
    self.password = pwd

    # 郵箱服務器地址和端口
    self.smtp_s = smtplib.SMTP_SSL(host=host, port=port)

    # 發(fā)送方郵箱賬號和授權碼
    self.smtp_s.login(user=msg_from, password=pwd)

  def send_text(self, to_user, content, subject, content_type='plain'):
    """
    發(fā)送文本郵件
    :param to_user: 對方郵箱
    :param content: 郵件正文
    :param subject: 郵件主題
    :param content_type: 內容格式：'plain' or 'html'
    :return:
    """
    msg = MIMEText(content, _subtype=content_type, _charset="utf8")

    msg["From"] = self.msg_from
    msg["To"] = to_user
    msg["subject"] = subject

    self.smtp_s.send_message(msg, from_addr=self.msg_from, to_addrs=to_user)

  def send_file(self, to_user, content, subject, reports_path, filename, content_type='plain'):
    """
    發(fā)送帶文件的郵件
    :param to_user: 對方郵箱
    :param content: 郵件正文
    :param subject: 郵件主題
    :param reports_path: 文件路徑
    :param filename: 郵件中顯示的文件名稱
    :param content_type: 內容格式
    """

    file_content = open(reports_path, "rb").read()

    msg = MIMEMultipart()

    text_msg = MIMEText(content, _subtype=content_type, _charset="utf8")
    msg.attach(text_msg)

    file_msg = MIMEApplication(file_content)
    file_msg.add_header('content-disposition', 'attachment', filename=filename)
    msg.attach(file_msg)

    msg["From"] = self.msg_from
    msg["To"] = to_user
    msg["subject"] = subject

    self.smtp_s.send_message(msg, from_addr=self.msg_from, to_addrs=to_user)

  def send_img(self, to_user, subject, content, filename, content_type='html'):
    '''
    發(fā)送帶圖片的郵件
    :param to_user: 對方郵箱
    :param subject: 郵件主題
    :param content: 郵件正文
    :param filename: 圖片路徑
    :param content_type: 內容格式
    '''
    subject = subject
    msg = MIMEMultipart('related')
    # Html正文必須包含<img src="cid:imageid" alt="imageid" width="100%" height="100%>
    content = MIMEText(content, _subtype=content_type, _charset="utf8")
    msg.attach(content)
    msg['Subject'] = subject
    msg['From'] = self.msg_from
    msg['To'] = to_user

    with open(filename, "rb") as file:
      img_data = file.read()

    img = MIMEImage(img_data)
    img.add_header('Content-ID', 'imageid')
    msg.attach(img)

    self.smtp_s.sendmail(self.msg_from, to_user, msg.as_string())