DRF 返回全部数据并保留 count 字段

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20

from rest_framework.pagination import BasePagination
from rest_framework.response import Response
from collections import OrderedDict


class TotalPagination(BasePagination):  
    """  
    不分页，显示总数  
    """  
    def paginate_queryset(self, queryset, request, view=None):  
        """  
        返回所有  
        """
        return list(queryset[:])  
  
    def get_paginated_response(self, data):  
        return Response(OrderedDict([  
            ('count', len(data)),  
            ('results', data)  
        ]))