From a573b80f0933a1a7d1f881ce7fa9893f049dbb81 Mon Sep 17 00:00:00 2001
From: Michal Kunc <michkunc@gmail.com>
Date: Tue, 24 Jan 2023 20:45:58 +0100
Subject: [PATCH] Add mark to unvoted movies

---
 templates/watchlist/index.html | 2 +-
 watchlist/views.py             | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/templates/watchlist/index.html b/templates/watchlist/index.html
index 2b360d5..097ca33 100644
--- a/templates/watchlist/index.html
+++ b/templates/watchlist/index.html
@@ -4,7 +4,7 @@
     <h1>Watchlist</h1> 
     <ul>
         {% for movie in object_list %}
-        <li><a href="{% url 'watchlist:detail' movie.id %}">{{movie.name}}</a> &mdash; {{movie.score}}</li>
+        <li><a href="{% url 'watchlist:detail' movie.id %}">{{movie.name}}</a>{% if movie in voted_movies %}<sup>*</sup>{% endif %} &mdash; {{movie.score}}</li>
         {% endfor %}
     </ul>
     {% if can_add_movie %}
diff --git a/watchlist/views.py b/watchlist/views.py
index 1defc5e..2a52170 100644
--- a/watchlist/views.py
+++ b/watchlist/views.py
@@ -44,6 +44,7 @@ class IndexView(generic.ListView):
     def get_context_data(self):
         context = super().get_context_data()
         context['can_add_movie'] = self.request.user.has_perm("watchlist.add_movie")
+        context['voted_movies'] = models.Movie.objects.filter(movievote__user=self.request.user)
         return context
 
 class DetailView(generic.DetailView):