django-mailbox/django_mailbox/management/commands/processincomingmessage.py

import email
import logging
import sys
try:
    from email import utils
except ImportError:
    import rfc822 as utils

from django.core.management.base import BaseCommand

from django_mailbox.models import Mailbox


logger = logging.getLogger(__name__)
logging.basicConfig(level=logging.INFO)


class Command(BaseCommand):
    help = "Receive incoming mail via stdin"

    def add_arguments(self, parser):
        parser.add_argument(
            'mailbox_name',
            nargs='?',
            help="The name of the mailbox that will receive the message"
        )

    def handle(self, mailbox_name=None, *args, **options):
        message = email.message_from_string(sys.stdin.read())
        if message:
            if mailbox_name:
                mailbox = self.get_mailbox_by_name(mailbox_name)
            else:
                mailbox = self.get_mailbox_for_message(message)
            mailbox.process_incoming_message(message)
            logger.info(
                "Message received from %s",
                message['from']
            )
        else:
            logger.warning("Message not processable.")

    def get_mailbox_by_name(self, name):
        mailbox, created = Mailbox.objects.get_or_create(
            name=name,
        )
        return mailbox

    def get_mailbox_for_message(self, message):
        email_address = utils.parseaddr(message['to'])[1][0:255]
        return self.get_mailbox_by_name(email_address)