добавил метод для проверки в избранном трек или нет

This commit is contained in:
Mike0001-droid 2024-06-10 17:25:34 +05:00
parent 177dcd0688
commit 0aba2ba891
1 changed files with 12 additions and 0 deletions

View File

@ -32,6 +32,18 @@ class SongViewSet(GenericViewSet):
serializer = FavoriteSongSerializer(favorite_songs)
return Response(serializer.data)
@action(detail=False, methods=['GET'], schema=DeleteSongSchema())
def check_is_favorite(self, request):
try:
song_obj = Song.objects.get(azura_id=request.data.get('azura_id'))
favorite_songs = FavoriteSong.objects.get(user=request.user, song=song_obj)
except ObjectDoesNotExist:
return Response({"error": 'Объекта не существует'}, status=status.HTTP_404_NOT_FOUND)
serializer = FavoriteSongSerializer(favorite_songs)
return Response(serializer.data)
@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'))