使用Django-Filters创建自定义字段

Question

我有一个模型，如下所示。该模型将属性类型作为外键。

class YachtGeneralInfo(models.Model):
    type = models.ForeignKey(
        YachtTypes,
        related_name="yachts_in_type",
        on_delete=models.PROTECT,
        blank=False,
        null=False,
    )
    ....
    ....

我编写了这样的视图类 -

class YachtGeneralInfoView(ListAPIView):

    pagination_class = PageNumberPagination
    serializer_class = YachtGeneralInfoSerializer
    filter_backends = [OrderingFilter, SearchFilter, DjangoFilterBackend]
    filterset_fields = [
        "status",
        "is_professional",
        "chartered_with__id",
        "harbour__id",
        "harbour__city__id",
        "model__id",
    ]
    search_fields = ["name", "company_name", "website", "owner__name"]

我想为字段type添加另一个过滤器。但是，我想为此查询参数提供多个值，例如[1,2,3]。

如果我直接使用QuerySet进行操作，那就是这样 -

queryset = YachtGeneralInfo.objects.filter(type__in=[1,2,3])

有没有办法编写自定义过滤器集字段来完成此操作，而不是直接在QuerySet中将其进行？

Answer 1

您可以为类型ID查找创建自定义过滤器集。

class YachtTypeFilterSet(django_filters.FilterSet):
    type_in = django_filters.CharFilter(
        method='filter_guest_level', field_name='guest_level')

    def filter_type_in(self, queryset, field_name, value):
        if value != "":
            id_array = json.loads(value)
            if isinstance(id_array, list):
                return queryset.filter(type__id__in=id_array)
        return queryset

然后，您可以在视图中设置该类。

class YachtGeneralInfoView(ListAPIView):
    ...
    filterset_class = YachtTypeFilterSet