рефакторинг
This commit is contained in:
parent
c4c1d5a9cb
commit
090d14b8e6
|
|
@ -15,25 +15,20 @@ from .models import Song, FavoriteSong
|
||||||
from .serializers import SongSerializer, FavoriteSongSerializer
|
from .serializers import SongSerializer, FavoriteSongSerializer
|
||||||
|
|
||||||
class SongViewSet(GenericViewSet):
|
class SongViewSet(GenericViewSet):
|
||||||
permission_classes_by_action = {
|
serializer_class = SongSerializer
|
||||||
'list': [IsAuthenticated],
|
permission_classes = (IsAuthenticated,)
|
||||||
'retrieve': [IsAuthenticated],
|
queryset = Song.objects.all()
|
||||||
'create_song': [IsAuthenticated]
|
|
||||||
}
|
|
||||||
|
|
||||||
def list(self, request):
|
def list(self, request):
|
||||||
try:
|
|
||||||
songs_pk = FavoriteSong.objects.filter(user=request.user, song__isnull=False).values_list('song_id', flat=True)
|
songs_pk = FavoriteSong.objects.filter(user=request.user, song__isnull=False).values_list('song_id', flat=True)
|
||||||
queryset = Song.objects.filter(id__in=songs_pk)
|
queryset = Song.objects.filter(id__in=songs_pk)
|
||||||
except ObjectDoesNotExist:
|
|
||||||
return Response({"error": 'Объекта не существует'}, status=status.HTTP_404_NOT_FOUND)
|
|
||||||
serializer = SongSerializer(queryset, many=True)
|
serializer = SongSerializer(queryset, many=True)
|
||||||
return Response(serializer.data)
|
return Response(f"{request.build_absolute_uri()}")
|
||||||
|
|
||||||
def retrieve(self, request, pk=None):
|
def retrieve(self, request, pk=None):
|
||||||
try:
|
try:
|
||||||
song_obj = Song.objects.get(azura_id=pk).pk
|
song_obj = Song.objects.get(azura_id=pk)
|
||||||
queryset = FavoriteSong.objects.get(user=request.user.pk, song=song_obj)
|
queryset = FavoriteSong.objects.get(user=request.user, song=song_obj)
|
||||||
except ObjectDoesNotExist:
|
except ObjectDoesNotExist:
|
||||||
return Response({"error": 'Объекта не существует'}, status=status.HTTP_404_NOT_FOUND)
|
return Response({"error": 'Объекта не существует'}, status=status.HTTP_404_NOT_FOUND)
|
||||||
serializer = FavoriteSongSerializer(queryset)
|
serializer = FavoriteSongSerializer(queryset)
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue