过滤器在 Django REST 框架中无法正常工作

Question

这是我的 TestViewSet。

class TestViewSet(ListAPIView, CreateAPIView, UpdateAPIView):
permission_classes = [
    IsAuthenticated,
]
pagination_class = CustomPagination
serializer_class = TestSerializer
model = Test
create_class = CreateTestSerializer
filter_backends = [DjangoFilterBackend]
filterset_fields = ['department__name', 'code', 'billing_category__id', 'billing_category__name']

def get_queryset(self):
    name_str = self.request.query_params.get("name")
    department_id = self.request.query_params.get("department_id")
    pk = self.kwargs.get("id")
    self.pagination_class.page_size = page_size
    if name_str is not None:
        return self.model.objects.filter(
            Q(name__icontains=name_str)
            | Q(name__iexact=name_str)
            | Q(name__istartswith = name_str)
            )
    elif pk is not None:
        return self.model.objects.filter(id=pk)
    elif department_id is not None:
        return self.model.objects.filter(department_id=department_id)
    else:
        return self.model.objects.all()

name 是 CharField，department 是 Test Model 中的 ForeignKey。

当我传递这个网址时 - http://127.0.0.1:8000/api/v1/package/test/?name=fun&department_id=7 它忽略了 deparment_id。我只是不知道为什么。单独两者都工作正常。我想知道为什么它忽略 deparment_id ？谢谢！！

Answer 1

if-elif-else只会输入一次，你需要把它们拆开，像这样：

bool(Q()) 为假

def get_queryset(self):
    name_str = self.request.query_params.get("name")
    department_id = self.request.query_params.get("department_id")
    pk = self.kwargs.get("id")
    self.pagination_class.page_size = page_size

    lookup = Q()
    if name_str is not None:
        lookup |= Q(name__icontains=name_str)
            | Q(name__iexact=name_str)
            | Q(name__istartswith = name_str)
    if department_id is not None:
        lookup &= Q(department_id=department_id)
        # lookup |= Q(department_id=department_id)

    if lookup:
        return self.model.objects.filter(lookup)
    elif pk is not None:
        return self.model.objects.filter(id=pk)
    else:
        return self.model.objects.all()