diff --git a/server/proj/audio/views.py b/server/proj/audio/views.py index 5f34ad2..5e470eb 100644 --- a/server/proj/audio/views.py +++ b/server/proj/audio/views.py @@ -34,8 +34,7 @@ class SongViewSet(ViewSet): @action(detail=False, methods=['post'], schema=DeleteSongSchema()) def delete_song(self, request): song_id = request.data['song_id'] - user_id = request.user.pk - item = FavoriteSong.objects.filter(song=song_id, user=user_id) + item = get_object_or_404(FavoriteSong, pk=song_id) item.delete() return Response(status=status.HTTP_202_ACCEPTED)