Refactor NFSeSoapClient

marcelobelli · Oct 27, 2018 · dcd0a07 · dcd0a07
1 parent bd54110
commit dcd0a07
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 16 deletions.
diff --git a/PyNFSe/base/cliente_comunicacao.py b/PyNFSe/base/cliente_comunicacao.py
diff --git a/PyNFSe/base/soap_client.py b/PyNFSe/base/soap_client.py
@@ -0,0 +1,13 @@
+from requests import Session
+from zeep import Client
+from zeep.transports import Transport
+
+
+class NFSeSoapClient(Client):
+
+    def __init__(self, url_env: str, certificate: tuple, is_production: bool=False):
+        session = Session()
+        session.cert = certificate
+        session.verify = is_production
+
+        super().__init__(url_env, transport=Transport(session=session))
diff --git a/PyNFSe/nfse/pr/curitiba/comunicacao.py b/PyNFSe/nfse/pr/curitiba/comunicacao.py
@@ -1,10 +1,10 @@
-from PyNFSe.base.cliente_comunicacao import ClienteComunicacao
+from PyNFSe.base.soap_client import NFSeSoapClient
 
 
 class Comunicacao:
 
     def __init__(self, url_ambiente, certificado, producao):
-        self._cliente = ClienteComunicacao(url_ambiente, certificado, producao)
+        self._cliente = NFSeSoapClient(url_ambiente, certificado, producao)
 
     def validar_xml(self, xml):
         return self._cliente.service.ValidarXml(xml)