lint cleanup

This commit is contained in:
alwaysjolley 2019-05-07 10:05:01 -04:00
parent fb3f0bd72a
commit b0c8901b0a
1 changed files with 5 additions and 8 deletions

View File

@ -81,7 +81,6 @@ class VaultSourcePlugin(SourcePlugin):
}, },
] ]
def get_certificates(self, options, **kwargs): def get_certificates(self, options, **kwargs):
"""Pull certificates from objects in Hashicorp Vault""" """Pull certificates from objects in Hashicorp Vault"""
data = [] data = []
@ -106,22 +105,21 @@ class VaultSourcePlugin(SourcePlugin):
secret = get_secret(client, mount, path) secret = get_secret(client, mount, path)
for cname in secret['data']: for cname in secret['data']:
#current_app.logger.info("Certificate Data: {0}".format(secret['data'][cname]))
if 'crt' in secret['data'][cname]: if 'crt' in secret['data'][cname]:
cert = secret['data'][cname]['crt'].split(cert_delimiter+'\n') cert = secret['data'][cname]['crt'].split(cert_delimiter + '\n')
elif 'pem' in secret['data'][cname]: elif 'pem' in secret['data'][cname]:
cert = secret['data'][cname]['pem'].split(cert_delimiter+'\n') cert = secret['data'][cname]['pem'].split(cert_delimiter + '\n')
else: else:
for key in secret['data'][cname]: for key in secret['data'][cname]:
if secret['data'][cname][key].startswith(cert_filter): if secret['data'][cname][key].startswith(cert_filter):
cert = secret['data'][cname][key].split(cert_delimiter+'\n') cert = secret['data'][cname][key].split(cert_delimiter + '\n')
break break
body = cert[0]+cert_delimiter body = cert[0] + cert_delimiter
if 'chain' in secret['data'][cname]: if 'chain' in secret['data'][cname]:
chain = secret['data'][cname]['chain'] chain = secret['data'][cname]['chain']
elif len(cert) > 1: elif len(cert) > 1:
if cert[1].startswith(cert_filter): if cert[1].startswith(cert_filter):
chain = cert[1]+cert_delimiter chain = cert[1] + cert_delimiter
else: else:
chain = None chain = None
else: else:
@ -214,7 +212,6 @@ class VaultDestinationPlugin(DestinationPlugin):
} }
] ]
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
super(VaultDestinationPlugin, self).__init__(*args, **kwargs) super(VaultDestinationPlugin, self).__init__(*args, **kwargs)