diff --git a/server/proj/audio/views.py b/server/proj/audio/views.py index aa6b3a5..85734a2 100644 --- a/server/proj/audio/views.py +++ b/server/proj/audio/views.py @@ -19,8 +19,8 @@ class SongViewSet(ViewSet): def retrieve(self, request, pk=None): user_pk = request.user.pk song_obj = get_object_or_404(Song, azura_id=pk).pk - queryset = FavoriteSong.objects.filter(user=user_pk, song=song_obj).exists() - if queryset: + queryset = FavoriteSong.objects.filter(user=user_pk, song=song_obj) + if queryset.exists(): serializer = FavoriteSongSerializer(queryset=queryset) resp = serializer.data resp.update(is_favorite=True)