不能找出正确的方式来实现Django过滤器

Question

我的问题是我必须查询数据库的用户的某些字段是des,gender,min_age,max_age。当我使用多个值时，它使用filter_queryset工作正常。然后我使用多个值gender其工作，但是当我没有在我的网址中包括gender代码中断基本上是因为split不起作用。 所以我认为使用多个if/else将有不同queryset.filter(des__in=[], gender__in=[])或gender不可用然后查询 - >queryset.filter(des__in=[])。我在这里错过了很多东西。请帮助

API - /v1/users/query_users?des=1,2&gender=Male,Female&min_age=2&max_age=4

class ProductFilter(django_filters.rest_framework.FilterSet): 
    min_age = django_filters.DateRangeFilter(name="dob", lookup_expr="gte") 
    max_age = django_filters.DateRangeFilter(name="dob", lookup_expr="lte") 
    class Meta: 
     model = UserProfile 
     fields = ['des', 'gender', 'min_age', 'max_age'] 


class QueryUserGroup(generics.ListAPIView): 
    queryset = UserProfile.objects.all() 
    serializer_class = UserProfileSerializer 
    filter_backends = (django_filters.rest_framework.DjangoFilterBackend,) 
    filter_class = ProductFilter 
    permission_classes = [permissions.IsAuthenticatedOrReadOnly] 

    def filter_queryset(self, queryset): 
     des_list = self.request.GET.getlist("des") 
     gender_list = self.request.GET.getlist("gender") 
     des_ids = des_list[0].split(',') 
     gender_ids = gender_list[0].split(',') 
     return queryset.filter(des__in=des_ids, gender__in=gender_ids) 

    def list(self, request, *args, **kwargs): 
     queryset = self.filter_queryset(self.get_queryset()) 
     serializer = UserProfileSerializer(queryset, many=True) 
     return JSONResponse({'data': serializer.data}, status=HTTP_200_OK) 

Answer 1

你应该重新安排你的代码，所以你只拆分字符串和过滤的queryset如果该字段的查询字符串。

由于您只访问列表中的第一个项目，您不妨使用get而不是getlist。

def filter_queryset(self, queryset): 
    des = self.request.GET.get("des") 
    if des is not None: 
     des_ids = des.split(',') 
     queryset = queryset.filter(des__in=des_ids) 

    gender = self.request.GET.get("gender") 
    if gender is not None: 
     gender_ids = gender.split(',') 
     queryset = queryset.filter(gender__in=gender_ids) 

    return queryset 

如果你有两个以上的领域，你可以通过字段名的列表循环削减重复的代码：

def filter_queryset(self, queryset): 
    for fieldname in ("des", "gender"): 
     value = self.request.GET.get(fieldname) 
     if value is not None: 
      ids = value.split(',') 
      queryset = queryset.filter(**{'%s__in' % fieldname: ids}) 

    return queryset