diff --git a/server/proj/audio/serializers.py b/server/proj/audio/serializers.py index ab87909..8222b8e 100644 --- a/server/proj/audio/serializers.py +++ b/server/proj/audio/serializers.py @@ -14,17 +14,6 @@ class FavoriteSongSerializer(serializers.ModelSerializer): def to_representation(self, instance): rep = super().to_representation(instance) rep["song"] = SongSerializer( - instance.song).data + instance.song.all(), many=True).data return rep -class OnlyFavoriteSongSerializer(serializers.ModelSerializer): - class Meta: - model = FavoriteSong - fields = ('song', ) - - def to_representation(self, instance): - rep = super().to_representation(instance) - rep["song"] = SongSerializer( - instance.song).data - return rep - \ No newline at end of file diff --git a/server/proj/audio/views.py b/server/proj/audio/views.py index aab79b1..0b072eb 100644 --- a/server/proj/audio/views.py +++ b/server/proj/audio/views.py @@ -20,7 +20,7 @@ class SongViewSet(ViewSet): def list(self, request): user_pk = request.user.pk queryset = get_list_or_404(FavoriteSong, user=user_pk) - serializer = OnlyFavoriteSongSerializer(queryset, many=True) + serializer = FavoriteSongSerializer(queryset, many=True) return Response(serializer.data) def retrieve(self, request, pk=None):