# 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