import logging

from paperless_mail.mail import MailAccountHandler, MailError
from paperless_mail.models import MailAccount


def process_mail_accounts():
    total_new_documents = 0
    for account in MailAccount.objects.all():
        try:
            total_new_documents += MailAccountHandler().handle_mail_account(
                account)
        except MailError as e:
            logging.getLogger(__name__).error(
                f"Error while processing mail account {account}: {e}",
                exc_info=True
            )

    if total_new_documents > 0:
        return f"Added {total_new_documents} document(s)."
    else:
        return "No new documents were added."


def process_mail_account(name):
    try:
        account = MailAccount.objects.get(name=name)
        MailAccountHandler().handle_mail_account(account)
    except MailAccount.DoesNotExist:
        logging.getLogger(__name__).error(f"Unknown mail acccount: {name}")