#! /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 :param file_name: nom du fichier avec les comptes :return: liste [[nom,montant,jours],...] """ file = open(file_name, "r") try: reader = csv.reader(file, delimiter = ",") r_comptes = list(reader) header = r_comptes.pop(0) # On récupère le premier élément (les headers) comptes = [] for c in r_comptes: # On parcourt les lignes pour typer les éléments correctement compte = {} for (i,h) in enumerate(header): if "nom" in h.lower(): compte["nom"] = c[i] if "montant" in h.lower(): compte["montant"] = eval(c[i]) if "jour" in h.lower(): compte["jours"] = int(c[i]) comptes += [compte] return comptes finally: file.close() def echange(l, seuil = 0): """ Algo recursif pour gérer les échanges :param l: liste des valeurs associées aux débits des comptes :param seuil: seuil à partir duquel on ne doit plus rien. :return: liste des valeurs associées aux débits des comptes après le remboursement """ 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] # Gestion du Seuil l = [i for i in l if abs(i[1]) >= seuil] if (len(l) > 1): #print(l) return [res] + echange(l,seuil=seuil) else: return [res] def forfait(compte): """ Gère le nombre resté à Pralo :param compte: comptes de ce que chacun a payé [[nom, montant,jours],...] :return: Comptes des crédits de chacun envers la communauté """ cout_total = sum([c["montant"] for c in compte]) print("Cout total de Pralo: {cout_tot}".format(cout_tot = cout_total)) nbr_jour = sum([c["jours"] for c in compte]) print("Le nombre de jour passé {njour}".format(njour = nbr_jour)) cout_jour = cout_total / nbr_jour print("Cout au jour: {cout}".format(cout = round(cout_jour,2))) print("\n") new_compte = [] for pers in compte: new_compte += [[pers["nom"], (pers["montant"] - pers["jours"] * cout_jour)]] return new_compte def normalise(compte): """Centre en 0 les comptes""" moyenne = sum([c["montant"] for c in compte]) / len(compte) compte_normalise = compte for (i,n) in enumerate(compte): compte_normalise[i]["montant"] = compte[i]["montant"] - moyenne return compte_normalise def affiche_final(donRec): """ Affiche qui donne quoi à qui à partir de la liste :param donRec: liste avec qui doit quoi à qui [[qui, àqui, quoi]...] """ for g in donRec: print("{don} donne {montant} à {rec}".format(don = g[0], rec = g[1], montant = round(g[2],2))) # ------------------------------ # 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") parser.add_option("-e","--seuil",action="store", type = "int", dest="seuil",default=0, help="Seuil à partir duquel on concidère qu'il n'est plus nécessaire de payer.") # Digestion (options, args) = parser.parse_args() if options.file_name: compte = extrait_from_file(options.file_name) # compte_normalise = normalise(compte) compte_normalise = forfait(compte) final = echange(compte_normalise, options.seuil) affiche_final(final) # ------------------------------ # Fin du programme # ------------------------------ # ----------------------------- # Reglages pour 'vim' # vim:set autoindent expandtab tabstop=4 shiftwidth=4: # cursor: 16 del