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

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):
    args = ""
    help = ""

    def add_arguments(self, parser):
        parser.add_argument('-nl', '--nologs', action='store_true', default=False, dest='nologs',
                            help='Disable logging (default False)')

    def handle(self, *args, **options):
        mailboxes = Mailbox.active_mailboxes.all()
        logging = True
        if options['nologs']:
            logging = False        

        if args:
            mailboxes = mailboxes.filter(
                name=' '.join(args)
            )
        for mailbox in mailboxes:
            if logging:
                logger.info(
                    'Gathering messages for %s',
                    mailbox.name
                )
            messages = mailbox.get_new_mail()
            for message in messages:
                if logging:
                    logger.info(
                        'Received %s (from %s)',
                        message.subject,
                        message.from_address
                    )