#!/usr/bin/env python # coding: utf-8 import os import sys import time if __name__ == "__main__": os.environ.setdefault("DJANGO_SETTINGS_MODULE", "cof.settings") from django.conf import settings settings.DEBUG = True from bda.models import Spectacle, Participant, ChoixSpectacle from bda.algorithm import Algorithm from django.db.models import Sum from django.db import connection start = time.time() shows = Spectacle.objects.all() members = Participant.objects.all() choices = ChoixSpectacle.objects.order_by('participant', 'priority').select_related().all() available_slots = Spectacle.objects.aggregate(Sum('slots'))['slots__sum'] cursor = connection.cursor() cursor.execute("SELECT SUM(`slots` * `price`) AS `total` FROM `bda_spectacle`;") total_price = cursor.fetchone()[0] print "%d spectacles" % len(shows) print "%d places" % available_slots print "%d participants" % len(members) print "%d demandes" % len(choices) print "%d places demandées" % (len(choices) + len(choices.filter(double=True).all())) print "%.02f€ à brasser" % total_price print "Récupération: %.2fs" % (time.time() - start) start_init = time.time() algo = Algorithm(shows, members, choices) print "Initialisation: %.2fs" % (time.time() - start_init) start_algo = time.time() results = algo(sys.argv[1]) print "Traitement: %.2fs" % (time.time() - start_algo) print len(connection.queries), "requêtes SQL effectuées" queries = list(connection.queries) total_slots = 0 total_losers = 0 for (_, members, losers) in results: total_slots += len(members) total_losers += len(losers) print "Placés %d\nDécus %d" % (total_slots, total_losers) print "Total %d" % (total_slots + total_losers) members2 = {} members_uniq = {} for (show, members, _) in results: for (member, _, _, _) in members: if member.id not in members_uniq: members_uniq[member.id] = member members2[member] = [] member.total = 0 member = members_uniq[member.id] members2[member].append(show) member.total += show.price if len(members) < show.slots: print "%d place(s) invendue(s) pour %s" % (show.slots - len(members), show) members2 = members2.items() print "Temps total: %.2fs" % (time.time() - start) print "Requêtes SQL:" for query in queries: print query['sql']