mirror of https://github.com/zulip/zulip.git
98 lines
3.5 KiB
Python
98 lines
3.5 KiB
Python
import logging
|
|
|
|
from typing import List
|
|
import configparser
|
|
|
|
import smtplib
|
|
from email.mime.multipart import MIMEMultipart
|
|
from email.mime.text import MIMEText
|
|
|
|
from django.conf import settings
|
|
from django.core.mail.backends.base import BaseEmailBackend
|
|
from django.core.mail import EmailMultiAlternatives
|
|
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) as e:
|
|
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 = MIMEMultipart('alternative')
|
|
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.attach(MIMEText(text, 'plain'))
|
|
msg.attach(MIMEText(html, 'html'))
|
|
|
|
smtp = smtplib.SMTP(settings.EMAIL_HOST)
|
|
smtp.starttls()
|
|
smtp.login(settings.EMAIL_HOST_USER, settings.EMAIL_HOST_PASSWORD)
|
|
smtp.sendmail(from_email, to, msg.as_string())
|
|
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, "r") 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:
|
|
self.log_email(email)
|
|
if get_forward_address():
|
|
self.send_email_smtp(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)
|