Pralo/pralo.py
2012-07-22 19:44:44 +02:00

125 lines
3.3 KiB
Python
Executable File

#! /usr/bin/env python
#
# ------------------------------
#
# La grosse fleme de faire les comptes à chaque fois
# Maintenant c'est toi qui les fait pour moi :D
#
# ------------------------------
# ------------------------------
# Imports
# ------------------------------
import csv
import os
import optparse
# ------------------------------
# Classes
# ------------------------------
# ------------------------------
# Fonctions
# ------------------------------
def extrait_from_file(file_name):
"""Extrait les informations à partir d'un fichier"""
file = open(file_name, "r")
try:
reader = csv.reader(file, delimiter = ",")
compte = list(reader)
for i in compte[1:]:
i[1] = int(i[1])
return compte
finally:
file.close()
def echange(l):
"""Algo recursif pour gérer les échanges"""
l.sort(key = lambda s: s[1])
m = l[0]
M = l[-1]
res = ['doneur','receveur',0]
# Si celui qui doit le plus doit moins que le crédit que celui qui a le plus grand crédit
# Il va donc tout lui donner
if abs(m[1]) <= abs(M[1]):
M2 = M[1] + m[1]
# print("{em} donne {em_n} à {eM} donc {em} devient {eM2}".format(em = m[0], em_n = abs(m[1]), eM = M[0] , eM2 = M2))
res = [m[0], M[0], abs(m[1])]
# On enleve le minimum
l.remove(m)
# ON change la valeur du max
l[-1] = [M[0],M2]
# Sinon il donne juste de quoi compenser
else:
m2 = m[1] + M[1]
# print("{em} donne {eM_n} à {eM} et devient {em2}".format(em = m[0] , em2 = m2 , eM = M[0], eM_n = M[1]))
res = [m[0], M[0], abs(M[1])]
# On eleve le max
l.remove(M)
# On change la valeur du min
l[0] = [m[0], m2]
if (len(l) > 1):
#print(l)
return [res] + echange(l)
else:
return [res]
def normalise(compte):
"""Centre en 0 les comptes"""
moyenne = sum([c[1] for c in compte]) / len(compte)
compte_normalise = compte
for (i,n) in enumerate(compte):
compte_normalise[i][1] = compte[i][1] - moyenne
return compte_normalise
def affiche_final(donRec):
"""Affiche qui donne quoi à qui à partir de la liste"""
for g in donRec:
print("{don} donne {montant} à {rec}".format(don = g[0], rec = g[1], montant = g[2]))
def elimine_seuil(liste, seuil):
"""Elimine les éléments de la liste qui sont sous le seuil """
pass
# ------------------------------
# Bloc principal
# ------------------------------
if __name__ == '__main__':
# Pour analyser les options qu'on lui demande
parser = optparse.OptionParser()
# options proposée
parser.add_option("-f","--file",action="store", type = "string", dest="file_name", help="Analyse les comptes à partir du fichier donné en argument")
# Digestion
(options, args) = parser.parse_args()
if options.file_name:
compte = extrait_from_file(options.file_name)[1:]
print(compte)
compte_normalise = normalise(compte)
final = echange(compte_normalise)
affiche_final(final)
# ------------------------------
# Fin du programme
# ------------------------------
# -----------------------------
# Reglages pour 'vim'
# vim:set autoindent expandtab tabstop=4 shiftwidth=4:
# cursor: 16 del