79 lines
2.7 KiB
Python
79 lines
2.7 KiB
Python
import markdown
|
|
import re
|
|
|
|
from flask import render_template
|
|
from flask_mail import Message
|
|
from jinja2 import Template
|
|
|
|
from crc import app, db, mail, session
|
|
|
|
from crc.models.email import EmailModel
|
|
from crc.models.file import FileDataModel
|
|
from crc.models.study import StudyModel
|
|
|
|
|
|
class EmailService(object):
|
|
"""Provides common tools for working with an Email"""
|
|
|
|
@staticmethod
|
|
def add_email(subject, sender, recipients, content, content_html,
|
|
cc=None, bcc=None, study_id=None, reply_to=None, attachment_files=None, workflow_id=None):
|
|
"""We will receive all data related to an email and store it"""
|
|
|
|
# Find corresponding study - if any
|
|
study = None
|
|
if type(study_id) == int:
|
|
study = db.session.query(StudyModel).get(study_id)
|
|
|
|
# Create EmailModel
|
|
email_model = EmailModel(subject=subject, sender=sender, recipients=str(recipients),
|
|
content=content, content_html=content_html, study=study,
|
|
cc=cc, bcc=bcc, workflow_id=workflow_id)
|
|
|
|
# Send mail
|
|
try:
|
|
msg = Message(subject,
|
|
sender=sender,
|
|
recipients=recipients,
|
|
body=content,
|
|
html=content_html,
|
|
cc=cc,
|
|
bcc=bcc,
|
|
reply_to=reply_to)
|
|
|
|
if attachment_files is not None:
|
|
for file in attachment_files:
|
|
file_data = session.query(FileDataModel).filter(FileDataModel.file_model_id==file['id']).first()
|
|
msg.attach(file['name'], file['type'], file_data.data)
|
|
|
|
mail.send(msg)
|
|
except Exception as e:
|
|
app.logger.error('An exception happened in EmailService', exc_info=True)
|
|
app.logger.error(str(e))
|
|
|
|
db.session.add(email_model)
|
|
db.session.commit()
|
|
return email_model
|
|
|
|
@staticmethod
|
|
def check_valid_email(email):
|
|
# regex from https://emailregex.com/
|
|
regex = r"(^[a-zA-Z0-9_.+-]+@[a-zA-Z0-9-]+\.[a-zA-Z0-9-.]+$)"
|
|
if re.search(regex, email):
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
def get_rendered_content(self, message, data):
|
|
template = Template(message)
|
|
rendered = template.render(data)
|
|
rendered_markdown = markdown.markdown(rendered)
|
|
wrapped = self.get_cr_connect_wrapper(rendered_markdown)
|
|
|
|
return rendered, wrapped
|
|
|
|
@staticmethod
|
|
def get_cr_connect_wrapper(email_body):
|
|
base_url = app.config['FRONTEND'] # The frontend url
|
|
return render_template('mail_content_template.html', email_body=email_body, base_url=base_url)
|