Merge branch 'master' of https://github.com/leotada/PyNFe

pynfe/entidades/certificado.py

@@ -33,7 +33,7 @@ class CertificadoA1(Certificado):
     def separar_arquivo(self, senha, caminho=False):
         """Separa o arquivo de certificado em dois: de chave e de certificado,
         e retorna a string. Se caminho for True grava na pasta temporaria e retorna
-        o caminho dos arquivos, apos o uso devem ser excluidos com o metodo excluir."""
+        o caminho dos arquivos, senao retorna o objeto. Apos o uso devem ser excluidos com o metodo excluir."""
 
         # Carrega o arquivo .pfx, erro pode ocorrer se a senha estiver errada ou formato invalido.
         pkcs12 = crypto.load_pkcs12(open(self.caminho_arquivo, "rb").read(), senha)
@@ -69,4 +69,3 @@ class CertificadoA1(Certificado):
             self.arquivos_temp.clear()
         except:
             pass
-

pynfe/processamento/assinatura.py

@@ -1,7 +1,11 @@
 # -*- coding: utf-8 -*-
 
 from pynfe.utils import etree, remover_acentos
+from pynfe.utils.flags import NAMESPACE_SIG
 import subprocess
+import signxml
+from signxml import XMLSigner
+from pynfe.entidades import CertificadoA1
 
 
 class Assinatura(object):
@@ -323,3 +327,38 @@ class AssinaturaA1(Assinatura):
                 return xml
         except Exception as e:
             raise e
+
+
+class AssinaturaA1SignXML(Assinatura):
+
+    def __init__(self, certificado, senha):
+        self.key, self.cert = CertificadoA1(certificado).separar_arquivo(senha)
+
+    def assinar(self, xml, retorna_string=False):
+        # busca tag que tem id(reference_uri), logo nao importa se tem namespace
+        reference = xml.find(".//*[@Id]").attrib['Id']
+
+        # retira acentos
+        xml_str = remover_acentos(etree.tostring(xml, encoding="unicode", pretty_print=False))
+        xml = etree.fromstring(xml_str)
+
+        signer = XMLSigner(
+            method=signxml.methods.enveloped, signature_algorithm="rsa-sha1",
+            digest_algorithm='sha1',
+            c14n_algorithm='http://www.w3.org/TR/2001/REC-xml-c14n-20010315')
+
+        ns = {None: signer.namespaces['ds']}
+        signer.namespaces = ns
+
+        ref_uri = ('#%s' % reference) if reference else None
+        signed_root = signer.sign(
+            xml, key=self.key, cert=self.cert, reference_uri=ref_uri)
+
+        ns = {'ns': NAMESPACE_SIG}
+        # coloca o certificado na tag X509Data/X509Certificate
+        tagX509Data = signed_root.find('.//ns:X509Data', namespaces=ns)
+        etree.SubElement(tagX509Data, 'X509Certificate').text = self.cert
+        if retorna_string:
+            return etree.tostring(signed_root, encoding="unicode", pretty_print=False)
+        else:
+            return signed_root

pynfe/processamento/comunicacao.py

@@ -6,7 +6,7 @@ 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, NAMESPACE_BETHA
 from pynfe.utils.webservices import NFCE, NFE, NFSE
-from .assinatura import AssinaturaA1
+from .assinatura import AssinaturaA1, AssinaturaA1SignXML
 from pynfe.entidades.certificado import CertificadoA1
 
 
@@ -260,7 +260,7 @@ class ComunicacaoSefaz(Comunicacao):
         etree.SubElement(inf_inut, 'xJust').text = justificativa
 
         # assinatura
-        a1 = AssinaturaA1(self.certificado, self.certificado_senha)
+        a1 = AssinaturaA1SignXML(self.certificado, self.certificado_senha)
         xml = a1.assinar(raiz)
 
         # Monta XML para envio da requisição

pynfe/utils/__init__.py

@@ -9,9 +9,6 @@ try:
 except ImportError:
     raise Exception('Falhou ao importar lxml/ElementTree')
 
-try:
-    from StringIO import StringIO
-except ImportError:
 from io import StringIO
 
 try: