使用 DRF 反向过滤 Django 模型

Question

我在django-models 中的过滤有问题，我正在使用django-rest-framework 处理这个序列化数据。

我想要在这里获取所有牛群记录，其中动物可能有species_type='Cow' 或空牛群。

这是我的模型。

models.py

class Herd(models.Model):
    name = models.CharField(max_length=25)
    description = models.TextField(max_length=250, null=True)

    created_at = models.DateTimeField(auto_now_add=True, editable=False)
    updated_at = models.DateTimeField(auto_now=True, editable=False)


class Animal(models.Model):
    name = models.CharField(max_length=25)
    species_type = models.CharField(max_length=25)
    breed = models.CharField(max_length=25)

    herd = models.ForeignKey(Herd, related_name='animals', on_delete=models.CASCADE)

    created_at = models.DateTimeField(auto_now_add=True, editable=False)
    updated_at = models.DateTimeField(auto_now=True, editable=False)

serializers.py

class AnimalSerializer(serializers.ModelSerializer):

    class Meta:
        model = Animal
        fields = [
            'name', 
            'species_type', 
            'breed'
          ]
        read_only_fields = ['id', 'created_at', 'updated_at']

class HerdSerializer(serializers.ModelSerializer):
    animals = AnimalSerializer(many=True, read_only=True)

    class Meta:
        model = Herd
        fields = [
            'id',
            'name',
            'description',
            'animals'
        ]
        read_only_fields = ['created_at', 'updated_at']

这是我处理所有 crud 操作的视图集。

views.py

class HerdViewset(viewsets.ModelViewSet):
    """
    This viewset automatically provides `list`, `create`, `retrieve`,
    `update` and `destroy` actions.
    """
    queryset = Herd.objects.all()
    serializer_class = HerdSerializer

现在，当我浏览HerdViewSet 端点/api/herd/ 时，我得到了所有有动物的牛群或空牛群的结果。但是某些畜群中的动物没有过滤species_type='Cow'，它仍然返回属于该畜群的所有动物，无论species_type是山羊、绵羊等。

Answer 1

您可以通过 Herd 中的species_type 进行过滤。因为你在外键中定义了related_name。

试试这个

from django.db.models import Count, Q

Herd.objects.annotate(animalCount=Count('animals')).filter(Q(animals__species_type='Cow')|Q(animalCount=0))

1. annotate 用于在结果中添加一个额外字段，因此一个新字段 animalCount 用于保存该畜群的动物数量。
2. Q 用于构建复杂的查询条件，所以对于这种情况，Q(animals__species_type='Cow')|Q(animalCount=0) 的意思是，按物种类型为“牛”的畜群中的动物进行过滤，或者该畜群中没有动物。