如何在Django REST Framework通用API视图中实现page_size=max？

没问题！我来教你怎么在DRF的通用ListAPIView里实现?page_size=max的功能——毕竟drf-extensions的这个特性确实只支持Viewsets，咱们自己动手扩展分页类就能搞定，步骤很清晰：

1. 自定义分页类 #

首先创建一个继承自DRF默认分页类（比如PageNumberPagination）的自定义分页类，重写get_page_size方法来处理page_size=max的请求：

from rest_framework.pagination import PageNumberPagination

class MaxPageSizePagination(PageNumberPagination):
    def get_page_size(self, request):
        # 获取请求中的page_size参数
        page_size_param = request.query_params.get('page_size')
        
        # 判断参数是否为"max"（不区分大小写）
        if page_size_param and page_size_param.lower() == 'max':
            # 两种可选实现方式：
            # 方式1：返回查询集的总数量，精准返回所有数据
            return self.queryset.count()
            # 方式2：返回一个足够大的固定值（比如10^9），适合不想查询总数量的场景
            # return 10**9
        
        # 其他情况使用默认的分页逻辑
        return super().get_page_size(request)

注意：如果选择方式1，self.queryset会在视图调用分页类时自动赋值，无需额外处理；如果担心查询总数量的性能开销，方式2会更高效。

2. 在ListAPIView中绑定自定义分页类 #

接下来在你的ListAPIView子类中，指定pagination_class为刚才创建的自定义分页类即可：

from rest_framework.generics import ListAPIView
from .models import YourTargetModel
from .serializers import YourModelSerializer

class YourModelListView(ListAPIView):
    queryset = YourTargetModel.objects.all()
    serializer_class = YourModelSerializer
    # 绑定自定义分页类
    pagination_class = MaxPageSizePagination
    # 可选：设置默认的每页数量
    page_size = 20

3. 额外优化（可选） #

如果担心返回超大数据集导致性能问题，可以给max模式加个上限，比如限制最多返回10000条数据：

def get_page_size(self, request):
    page_size_param = request.query_params.get('page_size')
    if page_size_param and page_size_param.lower() == 'max':
        # 取查询集总数和上限值的较小者
        return min(self.queryset.count(), 10000)
    return super().get_page_size(request)

现在你就可以通过请求/your-api-endpoint/?page_size=max来一次性获取所有数据了，效果和drf-extensions在Viewsets里的表现完全一致~

内容的提问来源于stack exchange，提问作者iamroot