54 lines
2.3 KiB
Python
54 lines
2.3 KiB
Python
from rest_framework.views import APIView
|
|
from rest_framework.response import Response
|
|
from rest_framework.authentication import SessionAuthentication
|
|
from rest_framework.permissions import IsAuthenticated
|
|
from playlist.models import Track, Profile
|
|
from playlist.serializers import TrackSerializer
|
|
from playlist.spotify import spt
|
|
from playlist.tasks import get_banter, get_and_dither_image
|
|
|
|
class TrackListView(APIView):
|
|
authentication_classes = [SessionAuthentication]
|
|
permission_classes = [IsAuthenticated]
|
|
def get(self, request):
|
|
tracks = Track.objects.all()
|
|
serializer = TrackSerializer(tracks, many=True, context={'request': request})
|
|
return Response(serializer.data)
|
|
|
|
def post(self, request):
|
|
(profile, _) = Profile.objects.get_or_create(user=request.user)
|
|
(can_nom, reason) = profile.can_nominate
|
|
if not can_nom:
|
|
return Response({'error': reason}, status=403)
|
|
(track, message) = Track.objects.create_from_spotify(request.data['spotify_link'], profile)
|
|
serializer = TrackSerializer(track, context={'request': request})
|
|
return Response(serializer.data)
|
|
|
|
def delete(self, request):
|
|
if request.user.is_superuser:
|
|
Track.objects.all().delete()
|
|
return Response(status=204)
|
|
else:
|
|
return Response({'error': 'Only superusers can delete all tracks'}, status=403)
|
|
|
|
class TrackDetailView(APIView):
|
|
authentication_classes = [SessionAuthentication]
|
|
permission_classes = [IsAuthenticated]
|
|
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)
|