Django REST分页相对URL而不是ListAPIView中的绝对URL

Question

Django Rest Framework generic.ListAPIView 将 next&prev 参数设置为绝对 URL。但我需要将它们设置为相对 URL。

我需要返回 JSON 如下

# {'count': 4, 'next': '?page=2', 'previous': None, 'results': [u'john', u'paul']}

不是这样

# {'count': 4, 'next': 'http://testserver/foobar?page=2', 'previous': None, 'results': [u'john', u'paul']}

Answer 1

这是@iankit建议的实现:

from rest_framework.utils.urls import remove_query_param, replace_query_param
class LeadListPagination(PageNumberPagination):
    page_size = 15

    def get_next_link(self):
        if not self.page.has_next():
            return None
        url = self.request.get_full_path()
        page_number = self.page.next_page_number()
        return replace_query_param(url, self.page_query_param, page_number)

    def get_previous_link(self):
        if not self.page.has_previous():
            return None
        url = self.request.get_full_path()
        page_number = self.page.previous_page_number()
        if page_number == 1:
            return remove_query_param(url, self.page_query_param)
        return replace_query_param(url, self.page_query_param, page_number)