adds vote api

playlist/views/__init__.py

@@ -5,3 +5,4 @@ from .api.artist import *
 from .api.user import *
 from .api.album import *
 from .api.profile import *
+from .api.vote import *

playlist/views/api/vote.py

@@ -0,0 +1,27 @@
+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 Vote
+from playlist.serializers import VoteSerializer
+
+class VoteListView(APIView):
+    authentication_classes = [SessionAuthentication]
+    permission_classes = [IsAuthenticated]
+
+    def get(self, request, user_id = None):
+        if user_id:
+            votes = Vote.objects.filter(user_id=user_id)
+        else:
+            votes = Vote.objects.all()
+        serializer = VoteSerializer(votes, many=True, context={'request': request})
+        return Response(serializer.data)
+
+class VoteDetailView(APIView):
+    authentication_classes = [SessionAuthentication]
+    permission_classes = [IsAuthenticated]
+
+    def get(self, request, id):
+        vote = Vote.objects.get(pk=id)
+        serializer = VoteSerializer(vote, context={'request': request})
+        return Response(serializer.data)