mirror/django-mailbox
1
0
Fork 1
mirror of https://github.com/coddingtonbear/django-mailbox.git synced 2026-07-10 06:48:19 +02:00
Code Activity

Added explicit getmail command arguments to parse one or more mailbox names and added maximum email read count

Browse source
This commit is contained in:
MSI Shafik 2023-05-18 10:24:43 +06:00
parent 462fdd3e49
commit 221585a254
3 changed files with 19 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

7
django_mailbox/management/commands/getmail.py
View file

@ -6,6 +6,11 @@ from django_mailbox.models import Mailbox
class Command(BaseCommand):
def add_arguments(self, parser):
parser.add_argument('-mb', '--mailboxes', type=str, nargs='+', help='Write mailbox names')
parser.add_argument('-mr', '--max_read', type=int,
help='Write maximum number of email to read from each mailbox')
def handle(self, *args, **options):
logging.basicConfig(level=logging.INFO)
Mailbox.get_new_mail_all_mailboxes(args)
Mailbox.get_new_mail_all_mailboxes(options)

16
django_mailbox/models.py
View file

@ -446,15 +446,19 @@ class Mailbox(models.Model):
save=False
)
def get_new_mail(self, condition=None):
def get_new_mail(self, condition=None, max_read=None):
"""Connect to this transport and fetch new messages."""
new_mail = []
connection = self.get_connection()
if not connection:
return
for message in connection.get_message(condition):
if max_read is not None and max_read <= 0:
break
msg = self.process_incoming_message(message)
if not msg is None:
if msg is not None:
if max_read is not None:
max_read -= 1
yield msg
self.last_polling = now()
if django.VERSION >= (1, 5): # Django 1.5 introduces update_fields
@ -463,18 +467,18 @@ class Mailbox(models.Model):
self.save()
@staticmethod
def get_new_mail_all_mailboxes(args=None):
def get_new_mail_all_mailboxes(options: dict):
mailboxes = Mailbox.active_mailboxes.all()
if args:
if options.get('mailboxes'):
mailboxes = mailboxes.filter(
name=' '.join(args)
name__in=options['mailboxes']
)
for mailbox in mailboxes:
logger.info(
'Gathering messages for %s',
mailbox.name
)
messages = mailbox.get_new_mail()
messages = mailbox.get_new_mail(max_read=options.get('max_read'))
for message in messages:
logger.info(
'Received %s (from %s)',