make mmg and firetext client params consistent

This commit is contained in:
Leo Hemsted
2016-06-01 15:24:19 +01:00
parent a97f3bbb4f
commit 0a8cb679d7
3 changed files with 32 additions and 15 deletions

View File

@@ -52,7 +52,7 @@ def create_app(app_name=None):
statsd_client.init_app(application) statsd_client.init_app(application)
firetext_client.init_app(application, statsd_client=statsd_client) firetext_client.init_app(application, statsd_client=statsd_client)
loadtest_client.init_app(application, statsd_client=statsd_client) loadtest_client.init_app(application, statsd_client=statsd_client)
mmg_client.init_app(application.config, statsd_client=statsd_client) mmg_client.init_app(application, statsd_client=statsd_client)
aws_ses_client.init_app(application.config['AWS_REGION'], statsd_client=statsd_client) aws_ses_client.init_app(application.config['AWS_REGION'], statsd_client=statsd_client)
notify_celery.init_app(application) notify_celery.init_app(application)
encryption.init_app(application) encryption.init_app(application)

View File

@@ -1,11 +1,12 @@
import logging import logging
from monotonic import monotonic from monotonic import monotonic
from requests import request, RequestException, HTTPError
from app.clients.sms import ( from app.clients.sms import (
SmsClient, SmsClient,
SmsClientException SmsClientException
) )
from flask import current_app
from requests import request, RequestException, HTTPError
from app.clients import STATISTICS_DELIVERED, STATISTICS_FAILURE from app.clients import STATISTICS_DELIVERED, STATISTICS_FAILURE
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@@ -55,10 +56,11 @@ class FiretextClient(SmsClient):
FireText sms client. FireText sms client.
''' '''
def init_app(self, config, statsd_client, *args, **kwargs): def init_app(self, current_app, statsd_client, *args, **kwargs):
super(SmsClient, self).__init__(*args, **kwargs) super(SmsClient, self).__init__(*args, **kwargs)
self.api_key = config.config.get('FIRETEXT_API_KEY') self.current_app = current_app
self.from_number = config.config.get('FIRETEXT_NUMBER') self.api_key = current_app.config.get('FIRETEXT_API_KEY')
self.from_number = current_app.config.get('FIRETEXT_NUMBER')
self.name = 'firetext' self.name = 'firetext'
self.statsd_client = statsd_client self.statsd_client = statsd_client
@@ -86,6 +88,14 @@ class FiretextClient(SmsClient):
if firetext_response['code'] != 0: if firetext_response['code'] != 0:
raise FiretextClientException(firetext_response) raise FiretextClientException(firetext_response)
response.raise_for_status() response.raise_for_status()
self.current_app.logger.info(
"API {} request on {} succeeded with {} '{}'".format(
"POST",
"https://www.firetext.co.uk/api/sendsms",
response.status_code,
firetext_response
)
)
except RequestException as e: except RequestException as e:
api_error = HTTPError.create(e) api_error = HTTPError.create(e)
logger.error( logger.error(
@@ -100,6 +110,6 @@ class FiretextClient(SmsClient):
raise api_error raise api_error
finally: finally:
elapsed_time = monotonic() - start_time elapsed_time = monotonic() - start_time
current_app.logger.info("Firetext request finished in {}".format(elapsed_time)) self.current_app.logger.info("Firetext request finished in {}".format(elapsed_time))
self.statsd_client.timing("notifications.clients.firetext.request-time", elapsed_time) self.statsd_client.timing("notifications.clients.firetext.request-time", elapsed_time)
return response return response

View File

@@ -1,5 +1,4 @@
import json import json
from flask import current_app
from monotonic import monotonic from monotonic import monotonic
from requests import (request, RequestException, HTTPError) from requests import (request, RequestException, HTTPError)
from app.clients import (STATISTICS_DELIVERED, STATISTICS_FAILURE) from app.clients import (STATISTICS_DELIVERED, STATISTICS_FAILURE)
@@ -57,10 +56,11 @@ class MMGClient(SmsClient):
MMG sms client MMG sms client
''' '''
def init_app(self, config, statsd_client, *args, **kwargs): def init_app(self, current_app, statsd_client, *args, **kwargs):
super(SmsClient, self).__init__(*args, **kwargs) super(SmsClient, self).__init__(*args, **kwargs)
self.api_key = config.get('MMG_API_KEY') self.current_app = current_app
self.from_number = config.get('MMG_FROM_NUMBER') self.api_key = current_app.config.get('MMG_API_KEY')
self.from_number = current_app.config.get('MMG_FROM_NUMBER')
self.name = 'mmg' self.name = 'mmg'
self.statsd_client = statsd_client self.statsd_client = statsd_client
@@ -84,12 +84,19 @@ class MMGClient(SmsClient):
headers={'Content-Type': 'application/json', headers={'Content-Type': 'application/json',
'Authorization': 'Basic {}'.format(self.api_key)}) 'Authorization': 'Basic {}'.format(self.api_key)})
if response.status_code != 200: if response.status_code != 200:
error = response.text raise MMGClientException(response.json())
raise MMGClientException(json.loads(error))
response.raise_for_status() response.raise_for_status()
self.current_app.logger.info(
"API {} request on {} succeeded with {} '{}'".format(
"POST",
"https://www.mmgrp.co.uk/API/json/api.php",
response.status_code,
response.json()
)
)
except RequestException as e: except RequestException as e:
api_error = HTTPError.create(e) api_error = HTTPError.create(e)
current_app.logger.error( self.current_app.logger.error(
"API {} request on {} failed with {} '{}'".format( "API {} request on {} failed with {} '{}'".format(
"POST", "POST",
"https://www.mmgrp.co.uk/API/json/api.php", "https://www.mmgrp.co.uk/API/json/api.php",
@@ -102,5 +109,5 @@ class MMGClient(SmsClient):
finally: finally:
elapsed_time = monotonic() - start_time elapsed_time = monotonic() - start_time
self.statsd_client.timing("notifications.clients.mmg.request-time", elapsed_time) self.statsd_client.timing("notifications.clients.mmg.request-time", elapsed_time)
current_app.logger.info("MMG request finished in {}".format(elapsed_time)) self.current_app.logger.info("MMG request finished in {}".format(elapsed_time))
return response return response