From 57ea98c7d7433b7aaf1d2b9add7ca74c321a34f3 Mon Sep 17 00:00:00 2001
From: Raphael Rodrigues Nogueira <raphael.rodrigues@gobtelecom.com.br>
Date: Thu, 9 Feb 2017 23:31:06 -0300
Subject: [PATCH] =?UTF-8?q?Altera=C3=A7=C3=B5es=20para=20emiss=C3=A3o=20de?=
 =?UTF-8?q?=20NFCe?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 pytrustnfe/nfe/__init__.py | 69 +++++++++++++++++++---------------------------
 1 file changed, 28 insertions(+), 41 deletions(-)

diff --git a/pytrustnfe/nfe/__init__.py b/pytrustnfe/nfe/__init__.py
index e2691d1..755f697 100644
--- a/pytrustnfe/nfe/__init__.py
+++ b/pytrustnfe/nfe/__init__.py
@@ -120,13 +120,10 @@ def _add_qrCode(xml, **kwargs):
     nfe.insert(1, infnfesupl)
     return etree.tostring(xml)
 
-
-def _send(certificado, method, sign, **kwargs):
+def _send(certificado, method, sign, mod='55', **kwargs):
     path = os.path.join(os.path.dirname(__file__), 'templates')
     xmlElem_send = render_xml(path, '%s.xml' % method, True, **kwargs)
-    modelo = xmlElem_send.find(".//{http://www.portalfiscal.inf.br/nfe}mod")
-    modelo = modelo.text if modelo is not None else '55'
-    if modelo == '65':
+    if mod == '65'and method == 'NfeAutorizacao':
         pagamento = etree.Element('pag')
         tipo_pagamento = etree.Element('tPag')
         valor = etree.Element('vPag')
@@ -165,13 +162,13 @@ def _send(certificado, method, sign, **kwargs):
             xml_send = signer.assina_xml(
                 xmlElem_send, kwargs['Id'])
 
-        if modelo == '65':
+        if mod == '65':
             xml_send = _add_qrCode(xml_send, **kwargs)
 
     else:
         xml_send = etree.tostring(xmlElem_send)
 
-    url = localizar_url(method,  kwargs['estado'], modelo,
+    url = localizar_url(method,  kwargs['estado'], mod,
                         kwargs['ambiente'])
     cabecalho = _build_header(method, **kwargs)
 
@@ -183,50 +180,40 @@ def _send(certificado, method, sign, **kwargs):
     }
 
 
-def autorizar_nfe(certificado, **kwargs):  # Assinar
+def autorizar_nfe(certificado, mod='55', **kwargs):  # Assinar
     _generate_nfe_id(**kwargs)
-    return _send(certificado, 'NfeAutorizacao', True, **kwargs)
-
-
-def retorno_autorizar_nfe(certificado, **kwargs):
-    return _send(certificado, 'NfeRetAutorizacao', False, **kwargs)
-
-
-def recepcao_evento_cancelamento(certificado, **kwargs):  # Assinar
-    return _send(certificado, 'RecepcaoEventoCancelamento', True, **kwargs)
-
-
-def inutilizar_nfe(certificado, **kwargs):  # Assinar
-    return _send(certificado, 'NfeInutilizacao', True, **kwargs)
-
-
-def consultar_protocolo_nfe(certificado, **kwargs):
-    return _send(certificado, 'NfeConsultaProtocolo', True, **kwargs)
-
-
-def nfe_status_servico(certificado, **kwargs):
-    return _send(certificado, 'NfeStatusServico', False, **kwargs)
+    return _send(certificado, 'NfeAutorizacao', True, mod, **kwargs)
 
+def retorno_autorizar_nfe(certificado, mod='55', **kwargs):
+    return _send(certificado, 'NfeRetAutorizacao', False, mod, **kwargs)
 
-def consulta_cadastro(certificado, **kwargs):
-    return _send(certificado, 'NfeConsultaCadastro', False, **kwargs)
 
+def recepcao_evento_cancelamento(certificado, mod='55', **kwargs):  # Assinar
+    return _send(certificado, 'RecepcaoEventoCancelamento', True, mod, **kwargs)
 
-def recepcao_evento_carta_correcao(certificado, **kwargs):  # Assinar
-    return _send(certificado, 'RecepcaoEventoCarta', True, **kwargs)
+def inutilizar_nfe(certificado, mod='55', **kwargs):  # Assinar
+    return _send(certificado, 'NfeInutilizacao', mod, True, **kwargs)
 
+def consultar_protocolo_nfe(certificado, mod='55', **kwargs):
+    return _send(certificado, 'NfeConsultaProtocolo', True, mod, **kwargs)
 
-def recepcao_evento_manifesto(certificado, **kwargs):  # Assinar
-    return _send(certificado, 'RecepcaoEventoManifesto', True, **kwargs)
+def nfe_status_servico(certificado, mod='55', **kwargs):
+    return _send(certificado, 'NfeStatusServico', False, mod, **kwargs)
 
+def consulta_cadastro(certificado, mod='55', **kwargs):
+    return _send(certificado, 'NfeConsultaCadastro', False, mod, **kwargs)
 
-def recepcao_evento_epec(certificado, **kwargs):  # Assinar
-    return _send(certificado, 'RecepcaoEventoEPEC', True, **kwargs)
+def recepcao_evento_carta_correcao(certificado, mod='55', **kwargs):  # Assinar
+    return _send(certificado, 'RecepcaoEventoCarta', True, mod, **kwargs)
 
+def recepcao_evento_manifesto(certificado, mod='55', **kwargs):  # Assinar
+    return _send(certificado, 'RecepcaoEventoManifesto', True, mod, **kwargs)
 
-def consulta_nfe_destinada(certificado, **kwargs):
-    return _send(certificado, 'NfeConsultaDest', False, **kwargs)
+def recepcao_evento_epec(certificado, mod='55', **kwargs):  # Assinar
+    return _send(certificado, 'RecepcaoEventoEPEC', True, mod, **kwargs)
 
+def consulta_nfe_destinada(certificado, mod='55', **kwargs):
+    return _send(certificado, 'NfeConsultaDest', False, mod, **kwargs)
 
-def download_nfe(certificado, **kwargs):
-    return _send(certificado, 'NfeDownloadNF', False, **kwargs)
+def download_nfe(certificado, mod='55', **kwargs):
+    return _send(certificado, 'NfeDownloadNF', False, mod, **kwargs)