Compare commits

..

4 Commits

Author SHA1 Message Date
a260bfc2c6 Fix: remove TODO 2020-08-03 11:02:14 +02:00
5c3a7f9b4f Feat: cleaning 2020-08-03 11:01:21 +02:00
a4feb1c310 Feat: turn Account to a single function 2020-08-03 10:52:14 +02:00
bf34d77bb0 Feat: remove old example 2020-08-03 10:47:55 +02:00
4 changed files with 60 additions and 99 deletions

3
TODO
View File

@@ -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

View File

@@ -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

View File

@@ -1,5 +0,0 @@
nom,montant, jours
pipo, 20 + 3, 3
hiphop, 40, 4
chic, 0, 4
Couette,50, 1
1 nom montant jours
2 pipo 20 + 3 3
3 hiphop 40 4
4 chic 0 4
5 Couette 50 1

View File

@@ -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)