mirror of
https://github.com/paperless-ngx/paperless-ngx.git
synced 2025-04-21 10:29:29 -05:00
34 lines
1.1 KiB
Python
34 lines
1.1 KiB
Python
import logging
|
|
|
|
from celery import shared_task
|
|
|
|
from paperless_mail.mail import MailAccountHandler
|
|
from paperless_mail.mail import MailError
|
|
from paperless_mail.models import MailAccount
|
|
from paperless_mail.models import MailRule
|
|
|
|
logger = logging.getLogger("paperless.mail.tasks")
|
|
|
|
|
|
@shared_task
|
|
def process_mail_accounts(account_ids: list[int] | None = None) -> str:
|
|
total_new_documents = 0
|
|
accounts = (
|
|
MailAccount.objects.filter(pk__in=account_ids)
|
|
if account_ids
|
|
else MailAccount.objects.all()
|
|
)
|
|
for account in accounts:
|
|
if not MailRule.objects.filter(account=account, enabled=True).exists():
|
|
logger.info(f"No rules enabled for account {account}. Skipping.")
|
|
continue
|
|
try:
|
|
total_new_documents += MailAccountHandler().handle_mail_account(account)
|
|
except MailError:
|
|
logger.exception(f"Error while processing mail account {account}")
|
|
|
|
if total_new_documents > 0:
|
|
return f"Added {total_new_documents} document(s)."
|
|
else:
|
|
return "No new documents were added."
|