[REF] barueri analisa_retorno_consulta

Author: marcelsavegnago

src/erpbrasil/edoc/provedores/barueri.py

@@ -9,7 +9,6 @@
 from erpbrasil.base import misc
 from erpbrasil.edoc.nfse import NFSe, ServicoNFSe
 from erpbrasil.edoc.resposta import RetornoSoap
-from zeep.helpers import serialize_object
 
 try:
     from urllib.parse import urljoin
@@ -18,7 +17,6 @@
 
 try:
     from nfselib.barueri import (
-        ConsultarNFeRecebidaNumero,
         NFeLoteEnviarArquivo,
         NFeLoteStatusArquivo,
         NFeLoteBaixarArquivo,
@@ -29,11 +27,6 @@
     barueri = False
 
 endpoint = "nfeservice/wsrps.asmx?WSDL"
-nsmap = {
-    "consulta": "http://www.barueri.sp.gov.br/nfe/ConsultaNFeRecebidaNumero"
-    "ConsultaNFeRecebidaNumero.v1.xsd",
-    "tipo": "https://servicos.barueri.sp.gov.br/nfewsxml/wsgeraxml.asmx?op=ConsultaNFeRecebidaNumero",
-}
 
 if barueri:
     servicos = {
@@ -50,7 +43,7 @@
             "NFeLoteBaixarArquivo", endpoint, NFeLoteBaixarArquivo, True
         ),
         "consulta_nfse_rps": ServicoNFSe(
-            "ConsultaNFeRecebidaNumero", 'nfewsxml/wsgeraxml.asmx?WSDL', ConsultarNFeRecebidaNumero, True
+            "NFeLoteStatusArquivo", endpoint, NFeLoteStatusArquivo, True
         ),
     }
 else:
@@ -119,100 +112,44 @@ def _prepara_cancelar_nfse_envio(self, doc_numero):
         pass
 
     def _prepara_consultar_nfse_rps(self, **kwargs):
-        rps_numero = kwargs.get("rps_number")
-        raiz = ConsultarNFeRecebidaNumero.NFeRecebidaNumero(
-            CPFCNPJTomador=self.cnpj_prestador,
-            CPFCNPJPrestador=self.cnpj_prestador,
-            NumeroNota=rps_numero,
+        protocolo = kwargs.get("lot_receipt_number")
+        raiz = NFeLoteStatusArquivo.NFeLoteStatusArquivo(
+                CPFCNPJContrib=self.cnpj_prestador, InscricaoMunicipal=self.im_prestador
+            ,
+            ProtocoloRemessa=protocolo,
         )
         return raiz
 
-    def analisa_retorno_consulta(
-        self, processo, number, company_cnpj_cpf, company_legal_name
-    ):
-        data = serialize_object(processo.resposta, target_cls=dict)
-        root = etree.Element("ConsultarNfeResposta")
-
-        if processo.webservice == "ConsultaNFeRecebidaNumero":
-            # if "ListaNfe" in data:
-            #     lista = etree.SubElement(root, "ListaNfe")
-            #     for nfe in data["ListaNfe"]:
-            #         item = etree.SubElement(lista, "CompNfe")
-            #         for k, v in nfe.items():
-            #             etree.SubElement(item, k).text = str(v)
-
-            if "ListaMensagemRetorno" in data:
-                msgs = etree.SubElement(root, "ListaMensagemRetorno")
-                for msg in data["ListaMensagemRetorno"]:
-                    item = etree.SubElement(msgs, "MensagemRetorno")
-                    for k, v in msg.items():
-                        etree.SubElement(item, k).text = str(v)
-                mensagem = ""
-
-            retorno = root
-            enviado = retorno.findall(".//CompNfe")
-            nao_encontrado = retorno.findall(".//MensagemRetorno")
-            # TODO: PAREI AQUI
-
-            if enviado:
-                # NFS-e já foi enviada
-
-                cancelada = retorno.findall(
-                    ".//tipo:NfseCancelamento", namespaces=nsmap
-                )
-
-                if cancelada:
-                    # NFS-e enviada foi cancelada
-
-                    data = retorno.findall(".//tipo:DataHora", namespaces=nsmap)[0].text
-                    data = datetime.strptime(data, "%Y-%m-%dT%H:%M:%S").strftime(
-                        "%m/%d/%Y"
+    def analisa_retorno_consulta(self, processo):
+        mensagem = ""
+        if processo.webservice == "NFeLoteStatusArquivo":
+
+            if processo.resposta:
+                lista_msgs = processo.resposta.ListaMensagemRetorno
+
+                if lista_msgs.Codigo != "OK200":
+                    mensagem += (
+                        lista_msgs.Codigo
+                        + " - "
+                        + lista_msgs.Mensagem
+                        + " - Correção: "
+                        + lista_msgs.Correcao
+                        + "\n"
                     )
-                    mensagem = "NFS-e cancelada em " + data
-
                 else:
-                    numero_retorno = retorno.findall(
-                        ".//tipo:InfNfse/tipo:Numero", namespaces=nsmap
-                    )[0].text
-                    cnpj_prestador_retorno = retorno.findall(
-                        ".//tipo:IdentificacaoPrestador/tipo:Cnpj", namespaces=nsmap
-                    )[0].text
-                    razao_social_prestador_retorno = retorno.findall(
-                        ".//tipo:PrestadorServico/tipo:RazaoSocial", namespaces=nsmap
-                    )[0].text
-
-                    variables_error = []
-
-                    if numero_retorno != number:
-                        variables_error.append("Número")
-                    if cnpj_prestador_retorno != misc.punctuation_rm(company_cnpj_cpf):
-                        variables_error.append("CNPJ do prestador")
-                    if razao_social_prestador_retorno != company_legal_name:
-                        variables_error.append("Razão Social de prestador")
-
-                    if variables_error:
-                        mensagem = (
-                            "Os seguintes campos não condizem com"
-                            " o provedor NFS-e: \n"
-                        )
-                        mensagem += "\n".join(variables_error)
-                    else:
-                        mensagem = "NFS-e enviada e corresponde com o provedor"
-
-            elif nao_encontrado:
-                # NFS-e não foi enviada
-
-                mensagem_erro = msg.get('Mensagem')
-                correcao = msg.get('Correcao')
-                codigo = msg.get('Codigo')
-                mensagem = (
-                    codigo + " - " + mensagem_erro + " - Correção: " + correcao + "\n"
-                )
-
-            else:
-                mensagem = "Erro desconhecido."
-
-        return mensagem
+                    status = int(processo.resposta.ListaNfeArquivosRPS.SituacaoArq)
+                    if status == 0:
+                        mensagem = "Validated"
+                    elif status == 1:
+                        mensagem = "Successfully Processed"
+                    elif status == 2:
+                        mensagem = "Processed with Error"
+                    elif status == -1:
+                        mensagem = "Batch not yet processed"
+                    elif status == -2:
+                        mensagem = "Batch not yet processed"
+
+        return status, mensagem
 
     def analisa_retorno_cancelamento(self, processo):
         pass