darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-01-31 23:26:23 -05:00
Code Issues Packages Projects Releases Wiki Activity

notify-api-412 use black to enforce python style standards

Browse Source
This commit is contained in:
Kenneth Kehl
2023-08-23 10:35:43 -07:00
parent a7898118d7
commit 026dc14021
586 changed files with 33990 additions and 23461 deletions
Download Patch File Download Diff File Expand all files Collapse all files

58
app/clients/performance_platform/performance_platform_client.py
View File

@@ -6,45 +6,49 @@ from flask import current_app
class PerformancePlatformClient:
@property
def active(self):
return self._active
def init_app(self, app):
self._active = app.config.get('PERFORMANCE_PLATFORM_ENABLED')
self._active = app.config.get("PERFORMANCE_PLATFORM_ENABLED")
if self.active:
self.performance_platform_url = app.config.get('PERFORMANCE_PLATFORM_URL')
self.performance_platform_endpoints = app.config.get('PERFORMANCE_PLATFORM_ENDPOINTS')
self.performance_platform_url = app.config.get("PERFORMANCE_PLATFORM_URL")
self.performance_platform_endpoints = app.config.get(
"PERFORMANCE_PLATFORM_ENDPOINTS"
)
def send_stats_to_performance_platform(self, payload):
if self.active:
bearer_token = self.performance_platform_endpoints[payload['dataType']]
bearer_token = self.performance_platform_endpoints[payload["dataType"]]
headers = {
'Content-Type': "application/json",
'Authorization': 'Bearer {}'.format(bearer_token)
"Content-Type": "application/json",
"Authorization": "Bearer {}".format(bearer_token),
}
resp = requests.post(
self.performance_platform_url + payload['dataType'],
self.performance_platform_url + payload["dataType"],
json=payload,
headers=headers
headers=headers,
)
if resp.status_code == 200:
current_app.logger.info(
"Updated performance platform successfully with payload {}".format(json.dumps(payload))
"Updated performance platform successfully with payload {}".format(
json.dumps(payload)
)
)
else:
current_app.logger.error(
"Performance platform update request failed for payload with response details: {} '{}'".format(
json.dumps(payload),
resp.status_code
json.dumps(payload), resp.status_code
)
)
resp.raise_for_status()
@staticmethod
def format_payload(*, dataset, start_time, group_name, group_value, count, period='day'):
def format_payload(
*, dataset, start_time, group_name, group_value, count, period="day"
):
"""
:param dataset - the name of the overall graph, as referred to in the endpoint.
:param start_time - UTC midnight of the day we're sending stats for
@@ -54,14 +58,16 @@ class PerformancePlatformClient:
:param period - the period that this data covers - "day", "week", "month", "quarter".
"""
payload = {
'_timestamp': start_time,
'service': 'govuk-notify',
'dataType': dataset,
'period': period,
'count': count,
"_timestamp": start_time,
"service": "govuk-notify",
"dataType": dataset,
"period": period,
"count": count,
group_name: group_value,
}
payload['_id'] = PerformancePlatformClient.generate_payload_id(payload, group_name)
payload["_id"] = PerformancePlatformClient.generate_payload_id(
payload, group_name
)
return payload
@staticmethod
@@ -69,12 +75,12 @@ class PerformancePlatformClient:
"""
group_name is the name of the group - eg "channel" or "status"
"""
payload_string = '{}{}{}{}{}'.format(
payload['_timestamp'],
payload['service'],
payload_string = "{}{}{}{}{}".format(
payload["_timestamp"],
payload["service"],
payload[group_name],
payload['dataType'],
payload['period']
payload["dataType"],
payload["period"],
)
_id = base64.b64encode(payload_string.encode('utf-8'))
return _id.decode('utf-8')
_id = base64.b64encode(payload_string.encode("utf-8"))
return _id.decode("utf-8")