Django - фильтрация связанных объектов

У меня есть загадка.

Вот мои модели: 

class StatusGroup(models.Model):

    name = models.TextField()

    def __str__(self):
        return self.name


class StatusDetail(models.Model):
    action = models.CharField(choices=[("CORRECT", "CORRECT"),
                                       ("INCORRECT", "INCORRECT")],
                              max_length=64)
    status_group = models.ForeignKey(to=StatusGroup,
                                      on_delete=models.CASCADE,
                                      related_name="status_details")

    def __str__(self):
        return f"Detail: {self.action}"

сериализаторы:

class StatusDetailSerializer(serializers.ModelSerializer):

    class Meta:
        model= models.StatusDetail
        fields = "__all__"

class StatusGroupSerializer(serializers.ModelSerializer):
    status_details = StatusDetailSerializer(many=True)

    class Meta:
        model = models.StatusGroup
        fields = [
            "pk",
            "status_details",
            "name"
        ]

И вид: 

class Status(viewsets.ModelViewSet):

    queryset = models.StatusGroup.objects.all()
    serializer_class = serializers.StatusGroupSerializer
    authentication_classes = []
    permission_classes = [permissions.AllowAny]

    filter_backends = (DjangoFilterBackend,)
    filterset_fields = ['status_details__action']

Когда я бью localhost:8000/api/status?status_details__action=INCORRECT

Я получаю:

[
    {
        "pk": 2,
        "status_details": [
            {
                "id": 3,
                "action": "CORRECT",
                "status_group": 2
            },
            {
                "id": 4,
                "action": "INCORRECT",
                "status_group": 2
            }
        ],
        "name": "Mixed"
    }
]

В то время как я хотел бы иметь:

[
    {
        "pk": 2,
        "status_details": [
            {
                "id": 4,
                "action": "INCORRECT",
                "status_group": 2
            }
        ],
        "name": "Mixed"
    }
]

Как заставить Django фильтровать связанные объекты? Я могу получить нужный мне результат в SQL консоли, но Django добавляет, все связанные объекты, которые принадлежат StatusGroup. У меня есть заблуждение, но я не знаю, что это такое

попробуйте этот способ (используя набор запросов):

from django.db.models import Prefetch

queryset = StatusGroup.objects.prefetch_related(
Prefetch('status_details', queryset=StatusDetail.objects.filter(action='what_you_want'), to_attr='action'))

и в вашем классе сериализатора: 

class StatusGroupSerializer(serializers.ModelSerializer):
status_details = StatusDetailSerializer(source='action', many=True, read_only=True)

Сначала вы можете создать класс FilterSet следующим образом:

from django_filters import rest_framework as filters

class StatusGroupFilter(filters.FilterSet):
    action = filters.CharFilter(
        field_name="status_details__action",
        lookup_expr="iexact"
    )

    class Meta:
        model = StatusGroup
        fields = ["action"]

Тогда в ваших представлениях:

class Status(viewsets.ModelViewSet):

    queryset = models.StatusGroup.objects.all()
    serializer_class = serializers.StatusGroupSerializer
    authentication_classes = []
    permission_classes = [permissions.AllowAny]

    filter_backends = (DjangoFilterBackend,)
    # Set the filterset class here
    filterset_class = StatusGroupFilter

Тогда в URL вы можете вызвать localhost:8000/api/status?action=INCORRECT

Вернуться на верх