diff --git a/server/proj/audio/views.py b/server/proj/audio/views.py index 3e83339..fc8dbd4 100644 --- a/server/proj/audio/views.py +++ b/server/proj/audio/views.py @@ -19,7 +19,7 @@ class SongViewSet(ViewSet): } def list(self, request): user_pk = request.user.pk - songs_pk = [i.pk for i in get_list_or_404(FavoriteSong, user=user_pk)] + songs_pk = [i.song.pk for i in get_list_or_404(FavoriteSong, user=user_pk)] queryset = Song.objects.filter(id__in = songs_pk) serializer = SongSerializer(queryset, many=True) return Response(serializer.data)