diff --git a/server/proj/audio/views.py b/server/proj/audio/views.py index fc0d698..9a3bc71 100644 --- a/server/proj/audio/views.py +++ b/server/proj/audio/views.py @@ -23,11 +23,14 @@ class SongViewSet(ViewSet): data = { "song": song[0].id, "user": request.user.pk + } serializer_1 = FavoriteSongSerializer(data=data) if serializer_1.is_valid(): serializer_1.save() - return Response(serializer_1.data, status=status.HTTP_201_CREATED) + resp = serializer_1.data + resp.update(is_favorite=True) + return Response(resp, status=status.HTTP_201_CREATED) else: return Response(serializer_1.errors, status=status.HTTP_400_BAD_REQUEST) else: @@ -48,7 +51,7 @@ class SongViewSet(ViewSet): "song": song[0].id, "user": request.user.pk } - + serializer_1 = FavoriteSongSerializer(data=data) if serializer_1.is_valid(): serializer_1.save()