diff --git a/server/proj/audio/views.py b/server/proj/audio/views.py index a2e818b..b3711a3 100644 --- a/server/proj/audio/views.py +++ b/server/proj/audio/views.py @@ -3,6 +3,7 @@ from rest_framework.viewsets import ViewSet, GenericViewSet from rest_framework.response import Response from rest_framework.decorators import action from rest_framework import status +from django.core.exceptions import ObjectDoesNotExist, MultipleObjectsReturned from django.shortcuts import get_object_or_404, get_list_or_404 from rest_framework.permissions import IsAuthenticated import requests @@ -25,8 +26,14 @@ class SongViewSet(GenericViewSet): return Response(serializer.data) def retrieve(self, request, pk=None): - song_obj = get_object_or_404(Song, azura_id=pk).pk - queryset = FavoriteSong.objects.get(user=request.user.pk, song=song_obj) + try: + song_obj = Song.objects.get(azura_id=pk).pk + queryset = FavoriteSong.objects.get(user=request.user.pk, song=song_obj) + except ObjectDoesNotExist: + print("Объект не сушествует") + except MultipleObjectsReturned: + print("Найдено более одного объекта") + serializer = FavoriteSongSerializer(queryset) return Response(serializer.data)