mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-02-24 20:31:19 -05:00
34 lines
1.1 KiB
Python
34 lines
1.1 KiB
Python
from notifications_python_client.base import BaseAPIClient
|
|
from app.notify_client import _attach_current_user
|
|
|
|
|
|
class ProviderClient(BaseAPIClient):
|
|
def __init__(self, base_url=None, client_id=None, secret=None):
|
|
super(self.__class__, self).__init__(
|
|
base_url=base_url or 'base_url',
|
|
client_id=client_id or 'client_id',
|
|
secret=secret or 'secret'
|
|
)
|
|
|
|
def init_app(self, app):
|
|
self.base_url = app.config['API_HOST_NAME']
|
|
self.client_id = app.config['ADMIN_CLIENT_USER_NAME']
|
|
self.secret = app.config['ADMIN_CLIENT_SECRET']
|
|
|
|
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 update_provider(self, provider_id, priority):
|
|
data = {
|
|
"priority": priority
|
|
}
|
|
_attach_current_user(data)
|
|
return self.post(url='/provider-details/{}'.format(provider_id), data=data)
|