1
0
Fork 1
mirror of https://github.com/coddingtonbear/django-mailbox.git synced 2026-07-09 22:38:19 +02:00

Cleaning up management commands for PEP-8 and consistency.

This commit is contained in:
Adam Coddington 2014-08-22 20:02:58 -07:00
parent 992375fc6b
commit db6190a295
2 changed files with 27 additions and 9 deletions

View file

@ -1,17 +1,30 @@
import logging
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):
def handle(self, *args, **options):
mailboxes = Mailbox.active_mailboxes.all()
if args:
mailboxes = mailboxes.filter(name = ' '.join(args))
mailboxes = mailboxes.filter(
name=' '.join(args)
)
for mailbox in mailboxes:
self.stdout.write('Gathering messages for %s\n' % mailbox.name)
logger.info(
'Gathering messages for %s',
mailbox.name
)
messages = mailbox.get_new_mail()
for message in messages:
self.stdout.write('Received %s (from %s)\n' % (
message.subject,
message.from_address
))
logger.info(
'Received %s (from %s)',
message.subject,
message.from_address
)

View file

@ -7,9 +7,11 @@ 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)]>"
command = "Receive incoming mail via stdin"
@ -22,14 +24,17 @@ class Command(BaseCommand):
else:
mailbox = self.get_mailbox_for_message(message)
mailbox.process_incoming_message(message)
logger.info("Message received from %s" % message['from'])
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,
)
name=name,
)
return mailbox
def get_mailbox_for_message(self, message):