votes left count

playlist/utils.py

@@ -25,6 +25,17 @@ def get_unvoted(user):
         return (Track.objects.get(pk=random_track), skipped)
     except IndexError:
         return (None, False)
+    
+def get_unvoted_count(user):
+    votes = set(Vote.objects.filter(user=user).values_list('track_id', flat=True))
+    tracks = set(Track.objects.filter(banter_done=True).values_list('pk', flat=True))
+    try:
+        queue = list(tracks - votes)
+        skipped = list(Vote.objects.filter(user=user).filter(skipped=True).values_list('track_id', flat=True))
+        return (len(queue), len(skipped))
+    except IndexError:
+        return (0, 0)
+    
 
 def track_from_json(json):
     """Creates the track, album and artist from spotify response."""