mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-05-29 10:30:20 -04:00
acfe8092fc
Currently requests to the API made from the admin app are going from PaaS admin app to the nginx router ELB, which then routes them back to the api app on PaaS. This makes sense for external requests, but for requests made from the admin app we could skip nginx and go directly to the api PaaS host, which should reduce load on the nginx instances and potentially reduce latency of the api requests. API apps on PaaS are checking the X-Custom-Forwarder header (which is set by nginx on proxy_pass requests) to only allow requests going through the proxy. This adds the custom header to the API client requests, so that they can pass that header check without going through nginx.
56 lines
1.9 KiB
Python
56 lines
1.9 KiB
Python
|
|
from app.notify_client import NotifyAdminAPIClient, _attach_current_user
|
|
from app.notify_client.models import InvitedUser
|
|
|
|
|
|
class InviteApiClient(NotifyAdminAPIClient):
|
|
def __init__(self):
|
|
super().__init__("a" * 73, "b")
|
|
|
|
def init_app(self, app):
|
|
super().init_app(app)
|
|
|
|
self.admin_url = app.config['ADMIN_BASE_URL']
|
|
|
|
def create_invite(self, invite_from_id, service_id, email_address, permissions, auth_type):
|
|
data = {
|
|
'service': str(service_id),
|
|
'email_address': email_address,
|
|
'from_user': invite_from_id,
|
|
'permissions': permissions,
|
|
'auth_type': auth_type,
|
|
'invite_link_host': self.admin_url,
|
|
}
|
|
data = _attach_current_user(data)
|
|
resp = self.post(url='/service/{}/invite'.format(service_id), data=data)
|
|
return InvitedUser(**resp['data'])
|
|
|
|
def get_invites_for_service(self, service_id):
|
|
endpoint = '/service/{}/invite'.format(service_id)
|
|
resp = self.get(endpoint)
|
|
invites = resp['data']
|
|
invited_users = self._get_invited_users(invites)
|
|
return invited_users
|
|
|
|
def check_token(self, token):
|
|
resp = self.get(url='/invite/service/{}'.format(token))
|
|
return InvitedUser(**resp['data'])
|
|
|
|
def cancel_invited_user(self, service_id, invited_user_id):
|
|
data = {'status': 'cancelled'}
|
|
data = _attach_current_user(data)
|
|
self.post(url='/service/{0}/invite/{1}'.format(service_id, invited_user_id),
|
|
data=data)
|
|
|
|
def accept_invite(self, service_id, invited_user_id):
|
|
data = {'status': 'accepted'}
|
|
self.post(url='/service/{0}/invite/{1}'.format(service_id, invited_user_id),
|
|
data=data)
|
|
|
|
def _get_invited_users(self, invites):
|
|
invited_users = []
|
|
for invite in invites:
|
|
invited_user = InvitedUser(**invite)
|
|
invited_users.append(invited_user)
|
|
return invited_users
|