From 024ca81ffe67d75690f2684bae595cea344e48e5 Mon Sep 17 00:00:00 2001 From: David Arias Date: Wed, 6 May 2020 13:43:20 -0300 Subject: [PATCH] Use correctly mkstemp in order to close file descriptors after using it. --- pytrustnfe/certificado.py | 15 +++++++-------- 1 file changed, 7 insertions(+), 8 deletions(-) diff --git a/pytrustnfe/certificado.py b/pytrustnfe/certificado.py index d6aab08..926d9c3 100644 --- a/pytrustnfe/certificado.py +++ b/pytrustnfe/certificado.py @@ -4,6 +4,7 @@ import tempfile from OpenSSL import crypto +import os class Certificado(object): @@ -29,15 +30,13 @@ def extract_cert_and_key_from_pfx(pfx, password): def save_cert_key(cert, key): - cert_temp = tempfile.mkstemp()[1] - key_temp = tempfile.mkstemp()[1] + fd_cert, cert_temp = tempfile.mkstemp() + fd_key, key_temp = tempfile.mkstemp() - arq_temp = open(cert_temp, "w") - arq_temp.write(cert) - arq_temp.close() + os.write(fd_cert, cert.encode()) + os.close(fd_cert) - arq_temp = open(key_temp, "w") - arq_temp.write(key) - arq_temp.close() + os.write(fd_key, key.encode()) + os.close(fd_key) return cert_temp, key_temp