Allow 'get_page' method for overriding

Signed-off-by: Paul Wayper <paulway@redhat.com>
This commit is contained in:
Paul Wayper 2020-11-04 12:02:10 +11:00
parent 56e4508123
commit 8ed5b1e70e

View File

@ -198,7 +198,7 @@ class PageNumberPagination(BasePagination):
return None return None
paginator = self.django_paginator_class(queryset, page_size) paginator = self.django_paginator_class(queryset, page_size)
page_number = request.query_params.get(self.page_query_param, 1) page_number = self.get_page(request)
if page_number in self.last_page_strings: if page_number in self.last_page_strings:
page_number = paginator.num_pages page_number = paginator.num_pages
@ -264,6 +264,12 @@ class PageNumberPagination(BasePagination):
return self.page_size return self.page_size
def get_page(self, request):
return _positive_int(
request.query_params.get(self.page_query_param, 1),
strict=True,
)
def get_next_link(self): def get_next_link(self):
if not self.page.has_next(): if not self.page.has_next():
return None return None