added environmental variable for redis endpoint

docker-compose.yaml

@@ -6,12 +6,16 @@ services:
  - 5000:5000
  volumes:
  - ../lbp-print-cache:/usr/src/app/cache
+ environment:
+ - REDIS_DOCKER=True
  queue-worker:
  build: .
  command: python3 worker.py
  depends_on:
  - redis
  volumes:
  - ../lbp-print-cache:/usr/src/app/cache
+ environment:
+ - REDIS_DOCKER=True
  redis:
  image: "redis:alpine"

processor.py

@@ -1,5 +1,5 @@
 import logging
-
+import os
 import urllib
 
 from redis import Redis
@@ -13,7 +13,9 @@
 
 logger = logging.getLogger()
 lbp_config.cache_dir = "cache"
-redis_connection = Redis(host="localhost")
+redis_endpoint = "redis" if os.environ.get("REDIS_DOCKER") == "True" else "localhost"
+logger.warning(f"Logging redis endpoint: {redis_endpoint}")
+redis_connection = Redis(host=redis_endpoint)
 q = Queue(connection=redis_connection)
 
 

worker.py

@@ -1,13 +1,14 @@
 #!/usr/bin/env python
 import sys
-
+import os
 from redis import Redis
 from rq import Connection, Worker
 
 
 def start_worker(queues: list = ["default"]):
+ redis_endpoint = "redis" if os.environ.get("REDIS_DOCKER") == "True" else "localhost"
  with Connection():
- w = Worker(queues, connection=Redis(host="localhost"))
+ w = Worker(queues, connection=Redis(host=redis_endpoint))
  w.work()