mirrors/wagtail
mirrors/wagtail
0
0
Fork 0
mirror of https://github.com/wagtail/wagtail.git synced 2024-12-01 11:41:20 +01:00
Code Issues Releases Wiki Activity

Refactor away .get_fields

Browse Source
This commit is contained in:
Tom Christie 2015-07-21 13:29:25 +01:00
parent f892c09308
commit 297f64509a
1 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
wagtail/contrib/wagtailapi/endpoints.py
View File

@ -32,6 +32,7 @@ class BaseAPIEndpoint(GenericViewSet):
pagination_class = WagtailPagination
serializer_class = WagtailSerializer
filter_classes = []
queryset = None # Set on subclasses or implement `get_queryset()`.
known_query_parameters = frozenset([
'limit',
@ -41,6 +42,7 @@ class BaseAPIEndpoint(GenericViewSet):
'search',
])
extra_api_fields = []
name = None # Set on subclass.
def listing_view(self, request):
queryset = self.get_queryset()
@ -88,26 +90,24 @@ class BaseAPIEndpoint(GenericViewSet):
if unknown_parameters:
raise BadRequestError("query parameter is not an operation or a recognised field: %s" % ', '.join(sorted(unknown_parameters)))
def get_fields(self, request):
"""
Return the set of fields that should be returned in the output
representation for listing views.
"""
if 'fields' in request.GET:
return set(request.GET['fields'].split(','))
return {'title'}
def get_serializer_context(self):
"""
The serialization context differs between listing and detail views.
"""
request = self.request
if self.action == 'listing_view':
if 'fields' in request.GET:
fields = set(request.GET['fields'].split(','))
else:
fields = {'title'}
return {
'request': request,
'view': self,
'fields': self.get_fields(request)
'fields': fields
}
return {
'request': request,
'view': self,