diff --git a/server/proj/audio/views.py b/server/proj/audio/views.py index 3d52a3f..0c07eb3 100644 --- a/server/proj/audio/views.py +++ b/server/proj/audio/views.py @@ -18,10 +18,10 @@ class SongViewSet(ViewSet): @action(detail=False, methods=['post'], schema=SongSchema()) def create_song(self, request): - song = Song.objects.get(azura_id = request.data['azura_id']) + song = Song.objects.filter(azura_id = request.data['azura_id']) if song: data = { - "song": song.id, + "song": song[0].id, "user": request.user.pk } serializer_1 = FavoriteSongSerializer(data=data)