From 2c1e2ed3737ad8967294a470c9237e8388e9ded7 Mon Sep 17 00:00:00 2001 From: Joey Payne Date: Mon, 1 Feb 2016 16:29:57 -0700 Subject: [PATCH] Raise error on connection lost. --- socketIO_client/__init__.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/socketIO_client/__init__.py b/socketIO_client/__init__.py index 1946323..2fe82fe 100644 --- a/socketIO_client/__init__.py +++ b/socketIO_client/__init__.py @@ -237,7 +237,7 @@ class SocketIO(object): callback, args = find_callback(args, kw) self._transport.emit(path, event, args, callback) - def wait(self, seconds=None, for_callbacks=False): + def wait(self, seconds=None, for_callbacks=False, raise_error=False): """Wait in a loop and process events as defined in the namespaces. - Omit seconds, i.e. call wait() without arguments, to wait forever. @@ -254,6 +254,8 @@ class SocketIO(object): pass next(self._heartbeat_pacemaker) except ConnectionError as e: + if raise_error: + raise try: warning = Exception('[connection error] %s' % e) warning_screen.throw(warning)