From 91f5302ce9ca8be506633f209ed4f7f5d35e8247 Mon Sep 17 00:00:00 2001 From: Gabo Date: Mon, 23 Sep 2024 12:46:59 +0200 Subject: [PATCH] Add queues names to yml --- src/QueueProcessor.py | 6 ++++-- src/ServiceConfig.py | 3 +-- 2 files changed, 5 insertions(+), 4 deletions(-) diff --git a/src/QueueProcessor.py b/src/QueueProcessor.py index 3f2c7cf..bb94a86 100644 --- 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 index 57daccc..9b73084 100644 --- 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: