Source code for cms_qe_newsletter.external_services.sync

from django.core.exceptions import ObjectDoesNotExist
from requests.exceptions import HTTPError

from ..models import SERVICE_MAILCHIMP, SUBSCRIBE, UNSUBSCRIBE, Subscriber, SubscribeTask
from . import mailchimp

SERVICE_TO_MODULE = {
    SERVICE_MAILCHIMP: mailchimp,
}


def sync_mailing_lists():
    """
    Download mailing lists from external services like Mailchimp.
    """
    mailchimp.sync_mailing_lists()


def sync_tasks():
    """
    Processing tasks yielding same tuple as :py:func:`~.sync_task` function.
    """
    tasks = SubscribeTask.objects.all()
    for task in tasks:
        result, message = True, ""
        try:
            result, message = sync_task(task)

        # pylint: disable=broad-except
        except Exception as exc:
            result, message = False, str(exc)

        finally:
            if result is False:
                task.failure(message)
            yield (result, f'{task}: {message}')


# pylint: disable=too-many-return-statements
def sync_task(task):
    """
    Processing task returning tuple of status (``True`` for success, ``False`` for
    some failure and ``None`` for warnings) and message.
    """
    if not task.should_process():
        return None, 'Skipped'

    service = SERVICE_TO_MODULE.get(task.mailing_list.external_service)
    if not service:
        return False, f'Unsupported service {task.mailing_list.external_service}'

    try:
        if task.type == SUBSCRIBE:
            sync_subscribe(service, task)
        elif task.type == UNSUBSCRIBE:
            sync_unsubscribe(service, task)

    except HTTPError as exc:
        text = exc.response.text
        if 'fake or invalid' in text:
            Subscriber.objects.filter(email=task.email).all().delete()
            task.delete()
            return None, f'Email is invalid or fake and is deleted\n{text}'
        return False, f'{text}\n{exc}'

    except ObjectDoesNotExist:
        task.delete()
        return None, 'Subscriber does not exist anymore, deleting task'

    except Exception as exc:  # pylint: disable=broad-except
        return False, str(exc)

    task.delete()
    return True, 'OK'


def sync_subscribe(service, task):
    subscriber = Subscriber.objects.get(mailing_list=task.mailing_list, email=task.email)

    external_id = service.sync_subscribe(
        task.mailing_list.external_id,
        task.email,
        task.first_name,
        task.last_name,
    )

    subscriber.external_id = external_id
    subscriber.save()


def sync_unsubscribe(service, task):
    service.sync_unsubscribe(
        task.mailing_list.external_id,
        task.email,
    )