From 1cdb1253564b7c8d6bd497ee21fc36a201d9c64f Mon Sep 17 00:00:00 2001
From: Tom Hubrecht <tom.hubrecht@ens.fr>
Date: Sat, 17 Apr 2021 00:23:33 +0200
Subject: [PATCH] Add a view to see the list of ballots

---
 elections/models.py                           |  5 +++
 .../templates/elections/ballots/select.html   | 25 +++++++++++++
 .../templates/elections/election_ballots.html | 36 +++++++++++++++++++
 elections/urls.py                             |  5 +++
 elections/utils.py                            | 25 +++++++++++++
 elections/views.py                            | 13 +++++++
 6 files changed, 109 insertions(+)
 create mode 100644 elections/templates/elections/ballots/select.html
 create mode 100644 elections/templates/elections/election_ballots.html

diff --git a/elections/models.py b/elections/models.py
index 5e7c203..570c5d2 100644
--- a/elections/models.py
+++ b/elections/models.py
@@ -14,6 +14,7 @@ from .staticdefs import (
     VALIDATE_FUNCTIONS,
 )
 from .utils import (
+    BallotsData,
     CastFunctions,
     ResultsData,
     TallyFunctions,
@@ -116,6 +117,10 @@ class Question(models.Model):
         results_function = getattr(ResultsData, BALLOT_TYPE[self.type])
         return results_function(self)
 
+    def display_ballots(self):
+        display_function = getattr(BallotsData, BALLOT_TYPE[self.type])
+        return display_function(self)
+
     @property
     def vote_type(self):
         return BALLOT_TYPE[self.type]
diff --git a/elections/templates/elections/ballots/select.html b/elections/templates/elections/ballots/select.html
new file mode 100644
index 0000000..fceeec1
--- /dev/null
+++ b/elections/templates/elections/ballots/select.html
@@ -0,0 +1,25 @@
+{% load i18n %}
+
+<table class="table">
+  <thead>
+    <tr>
+      {% for o in options %}
+      <th>{{ o }}</th>
+      {% endfor %}
+    </tr>
+  </thead>
+
+  <tbody>
+    {% for ballot in ballots.values %}
+    <tr>
+      {% for v in ballot %}
+      <td class="has-text-centered">
+        <span class="icon">
+          <i class="fas fa-{% if v %}check{% else %}times{% endif %}"></i>
+        </span>
+      </td>
+      {% endfor %}
+    </tr>
+    {% endfor %}
+  </tbody>
+</table>
diff --git a/elections/templates/elections/election_ballots.html b/elections/templates/elections/election_ballots.html
new file mode 100644
index 0000000..5d768fe
--- /dev/null
+++ b/elections/templates/elections/election_ballots.html
@@ -0,0 +1,36 @@
+{% extends "base.html" %}
+{% load i18n %}
+
+
+{% block content %}
+<div class="level">
+  {# Titre de l'élection #}
+  <div class="level-left is-flex-shrink-1">
+    <h1 class="title">{{ election.name }}</h1>
+  </div>
+
+  <div class="level-right">
+    <div class="level-item">
+      <a class="button is-primary" href="{% url 'election.view' election.pk %}">
+        <span class="icon">
+          <i class="fas fa-undo-alt"></i>
+        </span>
+        <span>{% trans "Retour" %}</span>
+      </a>
+    </div>
+  </div>
+</div>
+
+<div class="level">
+  <div class="level-left">
+    <h3 class="subtitle">{% trans "Liste des bulletins" %}</h3>
+  </div>
+</div>
+<hr>
+
+{# Tableau des bulletins par question #}
+{% for q in election.questions.all %}
+{{ q.display_ballots }}
+{% endfor %}
+
+{% endblock %}
diff --git a/elections/urls.py b/elections/urls.py
index 662954e..974f6fa 100644
--- a/elections/urls.py
+++ b/elections/urls.py
@@ -76,5 +76,10 @@ urlpatterns = [
         views.ElectionVotersView.as_view(),
         name="election.voters",
     ),
+    path(
+        "view/<int:pk>/ballots",
+        views.ElectionBallotsView.as_view(),
+        name="election.ballots",
+    ),
     path("vote/<int:pk>", views.VoteView.as_view(), name="election.vote"),
 ]
diff --git a/elections/utils.py b/elections/utils.py
index 373dea1..708753b 100644
--- a/elections/utils.py
+++ b/elections/utils.py
@@ -271,6 +271,31 @@ class ResultsData:
         )
 
 
+class BallotsData:
+    """Classe pour afficher les bulletins d'une question"""
+
+    def select(question):
+        from .models import Vote
+
+        votes = Vote.objects.filter(option__question=question)
+
+        options = list(question.options.all())
+
+        ballots = {}
+        for v in votes:
+            ballot = ballots.get(v.user, [False] * len(options))
+            ballot[options.index(v.option)] = True
+
+            ballots[v.user] = ballot
+
+        return render_to_string(
+            "elections/ballots/select.html", {"options": options, "ballots": ballots}
+        )
+
+    def rank(question):
+        return ""
+
+
 # #############################################################################
 #  Fonctions pour importer une liste de votant·e·s
 # #############################################################################
diff --git a/elections/views.py b/elections/views.py
index 34e20b0..46863d9 100644
--- a/elections/views.py
+++ b/elections/views.py
@@ -479,6 +479,19 @@ class ElectionVotersView(NotArchivedMixin, DetailView):
         return context
 
 
+class ElectionBallotsView(NotArchivedMixin, DetailView):
+    model = Election
+    template_name = "elections/election_ballots.html"
+
+    def get_queryset(self):
+        return (
+            super()
+            .get_queryset()
+            .filter(tallied=True)
+            .prefetch_related("questions__options")
+        )
+
+
 class VoteView(OpenElectionOnlyMixin, DetailView):
     model = Question