Feat: turn Account to a single function

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 

pralo.py

@@ -17,8 +17,6 @@ import os
 import optparse
 from pathlib import Path
 
-from account import Account
-
 def extrait_from_file(file_name):
     """
     Extrait les informations à partir d'un fichier
@@ -105,6 +103,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 +159,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_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)
             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)