cr-connect-workflow/crc/services/email_service.py

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)