Merge pull request #15 from jayvdb/outbox-backend

Add OutboxEmailBackend

Author: delneg

django_mail_admin/backends.py

@@ -1,8 +1,14 @@
+import logging
 import threading
 
+from django.core.mail.backends.base import BaseEmailBackend
 from django.core.mail.backends.smtp import EmailBackend
 
-from .models import Outbox
+from .mail import create
+from .models import Outbox, create_attachments
+from .utils import PRIORITY
+
+logger = logging.getLogger(__name__)
 
 
 class CustomEmailBackend(EmailBackend):
@@ -28,3 +34,35 @@ def __init__(self, host=None, port=None, username=None, password=None,
         self.ssl_certfile = configuration.email_ssl_certfile if ssl_certfile is None else ssl_certfile
         self.connection = None
         self._lock = threading.RLock()
+
+
+class OutboxEmailBackend(BaseEmailBackend):
+    def send_messages(self, email_messages):
+        for msg in email_messages:
+            try:
+                email = create(
+                    sender=msg.from_email,
+                    recipients=msg.to,
+                    cc=msg.cc,
+                    bcc=msg.bcc,
+                    subject=msg.subject,
+                    message=msg.body,
+                    headers=msg.extra_headers,
+                    priority=PRIORITY.medium,
+                )
+                alternatives = getattr(msg, 'alternatives', [])
+                for content, mimetype in alternatives:
+                    if mimetype == 'text/html':
+                        email.html_message = content
+                        email.save()
+
+                if msg.attachments:
+                    attachments = create_attachments(msg.attachments)
+                    email.attachments.add(*attachments)
+
+            except Exception:
+                if not self.fail_silently:
+                    raise
+                logger.exception('Email queue failed')
+
+        return len(email_messages)

tests/test_backends.py

@@ -1,10 +1,11 @@
+from django.core.mail import send_mail 
 from django.core.mail.backends.base import BaseEmailBackend
 from django.test import TestCase
 from django.test.utils import override_settings
 from django_mail_admin.mail import send
 from django_mail_admin.settings import get_backend
 from django_mail_admin.backends import CustomEmailBackend
-from django_mail_admin.models import Outbox
+from django_mail_admin.models import Outbox, OutgoingEmail
 
 
 class ErrorRaisingBackend(BaseEmailBackend):
@@ -42,3 +43,14 @@ def test_custom_email_backend(self):
         outbox.delete()
         with self.assertRaises(ValueError):
             backend2 = CustomEmailBackend()
+
+    @override_settings(DJANGO_MAIL_ADMIN={}, EMAIL_BACKEND='django_mail_admin.backends.OutboxEmailBackend')
+    def test_outbox_email_backend(self):
+        count_before = len(OutgoingEmail.objects.all())
+        sent_count = send_mail(
+            subject='Test subject', message='message.', 
+            from_email='from@example.com', recipient_list=['to@example.com'], 
+            fail_silently=False)
+        self.assertEqual(sent_count, 1)
+        count_after = len(OutgoingEmail.objects.all())
+        self.assertEqual(count_after, count_before + 1)