Browse Source

certificado separa arquivos de duas formas

pull/1/head
leotada 11 years ago
parent
commit
31f8402e90
  1. 16
      pynfe/entidades/certificado.py

16
pynfe/entidades/certificado.py

@ -2,6 +2,7 @@
from .base import Entidade from .base import Entidade
from OpenSSL import crypto from OpenSSL import crypto
import tempfile
class Certificado(Entidade): class Certificado(Entidade):
@ -27,13 +28,24 @@ class CertificadoA1(Certificado):
def __init__(self, caminho_arquivo=None): def __init__(self, caminho_arquivo=None):
self.caminho_arquivo = caminho_arquivo self.caminho_arquivo = caminho_arquivo
def separar_arquivo(self, senha, caminho_chave=None, caminho_cert=None):
def separar_arquivo(self, senha, caminho=False):
"""Separa o arquivo de certificado em dois: de chave e de certificado, """Separa o arquivo de certificado em dois: de chave e de certificado,
e retorna a string."""
e retorna a string. Se caminho for True grava e retorna objeto gravado
na pasta temporaria, que tem o caminho(name), apos o uso devem ser fechados com close."""
# Carrega o arquivo .pfx, erro pode ocorrer se a senha estiver errada ou formato invalido. # 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) pkcs12 = crypto.load_pkcs12(open(self.caminho_arquivo, "rb").read(), senha)
if caminho:
cert = crypto.dump_certificate(crypto.FILETYPE_PEM, pkcs12.get_certificate())
chave = crypto.dump_privatekey(crypto.FILETYPE_PEM, pkcs12.get_privatekey())
# cria arquivos temporarios
arqcert = tempfile.NamedTemporaryFile()
arqcert.write(cert)
arqchave = tempfile.NamedTemporaryFile()
arqchave.write(chave)
return arqchave, arqcert
else:
# Certificado # Certificado
cert = crypto.dump_certificate(crypto.FILETYPE_PEM, pkcs12.get_certificate()).decode('utf-8') cert = crypto.dump_certificate(crypto.FILETYPE_PEM, pkcs12.get_certificate()).decode('utf-8')
cert = cert.replace('\n', '') cert = cert.replace('\n', '')

Loading…
Cancel
Save