danidee10/django-notifs

"""Base backend."""

import abc
import logging

from django.utils.module_loading import import_string

from notifications.channels import BaseNotificationChannel
from notifications.providers import BaseNotificationProvider
from notifications.signals import post_bulk_send, post_send, pre_bulk_send, pre_send


class BaseBackend(metaclass=abc.ABCMeta):

    logging.getLogger().setLevel(logging.INFO)
    logging.basicConfig(level=logging.INFO)
    logger = logging.getLogger('django_notifs.backends.base')

    def __init__(self, notification_channel: BaseNotificationChannel):
        self.notification_channel = notification_channel

    @staticmethod
    def get_notification_provider(provider, context):
        provider_class = BaseNotificationProvider.get_provider_class(provider)

        return import_string(provider_class)(context)

    @classmethod
    def consume(cls, provider, payload, context):
        notification_provider = cls.get_notification_provider(provider, context)
        provider_class = notification_provider.__class__
        notification_provider.validate(payload)

        bulk = context.get('bulk', False)
        if bulk is True:
            pre_bulk_send.send(sender=provider_class, context=context, payload=payload)
            response = notification_provider.send_bulk(payload)
            post_bulk_send.send(
                sender=provider_class,
                context=context,
                payload=payload,
                response=response,
            )
        else:
            pre_send.send(sender=provider_class, context=context, payload=payload)
            response = notification_provider.send(payload)
            post_send.send(
                sender=provider_class,
                context=context,
                payload=payload,
                response=response,
            )

        cls.logger.info(
            'Sent notification with the %s provider with context: %s\n'
            % (provider, context)
        )

    @abc.abstractclassmethod
    def produce(self):
        raise NotImplementedError

    def run(self, countdown):
        for (
            provider,
            provider_class,
        ) in self.notification_channel.provider_paths.items():
            payload = self.notification_channel.build_payload(provider)
            context = self.notification_channel.get_context(provider)
            self.get_notification_provider(provider, context).validate(payload)

            self.logger.info(
                'Calling provider `%s`(%s) with args:'
                'payload: %s, context: %s, countdown %s'
                % (provider, provider_class, payload, context, countdown)
            )

            self.produce(provider, payload, context, countdown)