adds albums

playlist/views/__init__.py

@@ -3,3 +3,4 @@ from .api.root import *
 from .api.track import *
 from .api.artist import *
 from .api.user import *
+from .api.album import *

playlist/views/api/album.py

@@ -0,0 +1,24 @@
+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 Album
+from playlist.serializers import AlbumSerializer
+
+class AlbumListView(APIView):
+    authentication_classes = [SessionAuthentication]
+    permission_classes = [IsAuthenticated]
+
+    def get(self, request):
+        albums = Album.objects.all()
+        serializer = AlbumSerializer(albums, many=True, context={'request': request})
+        return Response(serializer.data)
+
+class AlbumDetailView(APIView):
+    authentication_classes = [SessionAuthentication]
+    permission_classes = [IsAuthenticated]
+
+    def get(self, request, id):
+        album = Album.objects.get(pk=id)
+        serializer = AlbumSerializer(album, context={'request': request})
+        return Response(serializer.data)

playlist/views/api/root.py

@@ -7,5 +7,7 @@ from rest_framework.reverse import reverse
 def api_root(request, format=None):
     return Response({
         'tracks': reverse('api-tracks', request=request, format=format),
-        'artists': reverse('api-artists', request=request, format=format)
+        'artists': reverse('api-artists', request=request, format=format),
+        'users': reverse('api-users', request=request, format=format),
+        'albums': reverse('api-albums', request=request, format=format),
     })