Allow for additional REDIS config such as password and port

data/userevent.py

@@ -7,14 +7,14 @@ class UserEventBuilder(object):
   Defines a helper class for constructing UserEvent and UserEventListener
   instances.
   """
-  def __init__(self, redis_host):
-    self._redis_host = redis_host
+  def __init__(self, redis_config):
+    self._redis_config = redis_config
 
   def get_event(self, username):
-    return UserEvent(self._redis_host, username)
+    return UserEvent(self._redis_config, username)
 
   def get_listener(self, username, events):
-    return UserEventListener(self._redis_host, username, events)
+    return UserEventListener(self._redis_config, username, events)
 
 
 class UserEventsBuilderModule(object):
@@ -26,8 +26,14 @@ class UserEventsBuilderModule(object):
       self.state = None
 
   def init_app(self, app):
-    redis_hostname = app.config.get('USER_EVENTS_REDIS_HOSTNAME')
-    user_events = UserEventBuilder(redis_hostname)
+    redis_config = app.config.get('USER_EVENTS_REDIS')
+    if not redis_config:
+      # This is the old key name.
+      redis_config = {
+        'host': app.config.get('USER_EVENTS_REDIS_HOSTNAME')
+      }
+
+    user_events = UserEventBuilder(redis_config)
 
     # register extension with app
     app.extensions = getattr(app, 'extensions', {})
@@ -43,8 +49,8 @@ class UserEvent(object):
   Defines a helper class for publishing to realtime user events
   as backed by Redis.
   """
-  def __init__(self, redis_host, username):
-    self._redis = redis.StrictRedis(host=redis_host, socket_connect_timeout=5)
+  def __init__(self, redis_config, username):
+    self._redis = redis.StrictRedis(socket_connect_timeout=5, **redis_config)
     self._username = username
 
   @staticmethod
@@ -74,10 +80,10 @@ class UserEventListener(object):
   Defines a helper class for subscribing to realtime user events as
   backed by Redis.
   """
-  def __init__(self, redis_host, username, events=set([])):
+  def __init__(self, redis_config, username, events=set([])):
     channels = [self._user_event_key(username, e) for e in events]
 
-    self._redis = redis.StrictRedis(host=redis_host, socket_connect_timeout=5)
+    self._redis = redis.StrictRedis(socket_connect_timeout=5, **redis_config)
     self._pubsub = self._redis.pubsub()
     self._pubsub.subscribe(channels)