Django rest 框架——过滤多对多字段

Question

Suppose I have a model like this one:

class Car(models.Model):
    images = models.ManyToManyField(Image)

class Image(models.Model):
    path = models.CharField()
    type = models.CharField()

I want to expose two API views:

• cars list
• car details

In list view I want to show only images that have type="thumbnail". In details view I want to show images of type="image".

This is more or less what the list should look like:

[{
    "id": 1,
    "images": [1, 2],
},
{
    "id": 2,
    "images": [3, 4],
}]

And the details view:

{
    "id": 1,
    "images": [5],
}

Note that different image ids are displayed depending on the view.

So far my serializer looks like this:

class CarSerializer(serializers.ModelSerializer):
    images = serializers.ManyPrimaryKeyRelatedField()

    class Meta:
        model = Car

List api view:

class CarList(generics.ListAPIView):
    model = Car
    serializer_class = CarSerializer

Details api view:

class CarDetails(generics.RetrieveAPIView):
    model = Car
    serializer_class = CarSerializer

This of course gives me all images in list as well as in details and forces clients to make additional calls to get image type that should be displayed.

Is there any generic way to do it? I have seen django-filter examples, but it seems that its only possible to filter which objects are listed, not what related objects in listed objects are listed.

Answer 1

I don´t know if you are still looking for this answer, but maybe it helps someone else.

First create a filter class like this:

class CarFilter(django_filters.FilterSet):
    having_image = django_filters.Filter(name="images", lookup_type='in')

    class Meta:
        model = Car

Then add the filter to your view:

class CarList(generics.ListAPIView):
    model = Car
    serializer_class = CarSerializer
    filter_class = CarFilter

And that´s all. Add "?having_image=1" to your query string and Django filter should do the trick for you.

Hope it helps..

Answer 2

I have found a other Stackoverflow question that has a solution that could be used here:

How can I apply a filter to a nested resource in Django REST framework?

(see under the "Solution" headline in the question itself)

Answer 3

I came up with this solution:
it's definitely not the optimized one but can be used for a quick fix or maybe for a hobby project.

So after serializing all the applications, I loop over the data to get the required results, append it inside an array and send it back as response.

Models file:

Views file: