добавил try except для извлечения объектов
This commit is contained in:
parent
6e54faf605
commit
6bd7251fde
|
|
@ -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
|
||||
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)
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue