mirror of
https://github.com/GSA/notifications-admin.git
synced 2025-12-13 16:43:54 -05:00
27 lines
888 B
Python
27 lines
888 B
Python
from app.notify_client import NotifyAdminAPIClient, _attach_current_user
|
|
|
|
# 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 get_api_keys(self, service_id):
|
|
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
|
|
)
|
|
|
|
|
|
api_key_api_client = ApiKeyApiClient()
|