diff --git a/server/proj/audio/serializers.py b/server/proj/audio/serializers.py index 1bc4e42..751389f 100644 --- a/server/proj/audio/serializers.py +++ b/server/proj/audio/serializers.py @@ -4,7 +4,7 @@ from .models import Song, FavoriteSong class SongSerializer(serializers.ModelSerializer): class Meta: model = Song - fields = ('id', 'unique_id', 'title', 'artist', 'album', 'genre', 'art') + fields = ('id', 'unique_id', 'azura_id', 'title', 'artist', 'album', 'genre', 'art') class FavoriteSongSerializer(serializers.ModelSerializer): diff --git a/server/proj/audio/views.py b/server/proj/audio/views.py index acf8f1b..2f86995 100644 --- a/server/proj/audio/views.py +++ b/server/proj/audio/views.py @@ -41,16 +41,14 @@ class SongViewSet(GenericViewSet): @action(detail=False, methods=['post'], schema=SongSchema()) def add_favorite(self, request): - song = get_object_or_404(Song, azura_id=request.data.get('azura_id')) - data = None - #Если трек есть в базе - if song: + data = None + try: + song = get_object_or_404(Song, azura_id=request.data.get('azura_id')) data = { "song": song, "user": request.user } - #Если трека нету в базе - else: + except ObjectDoesNotExist: file_url = f"http://82.97.242.49:10084/api/station/it-radio/file/{request.data['azura_id']}" API_KEY = "49226d3488aac3f5:18d88659c6c1c5e131a0ce0a94d55235" headers = {