CODEWITHSUNDEEP
pythondjangodjango-rest-framework
Flowerowl
Flowerowl

Reputation: 5

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

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...

Upvotes: 0

Views: 233

Answers (1)

Priyank Patel
Priyank Patel

Reputation: 3535

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')

Upvotes: 1

Related Questions