Dépouillement des questions de type condorcet

elections/migrations/0016_auto_20210329_1759.py

@@ -0,0 +1,69 @@
+# Generated by Django 2.2.19 on 2021-03-29 15:59
+
+import django.db.models.deletion
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("elections", "0015_condorcet"),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name="Duel",
+            fields=[
+                (
+                    "id",
+                    models.AutoField(
+                        auto_created=True,
+                        primary_key=True,
+                        serialize=False,
+                        verbose_name="ID",
+                    ),
+                ),
+                (
+                    "amount",
+                    models.PositiveSmallIntegerField(
+                        verbose_name="votes supplémentaires"
+                    ),
+                ),
+                (
+                    "loser",
+                    models.OneToOneField(
+                        on_delete=django.db.models.deletion.CASCADE,
+                        related_name="contests_lost",
+                        to="elections.Option",
+                    ),
+                ),
+                (
+                    "question",
+                    models.ForeignKey(
+                        on_delete=django.db.models.deletion.CASCADE,
+                        related_name="duels",
+                        to="elections.Question",
+                    ),
+                ),
+                (
+                    "winner",
+                    models.OneToOneField(
+                        on_delete=django.db.models.deletion.CASCADE,
+                        related_name="contests_won",
+                        to="elections.Option",
+                    ),
+                ),
+            ],
+        ),
+        migrations.AlterModelOptions(
+            name="rank",
+            options={"ordering": ["vote"]},
+        ),
+        migrations.AlterModelOptions(
+            name="vote",
+            options={"ordering": ["option"]},
+        ),
+        migrations.DeleteModel(
+            name="Matrix",
+        ),
+    ]

elections/migrations/0017_auto_20210329_1802.py

@@ -0,0 +1,32 @@
+# Generated by Django 2.2.19 on 2021-03-29 16:02
+
+import django.db.models.deletion
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("elections", "0016_auto_20210329_1759"),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name="duel",
+            name="loser",
+            field=models.ForeignKey(
+                on_delete=django.db.models.deletion.CASCADE,
+                related_name="contests_lost",
+                to="elections.Option",
+            ),
+        ),
+        migrations.AlterField(
+            model_name="duel",
+            name="winner",
+            field=models.ForeignKey(
+                on_delete=django.db.models.deletion.CASCADE,
+                related_name="contests_won",
+                to="elections.Option",
+            ),
+        ),
+    ]

elections/models.py

@@ -123,17 +123,26 @@ class Vote(models.Model):
     option = models.ForeignKey(Option, on_delete=models.CASCADE)
     user = models.ForeignKey(settings.AUTH_USER_MODEL, on_delete=models.CASCADE)
 
+    class Meta:
+        ordering = ["option"]
+
 
 class Rank(models.Model):
     vote = models.OneToOneField(Vote, on_delete=models.CASCADE)
     rank = models.PositiveSmallIntegerField(_("rang de l'option"))
 
+    class Meta:
+        ordering = ["vote"]
 
-class Matrix(models.Model):
-    winner = models.OneToOneField(
+
+class Duel(models.Model):
+    question = models.ForeignKey(
+        Question, related_name="duels", on_delete=models.CASCADE
+    )
+    winner = models.ForeignKey(
         Option, related_name="contests_won", on_delete=models.CASCADE
     )
-    loser = models.OneToOneField(
+    loser = models.ForeignKey(
         Option, related_name="contests_lost", on_delete=models.CASCADE
     )
     amount = models.PositiveSmallIntegerField(_("votes supplémentaires"))

elections/utils.py

@@ -2,6 +2,8 @@ import csv
 import io
 import random
 
+import numpy as np
+
 from django.contrib.auth.hashers import make_password
 from django.core.exceptions import ValidationError
 from django.core.mail import EmailMessage, get_connection
@@ -87,6 +89,53 @@ class TallyFunctions:
 
         Option.objects.bulk_update(options, ["nb_votes"])
 
+    def tally_rank(question):
+        """On dépouille un vote par classement et on crée la matrice des duels"""
+        from .models import Duel, Rank
+
+        def duel(a, b):
+            # Renvoie 1 si a est classé avant b, -1 si c'est l'inverse et
+            # 0 si a et b ont le même rang, ce qui permet d'avoir directement
+            # le graphe des duels en faisant le max avec la matrice nulle
+            return (a.rank < b.rank) - (a.rank > b.rank)
+
+        options = list(question.options.all())
+        ranks = Rank.objects.select_related("vote").filter(vote__option__in=options)
+        ranks_by_user = {}
+
+        for r in ranks:
+            user = r.vote.user
+            if user in ranks_by_user:
+                ranks_by_user[user].append(r)
+            else:
+                ranks_by_user[user] = [r]
+
+        ballots = []
+
+        # Pour chaque votant·e, on regarde son classement
+        for user in ranks_by_user:
+            votes = ranks_by_user[user]
+            ballots.append(np.array([[duel(x, y) for x in votes] for y in votes]))
+
+        # On additionne les classements de tout le monde
+        duels = np.maximum(sum(ballots), 0)
+
+        cells = []
+
+        # On enregistre la matrice
+        for i, line in enumerate(duels):
+            for j, cell in enumerate(line):
+                if cell > 0:
+                    cells.append(
+                        Duel(
+                            question=question,
+                            winner=options[j],
+                            loser=options[i],
+                            amount=cell,
+                        )
+                    )
+        Duel.objects.bulk_create(cells)
+
 
 class ValidateFunctions:
     """Classe pour valider les formsets selon le type de question"""