通过 SerializerMethodField 过滤 django REST 结果

Question

如何根据来自序列化程序的计算字段过滤结果？我试着把它当作任何其他领域，但 django 不喜欢它。

序列化器

class ImageSerializer(serializers.ModelSerializer):
    is_annotated = serializers.SerializerMethodField('has_annotation')

    class Meta:
        model = Image
        fields = '__all__'

    @staticmethod
    def has_annotation(image):
        return image.annotation_set.count() > 0

查看

class ImageViewSet(viewsets.ModelViewSet):
    serializer_class = ImageSerializer
    lookup_field = 'id'
    permission_classes = [
        Or(IsAdmin),
        IsAuthenticated
    ]

    def get_queryset(self):
        queryset = Image.objects

        is_annotated_filter = self.request.query_params.get('is_annotated', None)
        if is_annotated_filter is not None:
            queryset = queryset.filter.annotate(
                cnt=Count('annotation')).filter(cnt__gte=1)

        queryset_order = get_queryset_order(self.request)
        return queryset.order_by(queryset_order).all()

• http://www.django-rest-framework.org/api-guide/filtering/#filtering-against-query-parameters
• http://www.django-rest-framework.org/api-guide/fields/#serializermethodfield

Answer 1

我想你误解了：1）序列化器 MethodField 其主要目的是只读的，并且序列化器不打算用于过滤查询集。

我会像这样过滤它：

from django.db.models import Count

queryset.filter.annotate(
    cnt=Count('annotation_set')
).filter(cnt__gte=1)

但是...你可以做得更好：

1) 只需注释您的查询集，例如在您的 ViewSet 中

from django.db.models import Count
queryset = Image.objects.annotate(
    cnt_annotations=Count('annotation_set')
)

2) 然后在序列化程序中执行以下操作：

@staticmethod
def has_annotation(image):
    if hasattr(obj, 'has_annotation'):
        return bool(obj.cnt_annotations)
    else:
        return None