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)