Major update

bda/algorithm.py

@@ -0,0 +1,92 @@
+# coding: utf-8
+
+from django.conf import settings
+import random
+
+class Algorithm(object):
+
+    shows = None
+    ranks = None
+    origranks = None
+    double = None
+
+    def __init__(self, shows, members):
+        """Initialisation :
+            - on aggrège toutes les demandes pour chaque spectacle dans
+              show.requests
+            - on crée des tables de demandes pour chaque personne, afin de
+              pouvoir modifier les rankings"""
+        self.shows = []
+        showdict = {}
+        for show in shows:
+            showdict[show] = show
+            show.requests = []
+            self.shows.append(show)
+        self.ranks = {}
+        self.origranks = {}
+        self.double = {}
+        for member in members:
+            ranks = {}
+            double = {}
+            for i in range(1, settings.NUM_CHOICES + 1):
+                choice = getattr(member, "choice%d" % i)
+                if not choice:
+                    continue
+                # Noter les doubles demandes
+                if choice in double:
+                    double[choice] = True
+                else:
+                    showdict[choice].requests.append(member)
+                    ranks[choice] = i
+                    double[choice] = False
+            self.ranks[member] = ranks
+            self.double[member] = double
+            self.origranks[member] = dict(ranks)
+
+    def IncrementRanks(self, member, currank, increment = 1):
+        for show in self.ranks[member]:
+            if self.ranks[member][show] > currank:
+                self.ranks[member][show] -= increment
+
+    def appendResult(self, l, member, show):
+        l.append((member,
+                  self.ranks[member][show],
+                  self.origranks[member][show],
+                  self.double[member][show]))
+
+    def __call__(self, seed):
+        random.seed(seed)
+        results = []
+        for show in self.shows:
+            # On regroupe tous les gens ayant le même rang
+            groups = {}
+            for i in range(1, settings.NUM_CHOICES + 1):
+                groups[i] = []
+            for member in show.requests:
+                groups[self.ranks[member][show]].append(member)
+            # On passe à l'attribution
+            winners = []
+            losers = []
+            for i in range(1, settings.NUM_CHOICES + 1):
+                group = list(groups[i])
+                random.shuffle(group)
+                for member in group:
+                    if self.double[member][show]: # double
+                        if len(winners) + 1 < show.slots:
+                            self.appendResult(winners, member, show)
+                            self.appendResult(winners, member, show)
+                        elif not member.autoquit and len(winners) < show.slots:
+                            self.appendResult(winners, member, show)
+                            self.appendResult(losers, member, show)
+                        else:
+                            self.appendResult(losers, member, show)
+                            self.appendResult(losers, member, show)
+                            self.IncrementRanks(member, i, 2)
+                    else: # simple
+                        if len(winners) < show.slots:
+                            self.appendResult(winners, member, show)
+                        else:
+                            self.appendResult(losers, member, show)
+                            self.IncrementRanks(member, i)
+            results.append((show,winners,losers))
+        return results