Update pagination.py

Allow User to add extra fields to data before sending it to HTTP Response
This commit is contained in:
Payam 2017-06-17 09:51:53 +01:00 committed by GitHub
parent b069b0d33a
commit dd4fdcc418

View File

@ -225,13 +225,16 @@ class PageNumberPagination(BasePagination):
self.request = request self.request = request
return list(self.page) return list(self.page)
def get_paginated_response(self, data): def get_paginated_dictionary(self, data):
return Response(OrderedDict([ return OrderedDict([
('count', self.page.paginator.count), ('count', self.page.paginator.count),
('next', self.get_next_link()), ('next', self.get_next_link()),
('previous', self.get_previous_link()), ('previous', self.get_previous_link()),
('results', data) ('results', data)
])) ])
def get_paginated_response(self, data, extra_fields={}):
return Response(self.get_paginated_dictionary(data).update(extra_fields))
def get_page_size(self, request): def get_page_size(self, request):
if self.page_size_query_param: if self.page_size_query_param: