Fixed Accounting Logic, Added Base 16, status Retenciones

satcfdi · satcfdi · commit 06578a8f2a6d · 2025-02-17T19:43:47.000-06:00
diff --git a/satcfdi/accounting/models.py b/satcfdi/accounting/models.py
@@ -104,7 +104,7 @@ class PaymentsDetails(Payment):
     def __post_init__(self):
         if self.pago:
             self.impuestos = self.docto_relacionado.get('ImpuestosDR')
-            if self.impuestos is None:
+            if self.impuestos is None and self.comprobante['Version'] == '3.3':
                 self.impuestos = make_impuestos_dr_parcial(
                     conceptos=self.comprobante_pagado['Conceptos'],
                     imp_saldo_ant=self.docto_relacionado['ImpSaldoAnt'],
diff --git a/satcfdi/accounting/process.py b/satcfdi/accounting/process.py
@@ -130,6 +130,7 @@ def invoice_def():
 
         'SubTotal': (12, True, lambda i: i["SubTotal"]),
         'Descuento': (12, True, lambda i: i.get("Descuento")),
+        'IVA16 Base': (12, True, lambda i: i.get("Impuestos", {}).get("Traslados", {}).get(IVA16, {}).get("Base")),
         'IVA16 Tras': (12, True, lambda i: i.get("Impuestos", {}).get("Traslados", {}).get(IVA16, {}).get("Importe")),
         'IVA Ret': (12, True, lambda i: i.get("Impuestos", {}).get("Retenciones", {}).get(Impuesto.IVA, {}).get("Importe")),
         'ISR Ret': (12, True, lambda i: i.get("Impuestos", {}).get("Retenciones", {}).get(Impuesto.ISR, {}).get("Importe")),
@@ -153,6 +154,7 @@ def invoice_confirm_def():
 
         'SubTotal': (12, True, lambda i: i["SubTotal"]),
         'Descuento': (12, True, lambda i: i.get("Descuento")),
+        'IVA16 Base': (12, True, lambda i: i.get("Impuestos", {}).get("Traslados", {}).get(IVA16, {}).get("Base")),
         'IVA16 Tras': (12, True, lambda i: i.get("Impuestos", {}).get("Traslados", {}).get(IVA16, {}).get("Importe")),
         'IVA Ret': (12, True, lambda i: i.get("Impuestos", {}).get("Retenciones", {}).get(Impuesto.IVA, {}).get("Importe")),
         'ISR Ret': (12, True, lambda i: i.get("Impuestos", {}).get("Retenciones", {}).get(Impuesto.ISR, {}).get("Importe")),
@@ -180,6 +182,7 @@ def payment_def():
         'Parcialidad': (12, False, lambda i: i.docto_relacionado["NumParcialidad"] if i.pago else None),
         'Subtotal': (12, True, lambda i: i.sub_total),
         'Descuento': (12, True, lambda i: i.descuento),
+        'IVA16 Base': (12, True, lambda i: i.impuestos.get("Traslados", {}).get(IVA16, {}).get("Base")),
         'IVA16 Tras': (12, True, lambda i: i.impuestos.get("Traslados", {}).get(IVA16, {}).get("Importe")),
         'IVA Ret': (12, True, lambda i: i.impuestos.get("Retenciones", {}).get(Impuesto.IVA, {}).get("Importe")),
         'ISR Ret': (12, True, lambda i: i.impuestos.get("Retenciones", {}).get(Impuesto.ISR, {}).get("Importe")),
diff --git a/satcfdi/pacs/sat.py b/satcfdi/pacs/sat.py
@@ -12,6 +12,8 @@
 from itertools import islice
 from typing import Iterator
 from uuid import UUID
+
+from bs4 import BeautifulSoup
 from packaging import version
 
 import requests
@@ -432,6 +434,28 @@ def status(self, cfdi: CFDI) -> dict:
             for i in res
         }
 
+    def status_retencion(self, cfdi: CFDI) -> dict:
+        rfc_emisor = cfdi["Emisor"]["RfcE"]
+        rfc_receptor = cfdi["Receptor"]["Nacional"]["RfcR"]
+        uuid = cfdi["Complemento"]["TimbreFiscalDigital"]["UUID"]
+
+        params = {
+            'folio': uuid,
+            'rfcEmisor': rfc_emisor,
+            'rfcReceptor': rfc_receptor
+        }
+
+        res = requests.get('https://prodretencionverificacion.clouda.sat.gob.mx/Home/ConsultaRetencion', params)
+
+        if res.status_code == 200:
+            soup = BeautifulSoup(res.text, 'html.parser')
+            status = soup.find('span', string='Cancelado').string
+            date = soup.find('td', id='tdFechaCancelacion').get_text(strip=True)
+
+        return {
+            "estado": status, "fecha": date
+        }
+
     def validate(self, cfdi: CFDI):
         """
                 verify if the CFDI is valid based on its signatures and certificates
