diff --git a/app/__init__.py b/app/__init__.py index 6b85f33ab..50c1691f4 100644 --- a/app/__init__.py +++ b/app/__init__.py @@ -8,12 +8,14 @@ from werkzeug.local import LocalProxy from utils import logging from app.celery.celery import NotifyCelery from app.clients.sms.twilio import TwilioClient +from app.clients.sms.firetext import FiretextClient from app.encryption import Encryption db = SQLAlchemy() ma = Marshmallow() notify_celery = NotifyCelery() twilio_client = TwilioClient() +firetext_client = FiretextClient() encryption = Encryption() api_user = LocalProxy(lambda: _request_ctx_stack.top.api_user) @@ -30,6 +32,7 @@ def create_app(): init_app(application) logging.init_app(application) twilio_client.init_app(application) + firetext_client.init_app(application) notify_celery.init_app(application) encryption.init_app(application) diff --git a/app/celery/tasks.py b/app/celery/tasks.py index e7ae2bb3b..1be4afb48 100644 --- a/app/celery/tasks.py +++ b/app/celery/tasks.py @@ -1,5 +1,5 @@ -from app import notify_celery, twilio_client, encryption -from app.clients.sms.twilio import TwilioClientException +from app import notify_celery, encryption, firetext_client +from app.clients.sms.firetext import FiretextClientException from app.dao.templates_dao import get_model_templates from app.dao.notifications_dao import save_notification from app.models import Notification @@ -23,8 +23,8 @@ def send_sms(service_id, notification_id, encrypted_notification): save_notification(notification_db_object) try: - twilio_client.send_sms(notification['to'], template.content) - except TwilioClientException as e: + firetext_client.send_sms(notification['to'], template.content) + except FiretextClientException as e: current_app.logger.debug(e) save_notification(notification_db_object, {"status": "failed"}) diff --git a/app/clients/sms/firetext.py b/app/clients/sms/firetext.py new file mode 100644 index 000000000..1c22cb8dd --- /dev/null +++ b/app/clients/sms/firetext.py @@ -0,0 +1,52 @@ +import logging +from app.clients.sms import ( + SmsClient, + SmsClientException +) +from requests import request, RequestException, HTTPError + +logger = logging.getLogger(__name__) + + +class FiretextClientException(SmsClientException): + pass + + +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') + + def send_sms(self, to, content): + + data = { + "apiKey": self.api_key, + "from": self.from_number, + "to": to.replace('+', ''), + "message": content + } + + try: + response = request( + "POST", + "https://www.firetext.co.uk/api/sendsms", + data=data + ) + response.raise_for_status() + except RequestException as e: + api_error = HTTPError.create(e) + print( + "API {} request on {} failed with {} '{}'".format( + "POST", + "https://www.firetext.co.uk/api/sendsms", + api_error.status_code, + api_error.message + ) + ) + raise api_error + return response diff --git a/config.py b/config.py index 70eb584c8..8700d71a3 100644 --- a/config.py +++ b/config.py @@ -41,6 +41,8 @@ class Config(object): TWILIO_ACCOUNT_SID = os.getenv('TWILIO_ACCOUNT_SID') TWILIO_AUTH_TOKEN = os.getenv('TWILIO_AUTH_TOKEN') TWILIO_NUMBER = os.getenv('TWILIO_NUMBER') + FIRETEXT_NUMBER = os.getenv('FIRETEXT_NUMBER') + FIRETEXT_API_KEY = os.getenv("FIRETEXT_API_KEY") class Development(Config): diff --git a/environment_test.sh b/environment_test.sh index 2be312597..bbc44a855 100644 --- a/environment_test.sh +++ b/environment_test.sh @@ -12,4 +12,6 @@ export SQLALCHEMY_DATABASE_URI='postgresql://localhost/test_notification_api' export VERIFY_CODE_FROM_EMAIL_ADDRESS='no-reply@notify.works' export TWILIO_ACCOUNT_SID="test" export TWILIO_AUTH_TOKEN="test" -export TWILIO_NUMBER="test" \ No newline at end of file +export TWILIO_NUMBER="test" +export FIRETEXT_API_KEY="Firetext" +export FIRETEXT_NUMBER="Firetext" \ No newline at end of file diff --git a/requirements.txt b/requirements.txt index 184d57249..cca101658 100644 --- a/requirements.txt +++ b/requirements.txt @@ -13,6 +13,7 @@ itsdangerous==0.24 Flask-Bcrypt==0.6.2 credstash==1.8.0 boto3==1.2.3 +boto==2.39.0 celery==3.1.20 twilio==4.6.0 diff --git a/tests/app/celery/test_tasks.py b/tests/app/celery/test_tasks.py index 80b3b2afa..4bcdf452c 100644 --- a/tests/app/celery/test_tasks.py +++ b/tests/app/celery/test_tasks.py @@ -3,6 +3,9 @@ import pytest from app.celery.tasks import (send_sms, send_sms_code) from app import twilio_client, encryption from app.clients.sms.twilio import TwilioClientException +from app.celery.tasks import send_sms +from app import firetext_client +from app.clients.sms.firetext import FiretextClientException from app.dao import notifications_dao from sqlalchemy.exc import SQLAlchemyError from sqlalchemy.orm.exc import NoResultFound @@ -14,7 +17,7 @@ def test_should_send_template_to_correct_sms_provider_and_persist(sample_templat "to": "+441234123123" } mocker.patch('app.encryption.decrypt', return_value=notification) - mocker.patch('app.twilio_client.send_sms') + mocker.patch('app.firetext_client.send_sms') notification_id = uuid.uuid4() @@ -23,7 +26,7 @@ def test_should_send_template_to_correct_sms_provider_and_persist(sample_templat notification_id, "encrypted-in-reality") - twilio_client.send_sms.assert_called_once_with("+441234123123", sample_template.content) + firetext_client.send_sms.assert_called_once_with("+441234123123", sample_template.content) persisted_notification = notifications_dao.get_notification(sample_template.service_id, notification_id) assert persisted_notification.id == notification_id assert persisted_notification.to == '+441234123123' @@ -37,7 +40,7 @@ def test_should_persist_notification_as_failed_if_sms_client_fails(sample_templa "to": "+441234123123" } mocker.patch('app.encryption.decrypt', return_value=notification) - mocker.patch('app.twilio_client.send_sms', side_effect=TwilioClientException()) + mocker.patch('app.firetext_client.send_sms', side_effect=FiretextClientException()) notification_id = uuid.uuid4() @@ -46,7 +49,7 @@ def test_should_persist_notification_as_failed_if_sms_client_fails(sample_templa notification_id, "encrypted-in-reality") - twilio_client.send_sms.assert_called_once_with("+441234123123", sample_template.content) + firetext_client.send_sms.assert_called_once_with("+441234123123", sample_template.content) persisted_notification = notifications_dao.get_notification(sample_template.service_id, notification_id) assert persisted_notification.id == notification_id assert persisted_notification.to == '+441234123123' @@ -60,7 +63,7 @@ def test_should_not_send_sms_if_db_peristance_failed(sample_template, mocker): "to": "+441234123123" } mocker.patch('app.encryption.decrypt', return_value=notification) - mocker.patch('app.twilio_client.send_sms', side_effect=TwilioClientException()) + mocker.patch('app.firetext_client.send_sms', side_effect=FiretextClientException()) mocker.patch('app.db.session.add', side_effect=SQLAlchemyError()) notification_id = uuid.uuid4() @@ -70,7 +73,7 @@ def test_should_not_send_sms_if_db_peristance_failed(sample_template, mocker): notification_id, "encrypted-in-reality") - twilio_client.send_sms.assert_not_called() + firetext_client.send_sms.assert_not_called() with pytest.raises(NoResultFound) as e: notifications_dao.get_notification(sample_template.service_id, notification_id) assert 'No row was found for one' in str(e.value)