1 files changed, 15 insertions, 0 deletions
diff --git a/python/common.py b/python/common.py
index ce9c2fe1..05deab26 100644
--- a/python/common.py
+++ b/python/common.py
@@ -1,3 +1,6 @@
+import redis
+import logging
+
 def sanitize_float(intex_float):
     try:
         intex_float = intex_float.replace(",", "")
@@ -9,3 +12,15 @@ def sanitize_float(intex_float):
             return float(intex_float)
     except (AttributeError, ValueError):
         return intex_float
+
+def get_redis_queue():
+    q = redis.Redis(unix_socket_path='/run/redis/redis.sock')
+    try:
+        q.ping()
+    except redis.ConnectionError:
+        try:
+            q = redis.Redis(os.environ['REDIS_HOST'])
+        except KeyError:
+            logging.error("Please set redis host in REDIS_HOST")
+            sys.exit(1)
+    return q