You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
58 lines
2.1 KiB
58 lines
2.1 KiB
# -*- coding: utf-8 -*-
|
|
import os
|
|
|
|
from .base import Entidade
|
|
|
|
from OpenSSL import crypto
|
|
|
|
class Certificado(Entidade):
|
|
u"""Classe abstrata responsavel por definir o modelo padrao para as demais
|
|
classes de certificados digitais.
|
|
|
|
Caso va implementar um novo formato de certificado, crie uma classe que
|
|
herde desta."""
|
|
|
|
def __new__(cls, *args, **kwargs):
|
|
if cls == Certificado:
|
|
raise Exception('Esta classe nao pode ser instanciada diretamente!')
|
|
else:
|
|
return super(Certificado, cls).__new__(cls)
|
|
|
|
class CertificadoA1(Certificado):
|
|
u"""Implementa a entidade do certificado eCNPJ A1, suportado pelo OpenSSL,
|
|
e amplamente utilizado."""
|
|
|
|
caminho_arquivo = None
|
|
conteudo_x509 = None
|
|
pasta_temporaria = '/tmp/'
|
|
arquivo_chave = 'key.pem'
|
|
arquivo_cert = 'cert.pem'
|
|
|
|
def __init__(self, caminho_arquivo=None, conteudo_x509=None):
|
|
self.caminho_arquivo = caminho_arquivo or self.caminho_arquivo
|
|
self.conteudo_x509 = conteudo_x509 or self.conteudo_x509
|
|
|
|
def separar_arquivo(self, senha, caminho_chave=None, caminho_cert=None):
|
|
u"""Separa o arquivo de certificado em dois: de chave e de certificado,
|
|
em arquivos temporários separados"""
|
|
|
|
caminho_chave = caminho_chave or os.path.join(self.pasta_temporaria, self.arquivo_chave)
|
|
caminho_cert = caminho_cert or os.path.join(self.pasta_temporaria, self.arquivo_cert)
|
|
|
|
# Lendo o arquivo pfx no formato pkcs12 como binario
|
|
pkcs12 = crypto.load_pkcs12(open(self.caminho_arquivo, 'rb').read(), senha)
|
|
|
|
# Retorna a string decodificado da chave privada
|
|
key_str = crypto.dump_privatekey(crypto.FILETYPE_PEM, pkcs12.get_privatekey())
|
|
|
|
# Retorna a string decodificado do certificado
|
|
cert_str = crypto.dump_certificate(crypto.FILETYPE_PEM, pkcs12.get_certificate())
|
|
|
|
# Gravando a string no dicso
|
|
open(caminho_cert, 'wb').write(cert_str)
|
|
|
|
# Gravando a string no dicso
|
|
open(caminho_chave, 'wb').write(key_str)
|
|
|
|
return caminho_chave, caminho_cert
|
|
|