Compare commits
4 Commits
d7ceff8bb8
...
master
| Author | SHA1 | Date | |
|---|---|---|---|
| a260bfc2c6 | |||
| 5c3a7f9b4f | |||
| a4feb1c310 | |||
| bf34d77bb0 |
3
TODO
3
TODO
@@ -1,3 +0,0 @@
|
||||
Idées d'amélioration:
|
||||
- Gérer quand un même nom apparait plusieurs fois dans la liste
|
||||
- Pondre un document .tex près à compiler pour tout résumer
|
||||
64
account.py
64
account.py
@@ -1,64 +0,0 @@
|
||||
#!/usr/bin/env python
|
||||
# encoding: utf-8
|
||||
|
||||
|
||||
class Account(object):
|
||||
"""@todo"""
|
||||
|
||||
def __init__(self, account):
|
||||
"""Initiate the account as a list
|
||||
|
||||
:param account: list of tuples (length 2 with name and credit) [("bob", 4), ("hip",2)...]
|
||||
|
||||
"""
|
||||
self._account = account
|
||||
|
||||
def tribut(self, seuil = 0):
|
||||
"""Tells who owns what to who (recursive algorithm
|
||||
|
||||
: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
|
||||
"""
|
||||
self._account.sort(key = lambda s: s[1])
|
||||
m = self._account[0]
|
||||
M = self._account[-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
|
||||
self._account.remove(m)
|
||||
# ON change la valeur du max
|
||||
self._account[-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
|
||||
self._account.remove(M)
|
||||
# On change la valeur du min
|
||||
self._account[0] = [m[0], m2]
|
||||
|
||||
# Gestion du Seuil
|
||||
self._account = [i for i in self._account if abs(i[1]) >= seuil]
|
||||
|
||||
if (len(self._account) > 1):
|
||||
#print(self._account)
|
||||
return [res] + self.tribut(seuil=seuil)
|
||||
else:
|
||||
return [res]
|
||||
|
||||
def __add__(self, account):
|
||||
pass
|
||||
|
||||
# -----------------------------
|
||||
# Reglages pour 'vim'
|
||||
# vim:set autoindent expandtab tabstop=4 shiftwidth=4:
|
||||
# cursor: 16 del
|
||||
@@ -1,5 +0,0 @@
|
||||
nom,montant, jours
|
||||
pipo, 20 + 3, 3
|
||||
hiphop, 40, 4
|
||||
chic, 0, 4
|
||||
Couette,50, 1
|
||||
|
87
pralo.py
87
pralo.py
@@ -17,18 +17,15 @@ import os
|
||||
import optparse
|
||||
from pathlib import Path
|
||||
|
||||
from account import Account
|
||||
|
||||
def extrait_from_file(file_name):
|
||||
def account_from_file(file_name):
|
||||
"""
|
||||
Extrait les informations à partir d'un fichier
|
||||
Extract account from csv file
|
||||
|
||||
:param file_name: nom du fichier avec les comptes
|
||||
:return: liste [[nom,montant,parts],...]
|
||||
:param file_name: csv file name
|
||||
:return: list of dictionnaries [{nom:... ,montant:... ,parts:... },...]
|
||||
"""
|
||||
file = open(file_name, "r")
|
||||
try:
|
||||
reader = csv.reader(file, delimiter = ",")
|
||||
with open(file_name, "r") as f:
|
||||
reader = csv.reader(f, delimiter = ",")
|
||||
r_comptes = list(reader)
|
||||
header = r_comptes.pop(0) # On récupère le premier élément (les headers)
|
||||
|
||||
@@ -38,7 +35,7 @@ def extrait_from_file(file_name):
|
||||
compte = {}
|
||||
for (i,h) in enumerate(header):
|
||||
if "nom" in h.lower():
|
||||
compte["nom"] = c[i]
|
||||
compte["nom"] = c[i].capitalize()
|
||||
if "montant" in h.lower():
|
||||
compte["montant"] = eval(c[i])
|
||||
if "part" in h.lower():
|
||||
@@ -46,15 +43,13 @@ def extrait_from_file(file_name):
|
||||
comptes += [compte]
|
||||
|
||||
return comptes
|
||||
finally:
|
||||
file.close()
|
||||
|
||||
def forfait(compte, output = print):
|
||||
"""
|
||||
Gère le nombre de parts et affiche quelques stats
|
||||
Affiche quelques stats et calcule les crédits et dettes
|
||||
|
||||
:param compte: comptes de ce que chacun a payé [[nom, montant,parts],...]
|
||||
:return: Comptes des crédits de chacun envers la communauté
|
||||
:param compte: list of dictionnaries [{nom:... ,montant:... ,parts:... },...]
|
||||
:return: list of tuples [(name, cred_dept), ...]
|
||||
"""
|
||||
cout_total = sum([c["montant"] for c in compte])
|
||||
output("Cout total: {cout_tot}".format(cout_tot = cout_total))
|
||||
@@ -67,11 +62,11 @@ def forfait(compte, output = print):
|
||||
|
||||
output("\n")
|
||||
|
||||
new_compte = []
|
||||
cred_dept = []
|
||||
for pers in compte:
|
||||
new_compte += [[pers["nom"], (pers["montant"] - pers["parts"] * cout_jour)]]
|
||||
cred_dept += [[pers["nom"], (pers["montant"] - pers["parts"] * cout_jour)]]
|
||||
|
||||
return new_compte
|
||||
return cred_dept
|
||||
|
||||
def normalise(compte):
|
||||
"""Centre en 0 les comptes"""
|
||||
@@ -105,6 +100,48 @@ def join_comptes(compte1, compte2):
|
||||
joined = {k: c_dict1.get(k, 0) + c_dict2.get(k, 0) for k in set(c_dict1) | set(c_dict2)}
|
||||
return [(k, v) for k,v in joined.items()]
|
||||
|
||||
def tribut(account, seuil = 0):
|
||||
"""Tells who owns what to who (recursive algorithm)
|
||||
|
||||
: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
|
||||
"""
|
||||
account.sort(key = lambda s: s[1])
|
||||
m = account[0]
|
||||
M = account[-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
|
||||
account.remove(m)
|
||||
# ON change la valeur du max
|
||||
account[-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
|
||||
account.remove(M)
|
||||
# On change la valeur du min
|
||||
account[0] = [m[0], m2]
|
||||
|
||||
# Gestion du Seuil
|
||||
account = [i for i in account if abs(i[1]) >= seuil]
|
||||
|
||||
if (len(account) > 1):
|
||||
#print(account)
|
||||
return [res] + tribut(account, seuil=seuil)
|
||||
else:
|
||||
return [res]
|
||||
|
||||
if __name__ == '__main__':
|
||||
# Pour analyser les options qu'on lui demande
|
||||
parser = optparse.OptionParser()
|
||||
@@ -119,22 +156,18 @@ if __name__ == '__main__':
|
||||
path = Path(options.file_name)
|
||||
assert path.exists()
|
||||
if path.is_dir():
|
||||
god = []
|
||||
cred_dept = []
|
||||
for f in path.glob('*.csv'):
|
||||
print(f"Compte: {f.name}")
|
||||
compte = extrait_from_file(f)
|
||||
compte = account_from_file(f)
|
||||
compte_normalise = forfait(compte)
|
||||
god = join_comptes(god, compte_normalise)
|
||||
cred_dept = join_comptes(cred_dept, compte_normalise)
|
||||
|
||||
account = Account(god)
|
||||
else:
|
||||
compte = extrait_from_file(options.file_name)
|
||||
# compte_normalise = normalise(compte)
|
||||
compte = account_from_file(options.file_name)
|
||||
cred_dept = forfait(compte)
|
||||
#final = echange(compte_normalise, options.seuil)
|
||||
account = Account(cred_dept)
|
||||
|
||||
final = account.tribut(options.seuil)
|
||||
final = tribut(cred_dept, options.seuil)
|
||||
affiche_final(final)
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user