Voting update

Nu met 100% meer mogelijkheid tot stemmen. Top.

playlist/views.py

@@ -2,7 +2,9 @@ from django.shortcuts import render
 from django.views import View
 from django.http import HttpResponse
 from django.template.response import TemplateResponse
+from django.contrib.auth.mixins import LoginRequiredMixin
 import re
+from random import choice
 from .models import Track, Artist, Album
 from .models.utils import from_json
 from . import spotify
@@ -19,14 +21,27 @@ class TrackListView(View):
     def get(self, request):
         return HttpResponse("..")
 
-class TrackDetailView(View):
+class TrackDetailView(LoginRequiredMixin, View):
     def get(self, request, spotify_id):
         try:
             track = Track.objects.get(pk=spotify_id)
             return HttpResponse(str(track))
         except Track.DoesNotExist:
             return HttpResponse("No track")
-class NominateView(View):
+
+class VoteView(LoginRequiredMixin, View):
+    def get(self, request):
+        songs = Track.objects.all()
+        return TemplateResponse(request, "Vote.html", {"songs": songs})
+
+class VoteTrackView(LoginRequiredMixin, View):
+    def get(self, request):
+        pks = Track.objects.values_list('pk', flat=True)
+        track = Track.objects.get(pk=choice(pks))
+
+        return TemplateResponse(request, "SpotifyEmbed.html", {"id": track.id})
+
+class NominateView(LoginRequiredMixin, View):
     def get(self, request):
         return TemplateResponse(request, "Nominate.html", {})
     def post(self, request):