notifications-api/app/clients/sms/firetext.py

import logging
from monotonic import monotonic
from app.clients.sms import (
    SmsClient,
    SmsClientException
)
from flask import current_app
from requests import request, RequestException, HTTPError
from app.clients import ClientResponse, STATISTICS_DELIVERED, STATISTICS_FAILURE

logger = logging.getLogger(__name__)


class FiretextResponses(ClientResponse):
    def __init__(self):
        ClientResponse.__init__(self)
        self.__response_model__ = {
            '0': {
                "message": 'Delivered',
                "notification_statistics_status": STATISTICS_DELIVERED,
                "success": True,
                "notification_status": 'delivered'
            },
            '1': {
                "message": 'Declined',
                "success": False,
                "notification_statistics_status": STATISTICS_FAILURE,
                "notification_status": 'failed'
            },
            '2': {
                "message": 'Undelivered (Pending with Network)',
                "success": False,
                "notification_statistics_status": None,
                "notification_status": 'sent'
            }
        }


class FiretextClientException(SmsClientException):
    def __init__(self, response):
        self.code = response['code']
        self.description = response['description']

    def __str__(self):
        return "Code {} description {}".format(self.code, self.description)


class FiretextClient(SmsClient):
    '''
    FireText sms client.
    '''

    def init_app(self, config, *args, **kwargs):
        super(SmsClient, self).__init__(*args, **kwargs)
        self.api_key = config.config.get('FIRETEXT_API_KEY')
        self.from_number = config.config.get('FIRETEXT_NUMBER')
        self.name = 'firetext'

    def get_name(self):
        return self.name

    def send_sms(self, to, content, reference):

        data = {
            "apiKey": self.api_key,
            "from": self.from_number,
            "to": to.replace('+', ''),
            "message": content,
            "reference": reference
        }

        start_time = monotonic()
        try:
            response = request(
                "POST",
                "https://www.firetext.co.uk/api/sendsms/json",
                data=data
            )
            firetext_response = response.json()
            if firetext_response['code'] != 0:
                raise FiretextClientException(firetext_response)
            response.raise_for_status()
        except RequestException as e:
            api_error = HTTPError.create(e)
            logger.error(
                "API {} request on {} failed with {} '{}'".format(
                    "POST",
                    "https://www.firetext.co.uk/api/sendsms",
                    api_error.status_code,
                    api_error.message
                )
            )
            raise api_error
        finally:
            elapsed_time = monotonic() - start_time
            current_app.logger.info("Firetext request finished in {}".format(elapsed_time))
        return response