danidee10/django-notifs

from typing import List

from pydantic import BaseModel, Field

try:
    from sms import Message, get_connection

    HAS_DEPENDENCIES = True
except ImportError:
    HAS_DEPENDENCIES = False

from . import BaseNotificationProvider


class DjangoSmsSchema(BaseModel):
    body: str = Field(description='The message body')
    originator: str = Field(description='The originating phone number')
    recipients: List[str] = Field(description='The list of recipient phone numbers')


class DjangoSMSNotificationProvider(BaseNotificationProvider):
    name = 'django_sms'
    validator = DjangoSmsSchema
    package = 'django-sms'

    HAS_DEPENDENCIES = HAS_DEPENDENCIES

    @staticmethod
    def _get_sms_message(payload):
        message = Message()
        for key, value in payload.items():
            setattr(message, key, value)

        return message

    def send(self, payload):
        sms_message = self._get_sms_message(payload)

        return sms_message.send()

    def send_bulk(self, payloads):
        messages = (self._get_sms_message(payload) for payload in payloads)
        connection = get_connection()

        return connection.send_messages(messages)