1 files changed, 2 insertions, 10 deletions
diff --git a/python/task_runner.py b/python/task_runner.py
index 1e5738b7..67f68edc 100644
--- a/python/task_runner.py
+++ b/python/task_runner.py
@@ -1,23 +1,15 @@
 import logging
-import os
 import redis
-import socket
 import tasks
 from systemd.daemon import notify
 from systemd import journal
 from json import loads, dumps
 from db import dbconn
-from termcolor import colored
+from process_queue import get_redis_queue
 
 def run():
-    hostname = socket.gethostname()
     ET = dbconn('etdb')
-    if hostname == 'ziggy':
-        q = redis.Redis(unix_socket_path='/run/redis/redis.sock')
-        os.environ['OMP_NUM_THREADS'] = '8'
-    else:
-        q = redis.Redis(host='ziggy')
-        os.environ['OMP_NUM_THREADS'] = '4'
+    q = get_redis_queue()
     notify("READY=1")
     while True:
         f, args = loads(q.blpop("tasks")[1].decode('utf-8'))