mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-07-14 02:09:44 -04:00
82634f79de
When we first built the contact list feature this endpoint didn’t exist. Now it does we can remove the slightly cludgy looping-through-all-the-lists code.
35 lines
941 B
Python
35 lines
941 B
Python
from app.notify_client import NotifyAdminAPIClient, _attach_current_user
|
|
|
|
|
|
class ContactListApiClient(NotifyAdminAPIClient):
|
|
|
|
def create_contact_list(
|
|
self,
|
|
*,
|
|
service_id,
|
|
upload_id,
|
|
original_file_name,
|
|
row_count,
|
|
template_type,
|
|
):
|
|
data = {
|
|
"id": upload_id,
|
|
"original_file_name": original_file_name,
|
|
"row_count": row_count,
|
|
"template_type": template_type,
|
|
}
|
|
|
|
data = _attach_current_user(data)
|
|
job = self.post(url='/service/{}/contact-list'.format(service_id), data=data)
|
|
|
|
return job
|
|
|
|
def get_contact_lists(self, service_id):
|
|
return self.get(f'/service/{service_id}/contact-list')
|
|
|
|
def get_contact_list(self, *, service_id, contact_list_id):
|
|
return self.get(f'/service/{service_id}/contact-list/{contact_list_id}')
|
|
|
|
|
|
contact_list_api_client = ContactListApiClient()
|