mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-02-23 11:51:05 -05:00
5ddbe80ea9
The Notify API client changed in version 4 to take two arguments, not three (service ID was removed in favour of the combined API key). This gets a bit gnarly because the API key has to be at least a certain length so it can be substringed internally.
40 lines
1.3 KiB
Python
40 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.api_key = '{}-{}'.format(
|
|
app.config['ADMIN_CLIENT_USER_NAME'],
|
|
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)
|