From c9268f9aa9618fb9fd9c161a7b78272250bc774c Mon Sep 17 00:00:00 2001 From: oxid Date: Wed, 10 Feb 2021 20:11:15 +0300 Subject: [PATCH] Catch TimeoutError --- socketio/redis_manager.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/socketio/redis_manager.py b/socketio/redis_manager.py index cef7624..9ce0b44 100644 --- a/socketio/redis_manager.py +++ b/socketio/redis_manager.py @@ -78,7 +78,7 @@ class RedisManager(PubSubManager): # pragma: no cover if not retry: self._redis_connect() return self.redis.publish(self.channel, pickle.dumps(data)) - except redis.exceptions.ConnectionError: + except (redis.exceptions.ConnectionError, redis.exceptions.TimeoutError) as e: if retry: logger.error('Cannot publish to redis... retrying') retry = False @@ -96,9 +96,9 @@ class RedisManager(PubSubManager): # pragma: no cover self.pubsub.subscribe(self.channel) for message in self.pubsub.listen(): yield message - except redis.exceptions.ConnectionError: - logger.error('Cannot receive from redis... ' - 'retrying in {} secs'.format(retry_sleep)) + except (redis.exceptions.ConnectionError, redis.exceptions.TimeoutError) as e: + logger.error('Cannot receive from redis ({})... ' + 'retrying in {} secs'.format(e, retry_sleep)) connect = True time.sleep(retry_sleep) retry_sleep *= 2