ajuste busca de outros documentos no consulta cadastro (#363)

* ajuste busca de outros documentos no consulta cadastro * fix: ajuste tipo de documento no parametro * ajuste uppercase no tipo do documento * ajuste uppercase no tipo do documento --------- Co-authored-by: = <[email protected]>
TadaSoftware · Sep 25, 2024 · 5a74454 · 5a74454
1 parent 7a239bc
commit 5a74454
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/pynfe/processamento/comunicacao.py b/pynfe/processamento/comunicacao.py
@@ -238,11 +238,12 @@ def consulta_distribuicao(
 
         return self._post(url, xml)
 
-    def consulta_cadastro(self, modelo, cnpj):
+    def consulta_cadastro(self, modelo, documento, tipo='CNPJ'):
         """
         Consulta de cadastro
         :param modelo: Modelo da nota
-        :param cnpj: CNPJ da empresa
+        :param documento: Documento (CNPJ, CPF ou IE)
+        :tipo do documento: CNPJ, CPF, IE
         :return:
         """
         # UF que utilizam a SVRS - Sefaz Virtual do RS:
@@ -265,7 +266,10 @@ def consulta_cadastro(self, modelo, cnpj):
         info = etree.SubElement(raiz, "infCons")
         etree.SubElement(info, "xServ").text = "CONS-CAD"
         etree.SubElement(info, "UF").text = self.uf.upper()
-        etree.SubElement(info, "CNPJ").text = cnpj
+
+        # Monta tipo de documento CNPJ, CPF ou IE
+        etree.SubElement(info, tipo.upper()).text = documento
+
         # etree.SubElement(info, 'CPF').text = cpf
 
         # Monta XML para envio da requisição