Django Rest Framework：在 ViewSet 上打开分页（如 ModelViewSet 分页）

Question

我有一个这样的 ViewSet 来列出用户的数据：

class Foo(viewsets.ViewSet):

    def list(self, request):
        queryset = User.objects.all()
        serializer = UserSerializer(queryset, many=True)
        return Response(serializer.data)

我想像 ModelViewSet 的默认分页一样打开分页：

{
    "count": 55,
    "next": "http://myUrl/?page=2",
    "previous": null,
    "results": [{...},{...},...,{...}]
}

The official doc 说：

只有在您使用通用视图或视图集时才会自动执行分页

...但是我的结果集根本没有分页。如何分页？

Answer 1

这种方式也适用于 Django Rest apiViews Api 的

from rest_framework.pagination import PageNumberPagination
class FooList(APIView):

    page_size = 10

    def get(self, request):
        foo = Foo.objects.all()
        paginator = PageNumberPagination()
        paginator.page_size = page_size
        result_page = paginator.paginate_queryset(foo, request)
        serializer = FooSerializer(result_page, many=True)
        return paginator.get_paginated_response(serializer.data)

Answer 2

在DRF中使用viewsets和list进行分页。

这里我处理了一个异常如果 page 是 empty 它将显示 empty 记录。

在设置定义页面大小时，这个页面大小是全局的并且被paginator_queryset在视图中使用。

REST_FRAMEWORK = { 'PAGE_SIZE': 10, }

在 view.py 中：

from rest_framework import mixins, viewsets

class SittingViewSet(viewsets.GenericViewSet,
    mixins.ListModelMixin):

    serializer_class = SittingSerializer
    queryset = Sitting.objects.all()
    serializer = serializer_class(queryset, many=True)

    def list(self, request, *args, **kwargs):
        queryset =self.filter_queryset(Sitting.objects.all().order_by('id'))

        page = request.GET.get('page')

        try: 
            page = self.paginate_queryset(queryset)
        except Exception as e:
            page = []
            data = page
            return Response({
                "status": status.HTTP_200_OK,
                "message": 'No more record.',
                "data" : data
                })

        if page is not None:
            serializer = self.get_serializer(page, many=True)
            data = serializer.data
            return self.get_paginated_response(data)

        # serializer = self.get_serializer(queryset, many=True)
        return Response({
            "status": status.HTTP_200_OK,
            "message": 'Sitting records.',
            "data" : data
        })

注意：如果您不使用Order_by，它将显示exception，因为此列表 给出无序列表。

Answer 3

如果您想为特定的ViewSet 分页，但不需要自定义页面大小，则this answer 的一个稍微简单的变体：

REST_FRAMEWORK = {
    'PAGE_SIZE': 100
}

class FooViewSet(viewsets.ModelViewSet):
    pagination_class = PageNumberPagination

Answer 4

对于那些使用 DRF 3.1 或更高版本的用户，他们正在更改处理分页的默认方式。详情请见http://www.django-rest-framework.org/topics/3.1-announcement/。

现在，如果您想为 ModelViewSet 启用分页，您可以通过在 settings.py 文件中进行设置来全局执行此操作：

REST_FRAMEWORK = {
    'DEFAULT_PAGINATION_CLASS': 'rest_framework.pagination.PageNumberPagination',
    'PAGE_SIZE': 100
}

或者，如果您只希望它用于一个 ModelViewSet，您可以手动为该视图集设置 pagination_class。

from rest_framework.pagination import PageNumberPagination

class StandardResultsSetPagination(PageNumberPagination):
    page_size = 100
    page_size_query_param = 'page_size'
    max_page_size = 1000

class FooViewSet(viewsets.ModelViewSet):
    pagination_class = StandardResultsSetPagination

这还允许您调整仅针对该视图集处理分页的方式。

DRF 3.1 还引入了可以使用的新类型的默认分页方案，例如 LimitOffset 和 Cursor。

Answer 5

尝试提供类变量

paginate_by = 10 #This will paginate by 10 results per page.

创建一个自定义 ViewSet，它只执行 list 操作，作为您目前的情况。

class ListModelViewSet(mixins.ListModelMixin, viewsets.GenericViewSet):
    pass

现在用这个定制的视图继承你的类Foo

class Foo(ListModelViewSet):

    paginate_by = 10

    def list(self, request):
        queryset = User.objects.all()
        serializer = UserSerializer(queryset, many=True)
        return Response(serializer.data)

这应该可以帮助您使分页正常工作。

Answer 6

只有在您使用泛型时才会自动执行分页 视图或视图集

第一个障碍是将文档翻译成英文。他们想要传达的是你想要一个通用的视图集。通用视图集从 generic ApiViews 扩展而来，它具有用于对查询集和响应进行分页的额外类方法。

另外，您提供了自己的list 方法，但默认分页过程实际上由mixin 处理：

class ListModelMixin(object):
    """
    List a queryset.
    """
    def list(self, request, *args, **kwargs):
        queryset = self.filter_queryset(self.get_queryset())

        page = self.paginate_queryset(queryset)
        if page is not None:
            serializer = self.get_serializer(page, many=True)
            return self.get_paginated_response(serializer.data)

        serializer = self.get_serializer(queryset, many=True)
        return Response(serializer.data)

简单的解决方案，使用框架代码：

class Foo(mixins.ListModelMixin, viewsets.GenericViewSet):
    queryset = User.objects.all()
    serializer = UserSerializer

更复杂的解决方案是，如果您需要自定义 list 方法，那么您应该按照您认为合适的方式编写它，但采用上述混合代码 sn-p 的样式。