How can I get results from multiple models with Django REST Framework

Question

I have some models : User/Like/Dislike...

class Like(models.Model):
""" Like """
    user = models.ForeignKey(User)
    movie_id = models.IntegerField()
    created = models.DateTimeField(auto_now_add=True)

class Dislike(models.Model):
""" Dislike """
    user = models.ForeignKey(User)
    movie_id = models.IntegerField()
    created = models.DateTimeField(auto_now_add=True)

How can I get results like this with Django REST Framework by movie_id list and current user?

movie_id = [11111,22222,33333,44444]

{
"results": [
    {
        "movie_id": 125405, 
        "like": True, 
        "dislike": False,
    }, 
    ...
    ]
}

LikeSerializer:

class LikeSerializer(serializers.ModelSerializer):
    user = UserSerializer(required=False)

    class Meta:
        model = Like
        fields = ('id', 'user', 'movie_id', 'created')


class LikeViewSet(viewsets.ModelViewSet):
    model = Like
    serializer_class = LikeSerializer
    permission_classes = (IsAuthenticated,)

DisLikeSerializer:

class DislikeSerializer(serializers.ModelSerializer):
    user = UserSerializer(required=False)

    class Meta:
        model = Dislike
        fields = ('id', 'user', 'movie_id', 'created')


class DislikeViewSet(viewsets.ModelViewSet):
    model = Dislike
    serializer_class = DislikeSerializer
    permission_classes = (IsAuthenticated,)

Thanks...

added answer . try it . let me knw . working or not . not tested. — Priyank Patel
– Priyank Patel, Commented Mar 12, 2014 at 7:36

Priyank Patel · Accepted Answer · 2014-03-12 08:15:17Z

1

You can add custom field to response .

Here we can't use like . as its reserved keyword in python . See this code . implement same for dislike .

class LikeSerializer(serializers.ModelSerializer):
    user = UserSerializer(required=False)
    likes = serializers.SerializerMethodField('no_likes_get')

    def no_likes_get(self, like_obj):
        return Like.objects.filter(movie_id = like_obj.movie_id).count()

    class Meta:
        model = Like
        fields = ('id', 'user', 'movie_id', 'likes')

edited Mar 12, 2014 at 8:15

answered Mar 12, 2014 at 7:35

Priyank Patel

3,56523 silver badges20 bronze badges

Sign up to request clarification or add additional context in comments.

2 Comments

Flowerowl Over a year ago

thks for your answer, i have changed my models, Status:like/dislike/etc.

Flowerowl Over a year ago

i've never used SerializerMethodField, this time i try it , it's awesome!

Collectives™ on Stack Overflow

How can I get results from multiple models with Django REST Framework

1 Answer 1

2 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Your Answer

Sign up or log in

Post as a guest

Related