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)