Django Email Backend for Gov Notify

govnotify_email_backend.py

"""
Backend for Gov.Notify
"""

from django.core import mail
from django.core.mail.backends.base import BaseEmailBackend
from notifications_python_client.notifications import NotificationsAPIClient
from django.conf import settings

class GovNotifyEmailBackend(BaseEmailBackend):
    def __init__(self, fail_silently=False, **kwargs):
        self.fail_silently = fail_silently
        self.api_key = settings.GOV_NOTIFY_API_KEY
        self.template_id = settings.GOV_NOTIFY_EMAIL_TEMPLATE_ID
        self.notifications_client = NotificationsAPIClient(self.api_key)
    def open(self):
        pass
    
    def close(self):
        pass

    def __enter__(self):
        try:
            self.open()
        except Exception:
            self.close()
            raise
        return self

    def __exit__(self, exc_type, exc_value, traceback):
        self.close()

    def send_messages(self, email_messages):
        """
        Send one or more EmailMessage objects and return the number of email
        messages sent.
        """
        msg_count = 0
        for message in email_messages:
            for recipient in message.recipients():
                response = self.notifications_client.send_email_notification(
                    email_address = recipient,
                    template_id = self.template_id,
                    personalisation = {
                        'subject': message.subject,
                        'body': message.body
                    }
            )
            msg_count += 1
        return msg_count