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

Move the bulk get_new_mail to queryset

Browse source
This commit is contained in:
George Cheng 2020-07-26 14:21:21 +08:00
parent 505bc8d1c9
commit 0c6f6aa6ce
3 changed files with 27 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

18
django_mailbox/models.py
View file

@ -36,9 +36,23 @@ from django_mailbox.transports import Pop3Transport, ImapTransport, \
logger = logging.getLogger(__name__)
class MailboxQuerySet(models.QuerySet):
def get_new_mail(self):
count = 0
for mailbox in self.all():
logger.debug('Receiving mail for %s' % mailbox)
count += sum(1 for i in mailbox.get_new_mail())
logger.debug('Received %d %s.', count, 'mails' if count != 1 else 'mail')
class MailboxManager(models.Manager):
def get_queryset(self):
return MailboxQuerySet(self.model, using=self._db)
class ActiveMailboxManager(models.Manager):
def get_queryset(self):
return super().get_queryset().filter(
return MailboxQuerySet(self.model, using=self._db).filter(
active=True,
)
@ -106,7 +120,7 @@ class Mailbox(models.Model):
null=True
)
objects = models.Manager()
objects = MailboxManager()
active_mailboxes = ActiveMailboxManager()
@property