mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-03-01 14:51:04 -05:00
9e798506c5
This avoids the annoying problem where you can’t import a client unless the app has already been initialised.
33 lines
863 B
Python
33 lines
863 B
Python
|
|
from app.notify_client import NotifyAdminAPIClient, _attach_current_user
|
|
|
|
|
|
class ProviderClient(NotifyAdminAPIClient):
|
|
def __init__(self):
|
|
super().__init__("a" * 73, "b")
|
|
|
|
def get_all_providers(self):
|
|
return self.get(
|
|
url='/provider-details'
|
|
)
|
|
|
|
def get_provider_by_id(self, provider_id):
|
|
return self.get(
|
|
url='/provider-details/{}'.format(provider_id)
|
|
)
|
|
|
|
def get_provider_versions(self, provider_id):
|
|
return self.get(
|
|
url='/provider-details/{}/versions'.format(provider_id)
|
|
)
|
|
|
|
def update_provider(self, provider_id, priority):
|
|
data = {
|
|
"priority": priority
|
|
}
|
|
data = _attach_current_user(data)
|
|
return self.post(url='/provider-details/{}'.format(provider_id), data=data)
|
|
|
|
|
|
provider_client = ProviderClient()
|