Updates

Nu met 200% meer AI.

playlist/views.py

@@ -7,6 +7,7 @@ import re
 from .models import Track, Artist, Album, Vote
 from .utils import from_json, get_unvoted
 from . import spotify
+from .tasks import get_banter
 
 spt = spotify.Spotify()
 
@@ -96,4 +97,19 @@ class NominateView(LoginRequiredMixin, View):
                 track.artists.add(artist)
             track.album = album
             track.save()
+            get_banter.delay(track.id)
+            
             return TemplateResponse(request, "Nominate.html", {"nominated": track})
+
+class BanterView(LoginRequiredMixin, View):
+    def get(self, request, spotify_id):
+        try:
+            track = Track.objects.get(pk=spotify_id)
+            return TemplateResponse(request, "Banter.html", {
+                "track": track,
+                "waiting": ["Nog even..", "Bijna klaar.", "Oké...", "Interessant", "Momentje"]
+            })
+        except Track.DoesNotExist:
+            return HttpResponse("???")
+
+