Use the local APIClient rather than the one from the python-api-client

- ensures that all API calls set the request ID when talking to the API.
This commit is contained in:
Martyn Inglis
2016-11-30 17:01:44 +00:00
parent 655e6892ad
commit 08dc8fb13d
11 changed files with 24 additions and 29 deletions

View File

@@ -1,6 +1,4 @@
from notifications_python_client.base import BaseAPIClient
from app.notify_client import _attach_current_user
from app.notify_client import _attach_current_user, NotifyAdminAPIClient
# must match key types in notifications-api/app/models.py
KEY_TYPE_NORMAL = 'normal'
@@ -8,7 +6,7 @@ KEY_TYPE_TEAM = 'team'
KEY_TYPE_TEST = 'test'
class ApiKeyApiClient(BaseAPIClient):
class ApiKeyApiClient(NotifyAdminAPIClient):
def __init__(self):
super().__init__("a", "b", "c")

View File

@@ -1,7 +1,7 @@
from notifications_python_client.base import BaseAPIClient
from app.notify_client import NotifyAdminAPIClient
class EventsApiClient(BaseAPIClient):
class EventsApiClient(NotifyAdminAPIClient):
def __init__(self):
super().__init__("a", "b", "c")

View File

@@ -1,11 +1,9 @@
from notifications_python_client.base import BaseAPIClient
from app.notify_client import _attach_current_user
from app.notify_client import _attach_current_user, NotifyAdminAPIClient
from app.notify_client.models import InvitedUser
class InviteApiClient(BaseAPIClient):
class InviteApiClient(NotifyAdminAPIClient):
def __init__(self):
super().__init__("a", "b", "c")

View File

@@ -1,10 +1,9 @@
from collections import defaultdict
from notifications_python_client.base import BaseAPIClient
from app.notify_client import _attach_current_user
from app.notify_client import _attach_current_user, NotifyAdminAPIClient
class JobApiClient(BaseAPIClient):
class JobApiClient(NotifyAdminAPIClient):
JOB_STATUSES = {
'scheduled',

View File

@@ -1,7 +1,7 @@
from notifications_python_client.base import BaseAPIClient
from app.notify_client import NotifyAdminAPIClient
class NotificationApiClient(BaseAPIClient):
class NotificationApiClient(NotifyAdminAPIClient):
def __init__(self):
super().__init__("a", "b", "c")

View File

@@ -1,7 +1,7 @@
from notifications_python_client.base import BaseAPIClient
from app.notify_client import NotifyAdminAPIClient
class OrganisationsClient(BaseAPIClient):
class OrganisationsClient(NotifyAdminAPIClient):
def __init__(self):
super().__init__("a", "b", "c")

View File

@@ -1,8 +1,8 @@
from notifications_python_client.base import BaseAPIClient
from app.notify_client import _attach_current_user
from app.notify_client import _attach_current_user, NotifyAdminAPIClient
class ProviderClient(BaseAPIClient):
class ProviderClient(NotifyAdminAPIClient):
def __init__(self):
super().__init__("a", "b", "c")

View File

@@ -1,11 +1,10 @@
from __future__ import unicode_literals
from flask import url_for
from notifications_python_client.base import BaseAPIClient
from app.utils import BrowsableItem
from app.notify_client import _attach_current_user
from app.notify_client import _attach_current_user, NotifyAdminAPIClient
class ServiceAPIClient(BaseAPIClient):
class ServiceAPIClient(NotifyAdminAPIClient):
# Fudge assert in the super __init__ so
# we can set those variables later.
def __init__(self):

View File

@@ -1,7 +1,8 @@
from notifications_python_client.base import BaseAPIClient
from app.notify_client import NotifyAdminAPIClient
class StatusApiClient(BaseAPIClient):
class StatusApiClient(NotifyAdminAPIClient):
def __init__(self):
super().__init__("a", "b", "c")

View File

@@ -1,7 +1,7 @@
from notifications_python_client.base import BaseAPIClient
from app.notify_client import NotifyAdminAPIClient
class TemplateStatisticsApiClient(BaseAPIClient):
class TemplateStatisticsApiClient(NotifyAdminAPIClient):
def __init__(self):
super().__init__("a", "b", "c")

View File

@@ -1,6 +1,6 @@
from notifications_python_client.notifications import BaseAPIClient
from notifications_python_client.errors import HTTPError
from app.notify_client import NotifyAdminAPIClient
from app.notify_client.models import User
ALLOWED_ATTRIBUTES = {
@@ -10,7 +10,7 @@ ALLOWED_ATTRIBUTES = {
}
class UserApiClient(BaseAPIClient):
class UserApiClient(NotifyAdminAPIClient):
def __init__(self):
super().__init__("a", "b", "c")