From 52bd3e7aa320201c1ad428785f9b8dca5a56832b Mon Sep 17 00:00:00 2001 From: Iddo Tal Date: Mon, 18 Dec 2017 23:56:46 +0200 Subject: [PATCH] added kwargs to the __init__ of RedisManager in order to pass more parameters in this case used to optionally set redis client name add kwargs to zmq_manager.py and kombu_manager.py --- socketio/kombu_manager.py | 2 +- socketio/redis_manager.py | 4 +++- socketio/zmq_manager.py | 2 +- 3 files changed, 5 insertions(+), 3 deletions(-) diff --git a/socketio/kombu_manager.py b/socketio/kombu_manager.py index 9906673..c699ca0 100644 --- a/socketio/kombu_manager.py +++ b/socketio/kombu_manager.py @@ -38,7 +38,7 @@ class KombuManager(PubSubManager): # pragma: no cover name = 'kombu' def __init__(self, url='amqp://guest:guest@localhost:5672//', - channel='socketio', write_only=False): + channel='socketio', write_only=False, **kwargs): if kombu is None: raise RuntimeError('Kombu package is not installed ' '(Run "pip install kombu" in your ' diff --git a/socketio/redis_manager.py b/socketio/redis_manager.py index 30f7978..34ce594 100644 --- a/socketio/redis_manager.py +++ b/socketio/redis_manager.py @@ -33,12 +33,14 @@ class RedisManager(PubSubManager): # pragma: no cover name = 'redis' def __init__(self, url='redis://localhost:6379/0', channel='socketio', - write_only=False): + write_only=False, **kwargs): if redis is None: raise RuntimeError('Redis package is not installed ' '(Run "pip install redis" in your ' 'virtualenv).') self.redis = redis.Redis.from_url(url) + if "client_name" in kwargs: + self.redis.client_setname(kwargs["client_name"]) self.pubsub = self.redis.pubsub() super(RedisManager, self).__init__(channel=channel, write_only=write_only) diff --git a/socketio/zmq_manager.py b/socketio/zmq_manager.py index d8995a0..3a63f07 100644 --- a/socketio/zmq_manager.py +++ b/socketio/zmq_manager.py @@ -50,7 +50,7 @@ class ZmqManager(PubSubManager): # pragma: no cover def __init__(self, url='zmq+tcp://localhost:5555+5556', channel='socketio', - write_only=False): + write_only=False, **kwargs): if zmq is None: raise RuntimeError('zmq package is not installed ' '(Run "pip install pyzmq" in your '