написал ретрив
This commit is contained in:
parent
2069365376
commit
7fcda8ed59
|
|
@ -16,6 +16,18 @@ class SongViewSet(ViewSet):
|
||||||
serializer = SongSerializer(queryset, many=True)
|
serializer = SongSerializer(queryset, many=True)
|
||||||
return Response(serializer.data)
|
return Response(serializer.data)
|
||||||
|
|
||||||
|
def retrieve(self, request, pk=None):
|
||||||
|
user_pk = request.user.pk
|
||||||
|
song_obj = get_object_or_404(Song, azura_id=pk).pk
|
||||||
|
queryset = FavoriteSong.objects.filter(user=user_pk, song=song_obj).exists()
|
||||||
|
if queryset:
|
||||||
|
serializer = FavoriteSongSerializer(queryset=queryset)
|
||||||
|
resp = serializer.data
|
||||||
|
resp.update(is_favorite=True)
|
||||||
|
return Response(serializer.data, status=status.HTTP_201_CREATED)
|
||||||
|
else:
|
||||||
|
return Response('Трека нет... Стёпа извини...', status=status.HTTP_400_BAD_REQUEST)
|
||||||
|
|
||||||
@action(detail=False, methods=['post'], schema=SongSchema())
|
@action(detail=False, methods=['post'], schema=SongSchema())
|
||||||
def create_song(self, request):
|
def create_song(self, request):
|
||||||
song = Song.objects.filter(azura_id = request.data['azura_id'])
|
song = Song.objects.filter(azura_id = request.data['azura_id'])
|
||||||
|
|
@ -23,7 +35,6 @@ class SongViewSet(ViewSet):
|
||||||
data = {
|
data = {
|
||||||
"song": song[0].id,
|
"song": song[0].id,
|
||||||
"user": request.user.pk
|
"user": request.user.pk
|
||||||
|
|
||||||
}
|
}
|
||||||
serializer_1 = FavoriteSongSerializer(data=data)
|
serializer_1 = FavoriteSongSerializer(data=data)
|
||||||
if serializer_1.is_valid():
|
if serializer_1.is_valid():
|
||||||
|
|
@ -51,7 +62,6 @@ class SongViewSet(ViewSet):
|
||||||
"song": song[0].id,
|
"song": song[0].id,
|
||||||
"user": request.user.pk
|
"user": request.user.pk
|
||||||
}
|
}
|
||||||
|
|
||||||
serializer_1 = FavoriteSongSerializer(data=data)
|
serializer_1 = FavoriteSongSerializer(data=data)
|
||||||
if serializer_1.is_valid():
|
if serializer_1.is_valid():
|
||||||
serializer_1.save()
|
serializer_1.save()
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue