如何自定义DjangoRestFramework中的分页大小

在Django Rest Framework中，可以通过自定义分页类来自定义分页大小。下面是一个示例，演示如何自定义分页大小为每页100个对象。

1. 首先，创建一个名为pagination.py的文件，用于存放自定义的分页类代码。

from rest_framework.pagination import PageNumberPagination

class CustomPagination(PageNumberPagination):
    page_size = 100  # 指定每页的对象数量
    page_size_query_param = 'page_size'  # 声明前端传递分页大小的查询参数名
    max_page_size = None  # 允许的最大分页大小，如果为None，则不限制大小

在上述代码中，CustomPagination类继承自DRF提供的PageNumberPagination，通过设置page_size属性来指定每页显示的对象数量为100。还通过page_size_query_param属性来声明前端传递分页大小的查询参数名为page_size。如果没有提供该参数，将使用默认的page_size值。

2. 在使用分页的视图中，将自定义的分页类应用到相应的视图中。

from .pagination import CustomPagination
from rest_framework.views import APIView
from rest_framework.response import Response

class MyView(APIView):
    pagination_class = CustomPagination  # 将自定义的分页类应用到视图中

    def get(self, request):
        # 通过分页类的属性来获取分页大小和页码
        page_size = self.pagination_class().get_page_size(request)
        page_number = self.pagination_class().get_page_number(request)

        # 执行自定义分页逻辑，这里只是一个示例
        queryset = MyModel.objects.all()
        paginated_queryset = self.pagination_class().paginate_queryset(queryset, request)

        # 返回分页结果
        serializer = MySerializer(paginated_queryset, many=True)
        return self.pagination_class().get_paginated_response(serializer.data)

在上述代码中，将自定义的分页类应用到视图中的pagination_class属性中。通过调用self.pagination_class().get_page_size(request)方法和self.pagination_class().get_page_number(request)方法，可以获取到前端传递的分页大小和页码。

然后，执行自定义的分页逻辑，这里使用DRF提供的paginate_queryset方法对查询集进行分页。最后，将分页结果进行序列化，并使用self.pagination_class().get_paginated_response(serializer.data)方法返回分页结果。

以上就是自定义分页大小的简单示例。在实际应用中，你可以根据自己的需求来自定义更复杂的分页逻辑，包括根据不同用户的角色或权限来控制分页大小等。