import configparser import logging import smtplib from email.message import EmailMessage from typing import List from django.conf import settings from django.core.mail import EmailMultiAlternatives from django.core.mail.backends.base import BaseEmailBackend from django.template import loader def get_forward_address() -> str: config = configparser.ConfigParser() config.read(settings.FORWARD_ADDRESS_CONFIG_FILE) try: return config.get("DEV_EMAIL", "forward_address") except (configparser.NoSectionError, configparser.NoOptionError): return "" def set_forward_address(forward_address: str) -> None: config = configparser.ConfigParser() config.read(settings.FORWARD_ADDRESS_CONFIG_FILE) if not config.has_section("DEV_EMAIL"): config.add_section("DEV_EMAIL") config.set("DEV_EMAIL", "forward_address", forward_address) with open(settings.FORWARD_ADDRESS_CONFIG_FILE, "w") as cfgfile: config.write(cfgfile) class EmailLogBackEnd(BaseEmailBackend): def send_email_smtp(self, email: EmailMultiAlternatives) -> None: from_email = email.from_email to = get_forward_address() msg = EmailMessage() msg['Subject'] = email.subject msg['From'] = from_email msg['To'] = to text = email.body html = email.alternatives[0][0] # Here, we replace the email addresses used in development # with chat.zulip.org, so that web email providers like Gmail # will be able to fetch the illustrations used in the emails. localhost_email_images_base_uri = settings.ROOT_DOMAIN_URI + '/static/images/emails' czo_email_images_base_uri = 'https://chat.zulip.org/static/images/emails' html = html.replace(localhost_email_images_base_uri, czo_email_images_base_uri) msg.add_alternative(text, subtype="plain") msg.add_alternative(html, subtype="html") smtp = smtplib.SMTP(settings.EMAIL_HOST) smtp.starttls() smtp.login(settings.EMAIL_HOST_USER, settings.EMAIL_HOST_PASSWORD) smtp.send_message(msg) smtp.quit() def log_email(self, email: EmailMultiAlternatives) -> None: """Used in development to record sent emails in a nice HTML log""" html_message = 'Missing HTML message' if len(email.alternatives) > 0: html_message = email.alternatives[0][0] context = { 'subject': email.subject, 'from_email': email.from_email, 'recipients': email.to, 'body': email.body, 'html_message': html_message, } new_email = loader.render_to_string('zerver/email.html', context) # Read in the pre-existing log, so that we can add the new entry # at the top. try: with open(settings.EMAIL_CONTENT_LOG_PATH) as f: previous_emails = f.read() except FileNotFoundError: previous_emails = "" with open(settings.EMAIL_CONTENT_LOG_PATH, "w+") as f: f.write(new_email + previous_emails) def send_messages(self, email_messages: List[EmailMultiAlternatives]) -> int: for email in email_messages: if get_forward_address(): self.send_email_smtp(email) if settings.DEVELOPMENT_LOG_EMAILS: self.log_email(email) email_log_url = settings.ROOT_DOMAIN_URI + "/emails" logging.info("Emails sent in development are available at %s", email_log_url) return len(email_messages)