mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-02-06 11:23:48 -05:00
fc1ca3ab2f
The responses we get to paginated queries from the API are fairly consistent, so we should be able to reuse the code that takes JSON from the API and turns it into Python objects. This commits factors out that code so that it is reusable (by inheriting from it).
63 lines
1.5 KiB
Python
63 lines
1.5 KiB
Python
from abc import abstractmethod
|
|
|
|
from flask import abort
|
|
from notifications_utils.serialised_model import (
|
|
SerialisedModel,
|
|
SerialisedModelCollection,
|
|
)
|
|
|
|
|
|
class JSONModel(SerialisedModel):
|
|
|
|
def __init__(self, _dict):
|
|
# in the case of a bad request _dict may be `None`
|
|
self._dict = _dict or {}
|
|
for property in self.ALLOWED_PROPERTIES:
|
|
if property in self._dict:
|
|
setattr(self, property, self._dict[property])
|
|
|
|
def __bool__(self):
|
|
return self._dict != {}
|
|
|
|
def __hash__(self):
|
|
return hash(self.id)
|
|
|
|
def __eq__(self, other):
|
|
return self.id == other.id
|
|
|
|
def _get_by_id(self, things, id):
|
|
try:
|
|
return next(thing for thing in things if thing['id'] == str(id))
|
|
except StopIteration:
|
|
abort(404)
|
|
|
|
|
|
class ModelList(SerialisedModelCollection):
|
|
|
|
@property
|
|
@abstractmethod
|
|
def client_method(self):
|
|
pass
|
|
|
|
def __init__(self, *args):
|
|
self.items = self.client_method(*args)
|
|
|
|
|
|
class PaginatedModelList(ModelList):
|
|
|
|
response_key = 'data'
|
|
|
|
def __init__(self, *args, page=None, **kwargs):
|
|
try:
|
|
self.current_page = int(page)
|
|
except TypeError:
|
|
self.current_page = 1
|
|
response = self.client_method(
|
|
*args,
|
|
**kwargs,
|
|
page=self.current_page,
|
|
)
|
|
self.items = response[self.response_key]
|
|
self.prev_page = response.get('links', {}).get('prev', None)
|
|
self.next_page = response.get('links', {}).get('next', None)
|