Adicionado utilitário que sobrescreve métodos da lib suds para comunicação via wsdl

pynfe/processamento/comunicacao.py

@@ -1,6 +1,5 @@
 # -*- coding: utf-8 -*-
 import datetime
-import time
 import requests
 from pynfe.utils import etree, so_numeros
 from pynfe.utils.flags import NAMESPACE_NFE, NAMESPACE_SOAP, NAMESPACE_XSI, NAMESPACE_XSD, NAMESPACE_METODO, VERSAO_PADRAO, CODIGOS_ESTADOS, \
@@ -9,6 +8,7 @@ from pynfe.utils.webservices import NFCE, NFE, NFSE
 from .assinatura import AssinaturaA1
 from pynfe.entidades.certificado import CertificadoA1
 
+
 class Comunicacao(object):
     u"""Classe abstrata responsavel por definir os metodos e logica das classes
     de comunicação com os webservices da NF-e."""
@@ -386,7 +386,7 @@ class ComunicacaoNfse(Comunicacao):
         # url do serviço
         url = self._get_url(autorizador)
         # comunica via wsdl
-        return self._post2(url, xml, 'consulta')
+        return self._post_https(url, xml, 'consulta')
 
     def consulta_rps(self, autorizador, xml):
         if autorizador.upper() == 'BETHA':
@@ -445,45 +445,45 @@ class ComunicacaoNfse(Comunicacao):
             raise Exception('Autorizador nao encontrado!')
         return self.url
 
-    def _post(self, url, xml):
-        certificadoA1 = CertificadoA1(self.certificado)
-        chave, cert = certificadoA1.separar_arquivo(self.certificado_senha, caminho=True)
-        chave_cert = (cert, chave)
-
-        # Abre a conexão HTTPS
+    def _post(self, url, xml, metodo):
+        # cabecalho
+        cabecalho = self._cabecalho()
+        # comunicacao wsdl
         try:
-            xml_declaration='<?xml version="1.0" encoding="utf-8"?>'
-            #xml = etree.tostring(xml, encoding='unicode', pretty_print=False).replace('\n','').replace('ns0:','soapenv:').replace(':ns0',':soapenv')
-            xml = etree.tostring(xml, encoding='unicode', pretty_print=False).replace('\n','').replace('ns0:','').replace(':ns0','')
-            xml = xml_declaration + xml
-
-            print (url)
-            print (xml)
-            import ipdb
-            ipdb.set_trace()
-
-            # Faz o request com o servidor
-            result = requests.post(url, xml, headers=self._post_header(), cert=chave_cert, verify=False)
-            result.encoding='utf-8'
-            return result
-        except requests.exceptions.ConnectionError as e:
+            from suds.client import Client
+            cliente = Client(url)
+            # gerar nfse
+            if metodo == 'gerar':
+                return cliente.service.GerarNfse(cabecalho, xml)
+            elif metodo == 'consultaRps':
+                return cliente.service.ConsultarNfsePorRps(cabecalho, xml)
+            elif metodo == 'consultaFaixa':
+                return cliente.service.ConsultarNfseFaixa(cabecalho, xml)
+            elif metodo == 'cancelar':
+                return cliente.service.CancelarNfse(cabecalho, xml)
+            # TODO outros metodos
+            else:
+                pass
+        except Exception as e:
             raise e
-        finally:
-            certificadoA1.excluir()
 
-    def _post2(self, url, xml, metodo):
+    def _post_https(self, url, xml, metodo):
         # cabecalho
         cabecalho = self._cabecalho()
         # comunicacao wsdl
         try:
             from suds.client import Client
-            cliente = Client(url)
+            from pynfe.utils.https_nfse import HttpAuthenticated
+            
+            certificadoA1 = CertificadoA1(self.certificado)
+            chave, cert = certificadoA1.separar_arquivo(self.certificado_senha, caminho=True)
+
+            cliente = Client(url, transport = HttpAuthenticated(key=chave, cert=cert, endereco=url))
+
             # gerar nfse
             if metodo == 'gerar':
                 return cliente.service.GerarNfse(cabecalho, xml)
             elif metodo == 'consulta':
-                import ipdb
-                ipdb.set_trace()
                 return cliente.service.ConsultarNfsePorRps(cabecalho, xml)
             elif metodo == 'consultaRps':
                 return cliente.service.ConsultarNfsePorRps(cabecalho, xml)

pynfe/utils/https_nfse.py

@@ -0,0 +1,38 @@
+
+"""
+    @author: Junior Tada, Leonardo Tada
+
+    Classe que sobrescreve metodos da lib suds para comunicação via https com certificado digital
+
+"""
+
+from suds.transport.http import HttpTransport
+import urllib.request, http.client
+
+
+class HTTPSClientAuthHandler(urllib.request.HTTPSHandler):
+    def __init__(self, key, cert):
+        urllib.request.HTTPSHandler.__init__(self)
+        self.key = key
+        self.cert = cert
+
+    def https_open(self, req):
+        # Rather than pass in a reference to a connection class, we pass in
+        # a reference to a function which, for all intents and purposes,
+        # will behave as a constructor
+        return self.do_open(self.getConnection, req)
+
+    def getConnection(self, host, timeout=300):
+        return http.client.HTTPSConnection(host, key_file=self.key, cert_file=self.cert)
+
+
+class HttpAuthenticated(HttpTransport):
+    def __init__(self, key, cert, endereco, *args, **kwargs):
+        HttpTransport.__init__(self, *args, **kwargs)
+        self.key = key
+        self.cert = cert
+        self.endereco = endereco
+
+    def open(self, request):
+        opener = urllib.request.build_opener(HTTPSClientAuthHandler(self.key, self.cert))
+        return opener.open(self.endereco)

pynfe/utils/webservices.py

@@ -435,7 +435,6 @@ NFSE = {
 		'CONSULTA_SERVICO_TOMADO':'ConsultarNfseServicoTomado',
 		'SUBSTITUIR':'SubstituirNfse',
 		'HTTPS':'http://e-gov.betha.com.br/e-nota-contribuinte-ws/nfseWS?wsdl',
-		#'HOMOLOGACAO':'http://e-gov.betha.com.br/e-nota-contribuinte-test-ws/'
 		'HOMOLOGACAO':'http://e-gov.betha.com.br/e-nota-contribuinte-test-ws/nfseWS?wsdl'
 	},
 	#