Add queues names to yml

huridocs · Sep 23, 2024 · 91f5302 · 91f5302
1 parent 7c43628
commit 91f5302
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/src/QueueProcessor.py b/src/QueueProcessor.py
@@ -68,8 +68,10 @@ def process(message):
 
     service_config = ServiceConfig()
     logger = service_config.get_logger("redis_tasks")
-    queues_names = QUEUES_NAMES.split(" ")
     queue_processor = QueueProcessor(
-        service_config.redis_host, service_config.redis_port, queues_names, logger
+        service_config.redis_host,
+        service_config.redis_port,
+        service_config.queues_names,
+        logger,
     )
     queue_processor.start(process)
diff --git a/src/ServiceConfig.py b/src/ServiceConfig.py
@@ -16,8 +16,6 @@
 PDF_PROCESSED_PATH = f"{DATA_PATH}/processed_pdfs"
 PDF_FAILED = f"{DATA_PATH}/failed_pdfs"
 
-QUEUES_NAMES = os.environ.get("QUEUES_NAMES", "ocr")
-
 
 class ServiceConfig:
     def __init__(self):
@@ -45,6 +43,7 @@ def __init__(self):
             "service_port", default_service_port
         )
         self.service_url = f"http://{self.service_host}:{self.service_port}"
+        self.queues_names = self.get_parameter_from_yml("queues_names", "ocr").split()
 
     def get_parameter_from_yml(self, parameter_name: str, default: any):
         if parameter_name in self.config_from_yml: