diff options
Diffstat (limited to 'kazoo/protocol/connection.py')
-rw-r--r-- | kazoo/protocol/connection.py | 28 |
1 files changed, 5 insertions, 23 deletions
diff --git a/kazoo/protocol/connection.py b/kazoo/protocol/connection.py index d7d84d1..d672497 100644 --- a/kazoo/protocol/connection.py +++ b/kazoo/protocol/connection.py @@ -10,8 +10,6 @@ import ssl import sys import time -import six - from kazoo.exceptions import ( AuthFailedError, ConnectionDropped, @@ -808,23 +806,11 @@ class ConnectionHandler(object): try: response = sasl_cli.process(challenge=challenge) except puresasl.SASLError as err: - six.reraise( - SASLException, - SASLException("library error: %s" % err), - sys.exc_info()[2], - ) + raise SASLException("library error: %s" % err) from err except puresasl.SASLProtocolException as err: - six.reraise( - AuthFailedError, - AuthFailedError("protocol error: %s" % err), - sys.exc_info()[2], - ) + raise AuthFailedError("protocol error: %s" % err) from err except Exception as err: - six.reraise( - AuthFailedError, - AuthFailedError("Unknown error: %s" % err), - sys.exc_info()[2], - ) + raise AuthFailedError("Unknown error: %s" % err) from err if sasl_cli.complete and not response: break @@ -838,13 +824,9 @@ class ConnectionHandler(object): try: header, buffer, offset = self._read_header(timeout) - except ConnectionDropped: + except ConnectionDropped as ex: # Zookeeper simply drops connections with failed authentication - six.reraise( - AuthFailedError, - AuthFailedError("Connection dropped in SASL"), - sys.exc_info()[2], - ) + raise AuthFailedError("Connection dropped in SASL") from ex if header.xid != xid: raise RuntimeError( |