From a20994cb53d84e3e7d4e7f9c2b881bd8216ae7c4 Mon Sep 17 00:00:00 2001
From: Yoann Schneider <yschneider@teklia.com>
Date: Wed, 7 Sep 2022 13:18:34 +0000
Subject: [PATCH] Allow defining corpus_id via environment in dev mode

---
 arkindex_worker/worker/base.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/arkindex_worker/worker/base.py b/arkindex_worker/worker/base.py
index f892558e..fe27e0d3 100644
--- a/arkindex_worker/worker/base.py
+++ b/arkindex_worker/worker/base.py
@@ -119,6 +119,8 @@ class BaseWorker(object):
         logger.info(f"Worker will use {self.work_dir} as working directory")
 
         self.process_information = None
+        # corpus_id will be updated in configure() using the worker_run's corpus
+        # or in configure_for_developers() from the environment
         self.corpus_id = None
         self.user_configuration = {}
         self.support_cache = support_cache
@@ -170,6 +172,9 @@ class BaseWorker(object):
             required_secrets = []
             logger.warning("Running without any extra configuration")
 
+        # Define corpus_id from environment
+        self.corpus_id = os.environ.get("ARKINDEX_CORPUS_ID")
+
         # Load all required secrets
         self.secrets = {name: self.load_secret(name) for name in required_secrets}
 
-- 
GitLab