如何为视图集使用限制偏移分页

时间:2017-12-12 09:13:11

标签: django django-rest-framework drf-queryset

Views.py

class CountryViewSet(viewsets.ViewSet):   
    serializer_class = CountrySerializer
    pagination_class = LimitOffsetPagination
    def list(self,request):
        try:
            country_data = Country.objects.all()
            country_serializer = CountrySerializer(country_data,many=True)
            return Response(            
                data = country_serializer.data,
                content_type='application/json',            
                )
        except Exception as ex:
            return Response(
                data={'error': str(ex)},
                content_type='application/json',
                status=status.HTTP_400_BAD_REQUEST
                )

Settings.py 我添加了

'DEFAULT_PAGINATION_CLASS': 'rest_framework.pagination.LimitOffsetPagination',

在我的urls.py

router = routers.DefaultRouter(trailing_slash=False)

router.register(r'country', CountryViewSet, base_name='country')
urlpatterns = [
    url(r'^', include(router.urls)),
]

当我尝试使用此网址http://192.168.2.66:8001/v1/voucher/country时,它会返回所有数据。

但是当我尝试使用此网址http://192.168.2.66:8001/v1/voucher/country/?limit=2&offset=2

但它返回404错误。 我是django的新手。请帮助我:)

1 个答案:

答案 0 :(得分:2)

使用ModelViewSet而不是ViewSet。同时删除列表功能,它会自动发送响应。

from rest_framework.pagination import LimitOffsetPagination

class CountryViewSet(viewsets.ModelViewSet):
    """
    A simple ViewSet for viewing and editing country.
    """ 
    queryset = Country.objects.all()
    serializer_class = CountrySerializer
    pagination_class = LimitOffsetPagination
  

ModelViewSet类提供的操作是.list(),   .retrieve(),. create(),. update(),. partial_update()和.destroy()。

<强>更新

在你的settings.py

REST_FRAMEWORK = {
    'PAGE_SIZE': 10,
    # 'DEFAULT_PAGINATION_CLASS': 'rest_framework.pagination.LimitOffsetPagination',
}

更新2

或者,您可以使用paginate_querysetget_paginated_response

def list(self,request):
    country_data = Country.objects.all()

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

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

<强>参考:   marking-extra-actions-for-routing