mirror of
https://github.com/GSA/notifications-admin.git
synced 2025-12-15 01:23:25 -05:00
415e1a401a
Because we’re setting the API key and service ID after calling the `__init__` method of the client it wasn’t doing the thing where it splits the combined key into the two individual UUIDs. So we still need to set them directly, individually on the client.
38 lines
1.3 KiB
Python
38 lines
1.3 KiB
Python
from app.notify_client import _attach_current_user, NotifyAdminAPIClient
|
|
|
|
# must match key types in notifications-api/app/models.py
|
|
KEY_TYPE_NORMAL = 'normal'
|
|
KEY_TYPE_TEAM = 'team'
|
|
KEY_TYPE_TEST = 'test'
|
|
|
|
|
|
class ApiKeyApiClient(NotifyAdminAPIClient):
|
|
def __init__(self):
|
|
super().__init__("a" * 73, "b")
|
|
|
|
def init_app(self, app):
|
|
self.base_url = app.config['API_HOST_NAME']
|
|
self.service_id = app.config['ADMIN_CLIENT_USER_NAME']
|
|
self.api_key = app.config['ADMIN_CLIENT_SECRET']
|
|
|
|
def get_api_keys(self, service_id, key_id=None):
|
|
if key_id:
|
|
return self.get(url='/service/{}/api-keys/{}'.format(service_id, key_id))
|
|
else:
|
|
return self.get(url='/service/{}/api-keys'.format(service_id))
|
|
|
|
def create_api_key(self, service_id, key_name, key_type):
|
|
data = {
|
|
'name': key_name,
|
|
'key_type': key_type
|
|
}
|
|
data = _attach_current_user(data)
|
|
key = self.post(url='/service/{}/api-key'.format(service_id), data=data)
|
|
return key['data']
|
|
|
|
def revoke_api_key(self, service_id, key_id):
|
|
data = _attach_current_user({})
|
|
return self.post(
|
|
url='/service/{0}/api-key/revoke/{1}'.format(service_id, key_id),
|
|
data=data)
|