darkhelm/notifications-admin
1
0
Fork 0
mirror of https://github.com/GSA/notifications-admin.git synced 2026-08-18 05:29:38 -04:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Kenneth Kehl
2023-08-25 09:12:23 -07:00
parent c6eb007386
commit 8c9721d8e2
201 changed files with 31660 additions and 28105 deletions
Download Patch File Download Diff File Expand all files Collapse all files

73
app/notify_client/notification_api_client.py
View File

@@ -2,7 +2,6 @@ from app.notify_client import NotifyAdminAPIClient, _attach_current_user
class NotificationApiClient(NotifyAdminAPIClient):
def get_notifications_for_service(
self,
service_id,
@@ -20,16 +19,16 @@ class NotificationApiClient(NotifyAdminAPIClient):
include_one_off=None,
):
params = {
'page': page,
'page_size': page_size,
'template_type': template_type,
'status': status,
'include_jobs': include_jobs,
'include_from_test_key': include_from_test_key,
'format_for_csv': format_for_csv,
'to': to,
'include_one_off': include_one_off,
'count_pages': count_pages,
"page": page,
"page_size": page_size,
"template_type": template_type,
"status": status,
"include_jobs": include_jobs,
"include_from_test_key": include_from_test_key,
"format_for_csv": format_for_csv,
"to": to,
"include_one_off": include_one_off,
"count_pages": count_pages,
}
params = {k: v for k, v in params.items() if v is not None}
@@ -37,34 +36,37 @@ class NotificationApiClient(NotifyAdminAPIClient):
# if `to` is set it is likely PII like an email address or mobile which
# we do not want in our logs, so we do a POST request instead of a GET
method = self.post if to else self.get
kwargs = {'data': params} if to else {'params': params}
kwargs = {"data": params} if to else {"params": params}
if job_id:
return method(
url='/service/{}/job/{}/notifications'.format(service_id, job_id),
url="/service/{}/job/{}/notifications".format(service_id, job_id),
**kwargs
)
else:
if limit_days is not None:
params['limit_days'] = limit_days
return method(
url='/service/{}/notifications'.format(service_id),
**kwargs
)
params["limit_days"] = limit_days
return method(url="/service/{}/notifications".format(service_id), **kwargs)
def send_notification(self, service_id, *, template_id, recipient, personalisation, sender_id):
def send_notification(
self, service_id, *, template_id, recipient, personalisation, sender_id
):
data = {
'template_id': template_id,
'to': recipient,
'personalisation': personalisation,
"template_id": template_id,
"to": recipient,
"personalisation": personalisation,
}
if sender_id:
data['sender_id'] = sender_id
data["sender_id"] = sender_id
data = _attach_current_user(data)
return self.post(url='/service/{}/send-notification'.format(service_id), data=data)
return self.post(
url="/service/{}/send-notification".format(service_id), data=data
)
def get_notification(self, service_id, notification_id):
return self.get(url='/service/{}/notifications/{}'.format(service_id, notification_id))
return self.get(
url="/service/{}/notifications/{}".format(service_id, notification_id)
)
def get_api_notifications_for_service(self, service_id):
ret = self.get_notifications_for_service(
@@ -72,26 +74,31 @@ class NotificationApiClient(NotifyAdminAPIClient):
include_jobs=False,
include_from_test_key=True,
include_one_off=False,
count_pages=False
count_pages=False,
)
return ret
def update_notification_to_cancelled(self, service_id, notification_id):
return self.post(
url='/service/{}/notifications/{}/cancel'.format(service_id, notification_id),
data={})
url="/service/{}/notifications/{}/cancel".format(
service_id, notification_id
),
data={},
)
def get_notification_status_by_service(self, start_date, end_date):
return self.get(
url='service/monthly-data-by-service',
url="service/monthly-data-by-service",
params={
'start_date': str(start_date),
'end_date': str(end_date),
}
"start_date": str(start_date),
"end_date": str(end_date),
},
)
def get_notification_count_for_job_id(self, *, service_id, job_id):
return self.get(url='/service/{}/job/{}/notification_count'.format(service_id, job_id))["count"]
return self.get(
url="/service/{}/job/{}/notification_count".format(service_id, job_id)
)["count"]
notification_api_client = NotificationApiClient()