所有ID的DRF分页响应附加列表

Question

我的网站上有“下一个对象”功能，但我使用分页。我想添加“id”字段，该字段包含过滤和分类到分页响应中的所有对象的id

我尝试的所有操作都只返回当前页面ID的列表

class StandardResultsSetPagination(PageNumberPagination):
    page_size = 20
    page_size_query_param = 'page_size'
    max_page_size = 20


    def get_paginated_response(self, data, list_of_ids):
        return Response(OrderedDict([
            ('count', self.page.paginator.count),
            ('next', self.get_next_link()),
            ('previous', self.get_previous_link()),
            ('results', data),
            ('ids', list_of_ids)
        ]))

    


class RealestateViewSet(ModelViewSet):
    ...
    pagination_class = StandardResultsSetPagination

    def get_paginated_response(self, data, list_of_ids):
        """
        Return a paginated style `Response` object for the given output data.
        """
        assert self.paginator is not None
        return self.paginator.get_paginated_response(data, list_of_ids)

    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, list(queryset.values_list('id', flat=True)))

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

这将返回一个包含20个ID的列表，而不是所有过滤对象ID

你知道怎么做吗

Answer 1

您可以通过提供ID，然后将子类附加到ModelViewSet.filter_backends，来子类BaseFilterBackend来过滤查询集（来自get_queryset()）

在你的问题中，你的ids来自哪里并不清楚（来自URL参数？例如http://your_url?param1=value1），但一个简单的例子可能是这样的：

from rest_framework.filters import BaseFilterBackend

class LimitQsetFilter(BaseFilterBackend):
    def filter_queryset(self, request, queryset, view):
        ids = [1,2,3,4]
        queryset = queryset.filter(pk__in=ids)
        return queryset

class RealestateViewSet(ModelViewSet):
    ...
    filter_backends = [LimitQsetFilter,]
    ...

然后，您将看到分页时对象实例的筛选列表