35 lines
1.3 KiB
Python
35 lines
1.3 KiB
Python
from rest_framework import viewsets
|
|
from rest_framework.views import APIView
|
|
from rest_framework.response import Response
|
|
from playlist.models import Track
|
|
from playlist.serializers import TrackSerializer
|
|
|
|
class TrackViewSet(viewsets.ModelViewSet):
|
|
queryset = Track.objects.all()
|
|
serializer_class = TrackSerializer
|
|
|
|
class TrackListView(APIView):
|
|
def get(self, request):
|
|
tracks = Track.objects.all()
|
|
serializer = TrackSerializer(tracks, many=True, context={'request': request})
|
|
return Response(serializer.data)
|
|
|
|
class TrackDetailView(APIView):
|
|
def get(self, request, id):
|
|
track = Track.objects.get(pk=id)
|
|
serializer = TrackSerializer(track, context={'request': request})
|
|
return Response(serializer.data)
|
|
def put(self, request, id):
|
|
track = Track.objects.get(pk=id)
|
|
serializer = TrackSerializer(track, data=request.data, context={'request': request})
|
|
if serializer.is_valid():
|
|
serializer.save()
|
|
return Response(serializer.data)
|
|
return Response(serializer.errors, status=400)
|
|
def post(self, request):
|
|
serializer = TrackSerializer(data=request.data, context={'request': request})
|
|
if serializer.is_valid():
|
|
serializer.save()
|
|
return Response(serializer.data)
|
|
return Response(serializer.errors, status=400)
|