1 files changed, 1 insertions, 13 deletions
diff --git a/python/process_queue.py b/python/process_queue.py
index 107f0b80..43d7d9a7 100644
--- a/python/process_queue.py
+++ b/python/process_queue.py
@@ -1,5 +1,4 @@
 import datetime
-import redis
 import socket
 import pandas as pd
 import csv
@@ -16,6 +15,7 @@ import task_server.config as config
 import os
 from sqlalchemy import create_engine
 from bbg_helpers import init_bbg_session, retrieve_data, BBG_IP
+from common import get_redis_queue
 import re
 from analytics.utils import previous_twentieth
 from db import dbconn
@@ -81,18 +81,6 @@ def decode_dict(d):
     return {k.decode() if isinstance(k, bytes) else k: \
             v.decode() if isinstance(v, bytes) else v for k, v in d.items()}
 
-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
-
 def get_effective_date(d):
     return previous_twentieth(d + datetime.timedelta(days=1))