initial API views

playlist/serializers/__init__.py

@@ -0,0 +1,2 @@
+from .track import TrackSerializer
+from .artist import ArtistSerializer

playlist/serializers/artist.py

@@ -0,0 +1,10 @@
+from playlist.models import Artist
+from rest_framework import serializers
+
+class ArtistSerializer(serializers.HyperlinkedModelSerializer):
+    class Meta:
+        model = Artist
+        fields = ['id', 'name', 'url']
+        extra_kwargs = {
+            'url': {'view_name': 'api-artist', 'lookup_field': 'id'},
+        }

playlist/serializers/track.py

@@ -0,0 +1,11 @@
+from playlist.models import Track
+from rest_framework import serializers
+
+class TrackSerializer(serializers.HyperlinkedModelSerializer):
+    class Meta:
+        model = Track
+        fields = ['url', 'id', 'name', 'artists', 'nominated_by', 'explicit', 'duration_ms', 'popularity', 'banter', 'banter_done', 'old']
+        extra_kwargs = {
+            'url': {'view_name': 'api-track', 'lookup_field': 'id'},
+            'artists': {'view_name': 'api-artist', 'lookup_field': 'id'}
+        }

playlist/urls.py

@@ -4,6 +4,11 @@ from django.conf import settings
 from django.conf.urls.static import static
 
 urlpatterns = [
+    path('api/', views.api_root, name="api-root"),
+    path('api/track', views.TrackListView.as_view(), name="api-tracks"),
+    path('api/track/<id>', views.TrackDetailView.as_view(), name="api-track"),
+    path('api/artist/<id>', views.ArtistDetailView.as_view(), name="api-artist"),
+    path('api/artist', views.ArtistListView.as_view(), name="api-artists"),
     path('track/', views.NominateView.as_view(), name="nominate"),
     path('vote/', views.VoteView.as_view(), name="vote"),
     path('vote/undo', views.UndoView.as_view(), name="undo"),

playlist/views/__init__.py

@@ -0,0 +1,4 @@
+from .static import *
+from .api.track import *
+from .api.artist import *
+from .api.root import *

playlist/views/api/artist.py

@@ -0,0 +1,22 @@
+from itertools import permutations
+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 Artist
+from playlist.serializers import ArtistSerializer
+
+class ArtistListView(APIView):
+    authentication_classes = [SessionAuthentication]
+    permission_classes = [IsAuthenticated]
+
+    def get(self, request):
+        artists = Artist.objects.all()
+        serializer = ArtistSerializer(artists, many=True, context={'request': request})
+        return Response(serializer.data)
+
+class ArtistDetailView(APIView):
+    def get(self, request, id):
+        artist = Artist.objects.get(pk=id)
+        serializer = ArtistSerializer(artist, context={'request': request})
+        return Response(serializer.data)

playlist/views/api/root.py

@@ -0,0 +1,11 @@
+from rest_framework.decorators import api_view
+from rest_framework.response import Response
+from rest_framework.reverse import reverse
+
+
+@api_view(['GET'])
+def api_root(request, format=None):
+    return Response({
+        'tracks': reverse('api-tracks', request=request, format=format),
+        'artists': reverse('api-artists', request=request, format=format)
+    })

playlist/views/api/track.py

@@ -0,0 +1,34 @@
+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)

playlist/views/static.py

@@ -10,10 +10,10 @@ import os
 from datetime import UTC, datetime
 from collections import OrderedDict
 from dotenv import load_dotenv
-from .models import Track, Artist, Album, Vote, Background, Profile, Playlist
+from ..models import Track, Artist, Album, Vote, Background, Profile, Playlist
-from .utils import track_from_json, get_unvoted, get_unvoted_count
+from ..utils import track_from_json, get_unvoted, get_unvoted_count
-from .spotify import spt
+from ..spotify import spt
-from .tasks import get_banter, get_and_dither_image
+from ..tasks import get_banter, get_and_dither_image
 
 load_dotenv()