encode/django-rest-framework
1
1
Fork 0
mirror of https://github.com/encode/django-rest-framework.git synced 2024-11-28 04:24:00 +03:00
Code Issues Packages Projects Releases Wiki Activity

feat: added last and first page links to PageNumberPagination

Browse Source
This commit is contained in:
Wellington Adriano Nicoletti de Abreu 2023-07-01 20:38:21 -03:00
parent 8dd4250d02
commit 81439bb646
1 changed files with 20 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
rest_framework/pagination.py
View File

@ -231,8 +231,10 @@ class PageNumberPagination(BasePagination):
def get_paginated_response(self, data): def get_paginated_response(self, data):
return Response({ return Response({
'count': self.page.paginator.count, 'count': self.page.paginator.count,
'last': self.get_last_link(),
'next': self.get_next_link(), 'next': self.get_next_link(),
'previous': self.get_previous_link(), 'previous': self.get_previous_link(),
'first': self.get_first_link(),
'results': data, 'results': data,
}) })
@ -272,6 +274,17 @@ class PageNumberPagination(BasePagination):
) )
return self.page_size return self.page_size
def get_last_link(self):
if not self.page.has_next():
return None
next_page_number = self.page.next_page_number()
last_page_number = self.page.paginator.num_pages
# if next and last are the same it does not need to build the last url
if next_page_number == last_page_number:
return None
url = self.request.build_absolute_uri()
return replace_query_param(url, self.page_query_param, last_page_number)
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
@ -288,6 +301,13 @@ class PageNumberPagination(BasePagination):
return remove_query_param(url, self.page_query_param) return remove_query_param(url, self.page_query_param)
return replace_query_param(url, self.page_query_param, page_number) return replace_query_param(url, self.page_query_param, page_number)
def get_first_link(self):
# if the current number is greater than 2, then it makes sense to build the first url
if self.page.number <= 2:
return None
url = self.request.build_absolute_uri()
return replace_query_param(url, self.page_query_param, 1)
def get_html_context(self): def get_html_context(self):
base_url = self.request.build_absolute_uri() base_url = self.request.build_absolute_uri()