from models.core import os
import smtplib, email, base64
from email.mime.multipart import MIMEMultipart
from email.mime.text import MIMEText

mail_secret = os.getenv("SECRET_KEY")
msg = MIMEMultipart()

def send_email(subject, recipient_emails, body, html = False, attach = '', sender = ""):
    try :
        if sender == "": 
            sender = os.getenv("MAIL_SENDER")
        msg['From'] = sender
        msg['To'] = ', '.join(recipient_emails)
        msg['Subject'] = subject
        msg_content = MIMEText(body, 'plain', 'utf-8')
        if attach != '':
            with open(attach['localurl'], 'rb') as f:
                # set attachment mime and file name, the image type is png
                mime = MIMEBase(attach['type'], attach['extension'], filename=attach['filename'])
                # add required header data:
                mime.add_header('Content-Disposition', 'attachment', filename=attach['filename'])
                mime.add_header('X-Attachment-Id', '0')
                mime.add_header('Content-ID', '<0>')
                # read attachment file content into the MIMEBase object
                mime.set_payload(f.read())
                # encode with base64
                encoders.encode_base64(mime)
                # add MIMEBase object to MIMEMultipart object
                msg.attach(mime)
        if html is True:
            msg_content = MIMEText(body, 'html', 'utf-8')
        msg.attach(msg_content)
        mail = smtplib.SMTP(os.getenv('MAIL_SERVER'), os.getenv('MAIL_PORT')) 
        # mail = smtplib.SMTP_SSL(os.getenv('MAIL_SERVER'), os.getenv('MAIL_PORT'))
        mail.starttls()
        mail.login(os.getenv('MAIL_USERNAME'), os.getenv('MAIL_PASSWORD'))
        mail.sendmail(msg['From'], msg['To'], msg.as_string())
        mail.quit()
    except Exception as e:
        print(str(e))
        raise Exception(e._message)

def generate_email_token(email, type = 'email-confirm'):
    return mail_secret.dumps(email, salt=type)

def confirm_email_token(token, type = 'email-confirm'):
    try :
        return mail_secret.loads(token, salt=type, max_age=3600)
    except SignatureExpired:
        return False