Remove get_exception from crypto namespace

Fix removes get_exception in favor of native Python exception
handling. Also, added to_native to manage exception message.
pull/25956/head
Yanis Guenane 7 years ago committed by Toshio Kuratomi
parent db5c0b3b2b
commit ed15a0e9a7

@ -167,7 +167,7 @@ else:
pyopenssl_found = True
from ansible.module_utils.basic import AnsibleModule
from ansible.module_utils.pycompat24 import get_exception
from ansible.module_utils._text import to_native
class CertificateSigningRequestError(Exception):
@ -230,9 +230,8 @@ class CertificateSigningRequest(object):
csr_file = open(self.path, 'w')
csr_file.write(crypto.dump_certificate_request(crypto.FILETYPE_PEM, self.request))
csr_file.close()
except (IOError, OSError):
e = get_exception()
raise CertificateSigningRequestError(e)
except (IOError, OSError) as exc:
raise CertificateSigningRequestError(exc)
else:
self.changed = False
@ -245,10 +244,9 @@ class CertificateSigningRequest(object):
try:
os.remove(self.path)
except OSError:
e = get_exception()
if e.errno != errno.ENOENT:
raise CertificateSigningRequestError(e)
except OSError as exc:
if exc.errno != errno.ENOENT:
raise CertificateSigningRequestError(exc)
else:
self.changed = False
@ -305,9 +303,8 @@ def main():
try:
csr.generate(module)
except CertificateSigningRequestError:
e = get_exception()
module.fail_json(msg=str(e))
except CertificateSigningRequestError as exc:
module.fail_json(msg=to_native(exc))
else:
@ -318,9 +315,8 @@ def main():
try:
csr.remove()
except CertificateSigningRequestError:
e = get_exception()
module.fail_json(msg=str(e))
except CertificateSigningRequestError as exc:
module.fail_json(msg=to_native(exc))
result = csr.dump()

@ -114,6 +114,7 @@ else:
import os
from ansible.module_utils._text import to_native
class PrivateKeyError(Exception):
@ -147,8 +148,8 @@ class PrivateKey(object):
try:
self.privatekey.generate_key(crypto_type, self.size)
except (TypeError, ValueError):
raise PrivateKeyError(get_exception())
except (TypeError, ValueError) as exc:
raise PrivateKeyError(exc)
try:
privatekey_file = os.open(self.path,
@ -157,9 +158,9 @@ class PrivateKey(object):
os.write(privatekey_file, crypto.dump_privatekey(crypto.FILETYPE_PEM, self.privatekey))
os.close(privatekey_file)
except IOError:
except IOError as exc:
self.remove()
raise PrivateKeyError(get_exception())
raise PrivateKeyError(exc)
else:
self.changed = False
@ -173,10 +174,9 @@ class PrivateKey(object):
try:
os.remove(self.path)
except OSError:
e = get_exception()
if e.errno != errno.ENOENT:
raise PrivateKeyError(e)
except OSError as exc:
if exc.errno != errno.ENOENT:
raise PrivateKeyError(exc)
else:
self.changed = False
@ -230,9 +230,8 @@ def main():
try:
private_key.generate(module)
except PrivateKeyError:
e = get_exception()
module.fail_json(msg=str(e))
except PrivateKeyError as exc:
module.fail_json(msg=to_native(exc))
else:
if module.check_mode:
@ -242,9 +241,8 @@ def main():
try:
private_key.remove()
except PrivateKeyError:
e = get_exception()
module.fail_json(msg=str(e))
except PrivateKeyError as exc:
module.fail_json(msg=to_native(exc))
result = private_key.dump()

@ -99,6 +99,7 @@ else:
import os
from ansible.module_utils._text import to_native
class PublicKeyError(Exception):
@ -127,10 +128,9 @@ class PublicKey(object):
publickey_file = open(self.path, 'w')
publickey_file.write(crypto.dump_publickey(crypto.FILETYPE_PEM, privatekey))
publickey_file.close()
except (IOError, OSError):
e = get_exception()
raise PublicKeyError(e)
except AttributeError:
except (IOError, OSError) as exc:
raise PublicKeyError(exc)
except AttributeError as exc:
self.remove()
raise PublicKeyError('You need to have PyOpenSSL>=16.0.0 to generate public keys')
else:
@ -145,10 +145,9 @@ class PublicKey(object):
try:
os.remove(self.path)
except OSError:
e = get_exception()
if e.errno != errno.ENOENT:
raise PublicKeyError(e)
except OSError as exc:
if exc.errno != errno.ENOENT:
raise PublicKeyError(exc)
else:
self.changed = False
@ -205,9 +204,8 @@ def main():
try:
public_key.generate(module)
except PublicKeyError:
e = get_exception()
module.fail_json(msg=str(e))
except PublicKeyError as exc:
module.fail_json(msg=to_native(exc))
else:
if module.check_mode:
@ -217,9 +215,8 @@ def main():
try:
public_key.remove()
except PublicKeyError:
e = get_exception()
module.fail_json(msg=str(e))
except PublicKeyError as exc:
module.fail_json(msg=to_native(exc))
result = public_key.dump()

Loading…
Cancel
Save