Opción para especificar timeouts en las peticiones al webservice del SAT, 15 segs por defecto

Author: jahaziel-admintotal

cfdiclient/validacioncfdi.py

@@ -4,14 +4,16 @@
 
 
 class Validacion():
-  def __init__(self):
+  def __init__(self, verify=True, timeout=15):
     self.SOAP_URL = 'https://consultaqr.facturaelectronica.sat.gob.mx/ConsultaCFDIService.svc'
     self.SOAP_ACTION = 'http://tempuri.org/IConsultaCFDIService/Consulta'
     self.NSMAP = {
         's': 'http://schemas.xmlsoap.org/soap/envelope/',
         'des': 'http://DescargaMasivaTerceros.sat.gob.mx',
         'xd': 'http://www.w3.org/2000/09/xmldsig#'
     }
+    self.verify = verify
+    self.timeout = timeout
 
   def __generar_soapreq__(self, rfc_emisor, rfc_receptor, total, uuid):
      soapreq = ('<soapenv:Envelope xmlns:soapenv="http://schemas.xmlsoap.org/soap/envelope/" xmlns:tem="http://tempuri.org/">'
@@ -38,7 +40,13 @@ def obtener_estado(self, rfc_emisor, rfc_receptor, total, uuid):
         'SOAPAction': self.SOAP_ACTION,
     }
 
-    response = requests.post(self.SOAP_URL, data=soapreq, headers=headers, verify=True)
+    response = requests.post(
+        self.SOAP_URL, 
+        data=soapreq, 
+        headers=headers, 
+        verify=self.verify,
+        timeout=self.timeout,
+    )
 
 
     if response.status_code != requests.codes['ok']:

cfdiclient/webservicerequest.py

@@ -23,10 +23,11 @@ class WebServiceRequest(Utils):
 
     fault_xpath: str = 's:Body/s:Fault/faultstring'
 
-    def __init__(self, fiel: Fiel, verify: bool = True) -> None:
+    def __init__(self, fiel: Fiel, verify: bool = True, timeout: int = 15) -> None:
         super().__init__()
         self.signer = Signer(fiel)
         self.verify = verify
+        self.timeout = timeout
 
     def get_headers(self, token: str) -> dict:
         headers = {
@@ -74,6 +75,7 @@ def request(self, token: str = None, arguments: dict = None) -> etree.Element:
             data=soap_request,
             headers=headers,
             verify=self.verify,
+            timeout=self.timeout,
         )
 
         logger.debug('Response headers: %s', response.headers)