Implementação nota fiscal

pytrustnfe/Servidores.py

@@ -23,11 +23,11 @@ NFE_AMBIENTE_HOMOLOGACAO = 2
 
 
 def localizar_url(servico, estado):
-    return ESTADO_WS[estado]['servidor'], ESTADO_WS[estado][servico]
+    return ESTADO_WS[estado][2]['servidor'], ESTADO_WS[estado][2][servico]
 
 
 METODO_WS = {
-    WS_NFE_AUTORIZACAO:{
+    WS_NFE_AUTORIZACAO: {
         'webservice': 'NfeAutorizacao',
         'metodo': 'NfeAutorizacao',
     },

pytrustnfe/nfe/__init__.py

@@ -16,6 +16,7 @@ def _build_header(**kwargs):
     vals = {'estado': kwargs['estado'], 'soap_action': ''}
     return CabecalhoSoap(**vals)
 
+
 def _generate_nfe_id(**kwargs):
     for item in kwargs['NFes']:
         vals = {
@@ -33,66 +34,73 @@ def _generate_nfe_id(**kwargs):
         item['infNFe']['Id'] = gerar_chave(chave, 'NFe')
 
 
-
 def _send(certificado, method, **kwargs):
     path = os.path.join(os.path.dirname(__file__), 'templates')
 
     xml = render_xml(path, '%s.xml' % method, **kwargs)
+    xml = '<!DOCTYPE NFe [<!ATTLIST infNFe Id ID #IMPLIED>]>' + xml
 
     pfx_path = certificado.save_pfx()
     signer = Assinatura(pfx_path, certificado.password)
     xml_signed = signer.assina_xml(xml, kwargs['NFes'][0]['infNFe']['Id'])
 
+    xml_signed = xml_signed.replace(
+        '\n<!DOCTYPE NFe [\n<!ATTLIST infNFe Id ID #IMPLIED>\n]>\n', '')
+
     url = localizar_url(0,  'RS')
     cabecalho = _build_header(**kwargs)
-
-    return executar_consulta(certificado, url, cabecalho, xml_signed)
+    response, obj = executar_consulta(certificado, url, cabecalho, xml_signed)
+    return {
+        'sent_xml': xml_signed,
+        'received_xml': response,
+        'object': obj
+    }
 
 
 def autorizar_nfe(certificado, **kwargs):  # Assinar
     _generate_nfe_id(**kwargs)
-    _send(certificado, 'NfeAutorizacao', **kwargs)
+    return _send(certificado, 'NfeAutorizacao', **kwargs)
 
 
 def retorno_autorizar_nfe(certificado, **kwargs):
-    _send(certificado, 'NfeRetAutorizacao', **kwargs)
+    return _send(certificado, 'NfeRetAutorizacao', **kwargs)
 
 
 def recepcao_evento_cancelamento(certificado, **kwargs):  # Assinar
-    _send(certificado, 'RecepcaoEventoCancelamento', **kwargs)
+    return _send(certificado, 'RecepcaoEventoCancelamento', **kwargs)
 
 
 def inutilizar_nfe(certificado, **kwargs):  # Assinar
-    _send(certificado, 'NfeInutilizacao', **kwargs)
+    return _send(certificado, 'NfeInutilizacao', **kwargs)
 
 
 def consultar_protocolo_nfe(certificado, **kwargs):
-    _send(certificado, 'NfeConsultaProtocolo', **kwargs)
+    return _send(certificado, 'NfeConsultaProtocolo', **kwargs)
 
 
 def nfe_status_servico(certificado, **kwargs):
-    _send(certificado, 'NfeStatusServico.', **kwargs)
+    return _send(certificado, 'NfeStatusServico.', **kwargs)
 
 
 def consulta_cadastro(certificado, **kwargs):
-    _send(certificado, 'NfeConsultaCadastro.', **kwargs)
+    return _send(certificado, 'NfeConsultaCadastro.', **kwargs)
 
 
 def recepcao_evento_carta_correcao(certificado, **kwargs):  # Assinar
-    _send(certificado, 'RecepcaoEventoCarta.', **kwargs)
+    return _send(certificado, 'RecepcaoEventoCarta.', **kwargs)
 
 
 def recepcao_evento_manifesto(certificado, **kwargs):  # Assinar
-    _send(certificado, 'RecepcaoEventoManifesto', **kwargs)
+    return _send(certificado, 'RecepcaoEventoManifesto', **kwargs)
 
 
 def recepcao_evento_epec(certificado, **kwargs):  # Assinar
-    _send(certificado, 'RecepcaoEventoEPEC', **kwargs)
+    return _send(certificado, 'RecepcaoEventoEPEC', **kwargs)
 
 
 def consulta_nfe_destinada(certificado, **kwargs):
-    _send(certificado, 'NfeConsultaDest', **kwargs)
+    return _send(certificado, 'NfeConsultaDest', **kwargs)
 
 
 def download_nfe(certificado, **kwargs):
-    _send(certificado, 'NfeDownloadNF', **kwargs)
+    return _send(certificado, 'NfeDownloadNF', **kwargs)

pytrustnfe/nfe/assinatura.py

@@ -38,7 +38,6 @@ class Assinatura(object):
         self._checar_certificado()
         self._inicializar_cripto()
         try:
-            xml = '<!DOCTYPE NFe [<!ATTLIST infNFe Id ID #IMPLIED>]>' + xml
             doc_xml = libxml2.parseMemory(
                 xml, len(xml))
 

pytrustnfe/nfe/comunicacao.py

@@ -15,14 +15,26 @@ soap_body_path = './soap:Envelope/soap:Body'
 soap_fault_path = './soap:Envelope/soap:Body/soap:Fault'
 
 
+def _soap_xml(body):
+    xml = '<?xml version="1.0" encoding="utf-8"?>'
+    xml += '<soap12:Envelope xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xmlns:xsd="http://www.w3.org/2001/XMLSchema" xmlns:soap12="http://www.w3.org/2003/05/soap-envelope"><soap12:Header>'
+    xml += '<nfeCabecMsg xmlns="http://www.portalfiscal.inf.br/nfe/wsdl/NfeAutorizacao">'
+    xml += '<cUF>43</cUF><versaoDados>3.10</versaoDados></nfeCabecMsg></soap12:Header><soap12:Body>'
+    xml += '<nfeDadosMsg xmlns="http://www.portalfiscal.inf.br/nfe/wsdl/NfeAutorizacao">'
+    xml += body
+    xml += '</nfeDadosMsg></soap12:Body></soap12:Envelope>'
+    return xml.rstrip('\n')
+
+
 def executar_consulta(certificado, url, cabecalho, xmlEnviar):
     cert, key = extract_cert_and_key_from_pfx(certificado.pfx, certificado.password)
     cert_path, key_path = save_cert_key(cert, key)
     url = 'https://nfe-homologacao.sefazrs.rs.gov.br/ws/NfeAutorizacao/NFeAutorizacao.asmx'
     web_service = 'NfeAutorizacao/nfeAutorizacaoLote'
     client = HttpClient(url, cert_path, key_path)
-    xml_retorno = client.post_soap(xmlEnviar, web_service)
-
+    xmlEnviar = xmlEnviar.replace('<?xml version="1.0"?>', '')
+    xml_enviar = _soap_xml(xmlEnviar)
+    xml_retorno = client.post_soap(xml_enviar, web_service)
     return sanitize_response(xml_retorno)
 
 
@@ -36,16 +48,6 @@ class Comunicacao(object):
         self.cert = cert
         self.key = key
 
-    def _soap_xml(self, body):
-        xml = '<?xml version="1.0" encoding="utf-8"?>'
-        xml += '<soap12:Envelope xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xmlns:xsd="http://www.w3.org/2001/XMLSchema" xmlns:soap12="http://www.w3.org/2003/05/soap-envelope"><soap12:Header>'
-        xml += '<nfeCabecMsg xmlns="http://www.portalfiscal.inf.br/nfe/wsdl/NfeAutorizacao">'
-        xml += '<cUF>43</cUF><versaoDados>3.10</versaoDados></nfeCabecMsg></soap12:Header><soap12:Body>'
-        xml += '<nfeDadosMsg xmlns="http://www.portalfiscal.inf.br/nfe/wsdl/NfeAutorizacao">'
-        xml += body
-        xml += '</nfeDadosMsg></soap12:Body></soap12:Envelope>'
-        return xml.rstrip('\n')
-
     def _preparar_temp_pem(self):
         cert_path = '/tmp/' + uuid4().hex
         key_path = '/tmp/' + uuid4().hex

pytrustnfe/utils.py

@@ -53,7 +53,7 @@ def gerar_chave(obj_chave, prefix=None):
     assert isinstance(obj_chave, ChaveNFe), "Objeto deve ser do tipo ChaveNFe"
     obj_chave.validar()
 
-    chave_parcial = "%s%s%s%s%s%09d%d%08d" % (obj_chave.estado, obj_chave.emissao,
+    chave_parcial = "%s%s%s%s%s%09d%d%s" % (obj_chave.estado, obj_chave.emissao,
                                           obj_chave.cnpj, obj_chave.modelo,
                                           obj_chave.serie.zfill(3), obj_chave.numero,
                                           obj_chave.tipo, obj_chave.codigo)