from django.conf import settings as djsettings from django.core.mail.backends.base import BaseEmailBackend from orchestra.core.caches import get_request_cache from . import settings from .models import Message from .tasks import send_message class EmailBackend(BaseEmailBackend): """ A wrapper that manages a queued SMTP system. """ messages = 0 def send_messages(self, email_messages): if not email_messages: return cache = get_request_cache() key = 'mailer.sent_messages' sent_messages = cache.get(key) or 0 sent_messages += 1 cache.set(key, sent_messages) is_bulk = len(email_messages) > 1 if sent_messages > settings.MAILER_NON_QUEUED_MAILS_PER_REQUEST_THRESHOLD: is_bulk = True default_priority = Message.NORMAL if is_bulk else Message.CRITICAL num_sent = 0 for message in email_messages: priority = message.extra_headers.get('X-Mail-Priority', default_priority) content = message.message().as_string() for to_email in message.recipients(): message = Message( priority=priority, to_address=to_email, from_address=getattr(message, 'from_email', djsettings.DEFAULT_FROM_EMAIL), subject=message.subject, content=content, ) if priority == Message.CRITICAL: # send immidiately send_message.apply_async(message) else: message.save() num_sent += 1 return num_sent