Black lint all the things
This commit is contained in:
@ -1,5 +1,4 @@
|
||||
try:
|
||||
VERSION = __import__('pkg_resources') \
|
||||
.get_distribution(__name__).version
|
||||
VERSION = __import__("pkg_resources").get_distribution(__name__).version
|
||||
except Exception as e:
|
||||
VERSION = 'unknown'
|
||||
VERSION = "unknown"
|
||||
|
@ -43,38 +43,30 @@ def create_csr(cert, chain, csr_tmp, key):
|
||||
assert isinstance(key, str)
|
||||
|
||||
with mktempfile() as key_tmp:
|
||||
with open(key_tmp, 'w') as f:
|
||||
with open(key_tmp, "w") as f:
|
||||
f.write(key)
|
||||
|
||||
with mktempfile() as cert_tmp:
|
||||
with open(cert_tmp, 'w') as f:
|
||||
with open(cert_tmp, "w") as f:
|
||||
if chain:
|
||||
f.writelines([cert.strip() + "\n", chain.strip() + "\n"])
|
||||
else:
|
||||
f.writelines([cert.strip() + "\n"])
|
||||
|
||||
output = subprocess.check_output([
|
||||
"openssl",
|
||||
"x509",
|
||||
"-x509toreq",
|
||||
"-in", cert_tmp,
|
||||
"-signkey", key_tmp,
|
||||
])
|
||||
subprocess.run([
|
||||
"openssl",
|
||||
"req",
|
||||
"-out", csr_tmp
|
||||
], input=output)
|
||||
output = subprocess.check_output(
|
||||
["openssl", "x509", "-x509toreq", "-in", cert_tmp, "-signkey", key_tmp]
|
||||
)
|
||||
subprocess.run(["openssl", "req", "-out", csr_tmp], input=output)
|
||||
|
||||
|
||||
class CSRExportPlugin(ExportPlugin):
|
||||
title = 'CSR'
|
||||
slug = 'openssl-csr'
|
||||
description = 'Exports a CSR'
|
||||
title = "CSR"
|
||||
slug = "openssl-csr"
|
||||
description = "Exports a CSR"
|
||||
version = csr.VERSION
|
||||
|
||||
author = 'jchuong'
|
||||
author_url = 'https://github.com/jchuong'
|
||||
author = "jchuong"
|
||||
author_url = "https://github.com/jchuong"
|
||||
|
||||
def export(self, body, chain, key, options, **kwargs):
|
||||
"""
|
||||
@ -93,7 +85,7 @@ class CSRExportPlugin(ExportPlugin):
|
||||
create_csr(body, chain, output_tmp, key)
|
||||
extension = "csr"
|
||||
|
||||
with open(output_tmp, 'rb') as f:
|
||||
with open(output_tmp, "rb") as f:
|
||||
raw = f.read()
|
||||
# passphrase is None
|
||||
return extension, None, raw
|
||||
|
@ -4,7 +4,8 @@ from lemur.tests.vectors import INTERNAL_PRIVATE_KEY_A_STR, INTERNAL_CERTIFICATE
|
||||
|
||||
def test_export_certificate_to_csr(app):
|
||||
from lemur.plugins.base import plugins
|
||||
p = plugins.get('openssl-csr')
|
||||
|
||||
p = plugins.get("openssl-csr")
|
||||
options = []
|
||||
with pytest.raises(Exception):
|
||||
p.export(INTERNAL_CERTIFICATE_A_STR, "", "", options)
|
||||
|
Reference in New Issue
Block a user