1
0
Fork 0

Adding optional argument to processincomingmessages management command for allowing one to specify a mailbox name.

This commit is contained in:
Adam Coddington 2012-10-27 14:54:54 -07:00
parent 22b31a5c2f
commit 4ceeb4dae7
2 changed files with 17 additions and 6 deletions

View file

@ -11,18 +11,27 @@ logger = logging.getLogger(__name__)
logging.basicConfig(level=logging.INFO)
class Command(BaseCommand):
def handle(self, *args, **options):
args = "<[Mailbox Name (optional)]>"
command = "Receive incoming mail via stdin"
def handle(self, mailbox_name=None, *args, **options):
message = email.message_from_string(sys.stdin.read())
if message:
mailbox = self.get_mailbox_for_message(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_for_message(self, message):
email_address = rfc822.parseaddr(message['to'])[1][0:255]
def get_mailbox_by_name(self, name):
mailbox, created = Mailbox.objects.get_or_create(
name=email_address,
name=name,
)
return mailbox
def get_mailbox_for_message(self, message):
email_address = rfc822.parseaddr(message['to'])[1][0:255]
return self.get_mailbox_by_name(email_address)