improved redis endpoint env variable handling

programminghistorian · Feb 10, 2020 · d4e59c6 · d4e59c6
1 parent a990ff9
commit d4e59c6
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/docker-compose.yaml b/docker-compose.yaml
@@ -7,7 +7,7 @@ services:
  volumes:
  - ../lbp-print-cache:/usr/src/app/cache
  environment:
- - REDIS_DOCKER=True
+ - REDIS_ENDPOINT=redis
  queue-worker:
  build: .
  command: python3 worker.py

diff --git a/processor.py b/processor.py
@@ -13,7 +13,7 @@
 
 logger = logging.getLogger()
 lbp_config.cache_dir = "cache"
-redis_endpoint = "redis" if os.environ.get("REDIS_DOCKER") == "True" else "localhost"
+redis_endpoint = os.getenv("REDIS_ENDPOINT", "localhost")
 logger.warning(f"Logging redis endpoint: {redis_endpoint}")
 redis_connection = Redis(host=redis_endpoint)
 q = Queue(connection=redis_connection)

diff --git a/worker.py b/worker.py
@@ -6,7 +6,7 @@
 
 
 def start_worker(queues: list = ["default"]):
- redis_endpoint = "redis" if os.environ.get("REDIS_DOCKER") == "True" else "localhost"
+ redis_endpoint = os.getenv("REDIS_ENDPOINT", "localhost")
  with Connection():
  w = Worker(queues, connection=Redis(host=redis_endpoint))
  w.work()