Optimized Rounding

Author: satcfdi

satcfdi/create/cfd/cfdi40.py

@@ -335,7 +335,6 @@ def __post_init__(self):
 def _make_conceptos(conceptos, decimals):
     rnd_fn = lambda v: round(v, decimals)
     rnd_traslados_tracker = RoundTracker(decimals)
-    rnd_retenciones_tracker = RoundTracker(decimals)
 
     def make_concepto(concepto):
         impuestos = concepto.get("Impuestos") or {}
@@ -353,9 +352,9 @@ def make_concepto(concepto):
         else:
             impuestos = {
                 imp_t: [
-                    make_impuesto(i, base=base, rnd_tracker=rnd_tracker) for i in imp
+                    make_impuesto(i, base=base, rnd_fn=rnd_tracker) for i in imp
                 ]
-                for imp_t, imp, rnd_tracker in [('Traslados', trasladados, rnd_traslados_tracker), ('Retenciones', retenciones, rnd_retenciones_tracker)] if imp
+                for imp_t, imp, rnd_tracker in [('Traslados', trasladados, rnd_traslados_tracker), ('Retenciones', retenciones, rnd_fn)] if imp
             }
             concepto['Impuestos'] = impuestos or None
             concepto["ObjetoImp"] = "02" if impuestos else "01"

satcfdi/create/compute.py

@@ -50,7 +50,7 @@ def encode_impuesto(impuesto, tipo_factor, tasa_cuota: Decimal = None):
     return impuesto
 
 
-def make_impuesto(impuesto: dict, base, rnd_tracker):
+def make_impuesto(impuesto: dict, base, rnd_fn):
     _impuesto = impuesto["Impuesto"]
     tipo_factor = impuesto['TipoFactor']
     tasa_cuota = impuesto['TasaOCuota']
@@ -62,7 +62,7 @@ def make_impuesto(impuesto: dict, base, rnd_tracker):
     else:
         match tipo_factor:
             case "Tasa":
-                importe = rnd_tracker.round(base * tasa_cuota)
+                importe = rnd_fn(base * tasa_cuota)
             case "Cuota":
                 importe = tasa_cuota
             case "Exento":
@@ -102,6 +102,9 @@ def peak(self, value):
             rounded = round(value, self.decimals)
         return rounded
 
+    def __call__(self, value):
+        return self.round(value)
+
 
 def group_impuestos(elements, pfx="", ofx=""):
     retenciones = aggregate(