diff --git a/pynfe/processamento/comunicacao.py b/pynfe/processamento/comunicacao.py index 8c2a272..ec4d0c7 100644 --- 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