From 63311d1d2608518f7b9f20bc39973d6fbb36d7ed Mon Sep 17 00:00:00 2001 From: Alexander-D-Karpov Date: Fri, 8 Dec 2023 16:35:32 +0300 Subject: [PATCH] updated serializer --- akarpov/music/api/serializers.py | 14 +++++++++----- 1 file changed, 9 insertions(+), 5 deletions(-) diff --git a/akarpov/music/api/serializers.py b/akarpov/music/api/serializers.py index e24fe3f..bb160ea 100644 --- a/akarpov/music/api/serializers.py +++ b/akarpov/music/api/serializers.py @@ -32,10 +32,12 @@ class SongSerializer(serializers.ModelSerializer): @extend_schema_field(serializers.BooleanField) def get_liked(self, obj): - if self.context["request"].user.is_authenticated: - return SongUserRating.objects.filter( - song=obj, user=self.context["request"].user, like=True - ).exists() + if "request" in self.context: + if self.context["request"].user.is_authenticated: + return SongUserRating.objects.filter( + song=obj, user=self.context["request"].user, like=True + ).exists() + return None class Meta: model = Song @@ -66,7 +68,9 @@ class ListSongSerializer(SetUserModelSerializer): def get_liked(self, obj): if "likes" in self.context: return self.context["likes"] - return obj.id in self.context["likes_ids"] + if "likes_ids" in self.context: + return obj.id in self.context["likes_ids"] + return None class Meta: model = Song