mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-05-26 08:09:51 -04:00
38 lines
1.1 KiB
Python
38 lines
1.1 KiB
Python
from app.notify_client import NotifyAdminAPIClient
|
|
|
|
|
|
class OrganisationsClient(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_organisation(self, id):
|
|
return self.get(url='/organisation/{}'.format(id))
|
|
|
|
def get_organisations(self):
|
|
return self.get(url='/organisation')['organisations']
|
|
|
|
def get_letter_organisations(self):
|
|
return self.get(url='/dvla_organisations')
|
|
|
|
def create_organisation(self, logo, name, colour):
|
|
data = {
|
|
"logo": logo,
|
|
"name": name,
|
|
"colour": colour
|
|
}
|
|
return self.post(url="/organisation", data=data)
|
|
|
|
def update_organisation(self, org_id, logo, name, colour):
|
|
data = {
|
|
"logo": logo,
|
|
"name": name,
|
|
"colour": colour
|
|
}
|
|
return self.post(url="/organisation/{}".format(org_id), data=data)
|