forked from mirror/django-mailbox
52 lines
1.5 KiB
Python
52 lines
1.5 KiB
Python
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):
|
|
args = "<[Mailbox Name (optional)]>"
|
|
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)
|