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


class Send_Mailbox(object):
    '''
    host_server:主机地址-> 默认是QQ邮箱的主机地址
    sender:发件人邮箱号
    code:授权码
    target:收件人邮箱号

    title：邮件标题
    content：邮件正文内容
    filepath：邮件需要添加的附件地址
    filename：邮件发送时的问价名
    '''

    def __init__(self, sender, code, target, host_server='smtp.qq.com'):
        self.host_server = host_server
        self.sender = sender
        self.code = code
        self.target = target

    def send_out(self):
        # 创建SMTP
        smtp = smtplib.SMTP(self.host_server)
        try:
            # 登录操作
            smtp.login(self.sender, self.code)
            # 发送
            smtp.sendmail(self.sender, self.target, self.msg.as_string())
            print("发送成功")
        except:
            print("发送失败,请检查校验码是否正确或者失效,邮箱地址是否正确")

    def Set_mail_data(self, title, content, filepath=None, filename=None):
        if filepath:
            try:
                with open(filepath, 'rb') as f:
                    if not filename:
                        filename = filepath.split('/')[-1]
                    # 创建附件实例
                    attachment = MIMEApplication(f.read())
                    attachment.add_header('Content-Disposition', 'attachment', filename=filename)
                    msg = MIMEMultipart()  # 带附件的实例
                    msg['Subject'] = title  # 设置邮箱标题
                    msg['From'] = self.sender  # 设置发件人邮箱
                    msg['To'] = self.target  # 设置邮件接收方
                    msg.attach(MIMEText(content))  # 设置邮件内容
                    msg.attach(attachment)  # 添加附件
                    self.msg = msg
            except FileNotFoundError:
                print('文件路径错误')
        else:
            msg = MIMEText(content)  # 不带附件的实例
            msg['Subject'] = title  # 设置邮箱标题
            msg['From'] = self.sender  # 设置发件人邮箱
            msg['To'] = self.target  # 设置邮件接收方
            self.msg = msg