Djano Pagination通用LISTCREATEAPIview中的条件query_set

时间:2016-04-22 09:28:07

标签: python pagination django-rest-framework django-class-based-views

我有一个URL中有一些query_params,我已经在这个URL上应用了分页。

urls.py

url(r'^users/(?P<pk>[0-9]+)/workouts/get/$',
        WorkoutList.as_view(serializer_class=WorkoutSerializer), name='list'),

views.py

class WorkoutList(generics.ListCreateAPIView):
    queryset = Workout.objects.all()
    serializer_class = WorkoutSerializer
    permission_classes = (UserPermissions,)

    def get_queryset(self):
        query_set = super(WorkoutList, self).get_queryset()
        query_params = self.request.QUERY_PARAMS.dict()
        try:
            date = string_to_date_convertor(query_params['date'])
        except KeyError:
            print 'Exception'
            return Response(status=status.HTTP_406_NOT_ACCEPTABLE)

        if 'date' in query_params:
            query_set = Workout.objects.filter(created__contains=date, user_id = self.kwargs['pk'])

        elif 'date' in query_params and 'exclude_app_install_time' in query_params:
            query_set = Workout.objects.filter(created__contains=date, time_reg = query_params['exclude_app_install_time'])    


        return query_set

现在我已经尝试除了阻止日期,如果日期参数不存在,只返回4xx Http状态代码。如果没有参数,那么只返回4xx状态码。

如果两个条件块都失败,则返回query_set并执行此query_set

  

queryset = Workout.objects.all()    我的锻炼表包含数百万条记录,我不想退回整个表格。那将是灾难性的。

所以我添加了另一个像这样的条件块

 elif query_param is None:
       query_set = None

简单来说,如果任何异常或条件块失败,那么只返回4xx。

1 个答案:

答案 0 :(得分:3)

当任何异常或条件阻止失败时,您可以使用QuerySet.objects.None() 返回EmptyQuerySet

# return empty queryset
queryset = Workout.objects.none()