python - djangorest框架中django过滤器后端的条件？

Question

我在几个字段上使用 Django 过滤器后端，效果非常好。但我想在获得特定条件时过滤字段，例如，如果 user_type 是基本获取过滤器查询，否则从模型中获取所有对象。我的过滤器代码在这里:

http_method_names = ['get']
serializer_class = SearchSerializer
pagination_class = LargeResultsSetPagination


filter_backends = (DjangoFilterBackend,)
filter_fields = ('property_zipcode', 'property_state', 'property_county',)

我需要的是:

def get_queryset(self):
    if self.request.query_params.get('basic',None):
       filter_backends = (DjangoFilterBackend,)
       filter_fields = ('property_zipcode', 'property_state', 'property_county',)
       return filtered_query # I want to return filter query from here.

    queryset = property.objects.all();
    return queryset

我是 Django Rest 框架和 Django 过滤器后端的新手。如果有人尝试做这件事，请帮助我。

Answer 1

这就是您所需要的:

class YourClass(ModelViewSet):
    serializer_class = SearchSerializer
    pagination_class = LargeResultsSetPagination
    filter_backends = (DjangoFilterBackend,)
    filter_fields = ('property_zipcode', 'property_state', 'property_county',)

    def filter_queryset(self, queryset):
        if self.request.query_params.get('basic',None):
            queryset = super(YourClass,self).filter_queryset(self.get_queryset())
        else:
            queryset=self.get_queryset()
        return queryset

覆盖get_queryset不会按您的期望工作，因为filter_queryset方法是在get_queryset方法之后由框架调用的，DRF get_queryset 然后通过 filter_queryset 方法对其进行过滤。