Depend only on RetrieveWorkerRun to get all the informations needed

Closes #115 (closed)

arkindex_worker/worker/base.py

@@ -111,6 +111,11 @@ class BaseWorker(object):
             logger.warning(
                 "Missing WORKER_VERSION_ID environment variable, worker is in read-only mode"
             )
+        self.worker_run_id = os.environ.get("WORKER_RUN_ID")
+        if not self.worker_run_id:
+            logger.warning(
+                "Missing WORKER_RUN_ID environment variable, worker is in read-only mode"
+            )
 
         logger.info(f"Worker will use {self.work_dir} as working directory")
 
@@ -130,7 +135,11 @@ class BaseWorker(object):
            or when no worker version ID is provided.
         :rtype: bool
         """
-        return self.args.dev or self.worker_version_id is None
+        return (
+            self.args.dev
+            or self.worker_version_id is None
+            or self.worker_run_id is None
+        )
 
     def configure_for_developers(self):
         assert self.is_read_only
@@ -170,18 +179,14 @@ class BaseWorker(object):
         self.api_client = ArkindexClient(**options_from_env())
         logger.debug(f"Setup Arkindex API client on {self.api_client.document.url}")
 
+        # Load worker run information
+        worker_run = self.request("RetrieveWorkerRun", id=self.worker_run_id)
+
         # Load process information
-        assert os.environ.get(
-            "ARKINDEX_PROCESS_ID"
-        ), "ARKINDEX_PROCESS_ID environment variable is not defined"
-        self.process_information = self.request(
-            "RetrieveDataImport", id=os.environ["ARKINDEX_PROCESS_ID"]
-        )
+        self.process_information = worker_run["process"]
 
         # Retrieve initial configuration from API
-        worker_version = self.request(
-            "RetrieveWorkerVersion", id=self.worker_version_id
-        )
+        worker_version = worker_run["worker_version"]
         logger.info(
             f"Loaded worker {worker_version['worker']['name']} revision {worker_version['revision']['hash'][0:7]} from API"
         )
@@ -200,18 +205,10 @@ class BaseWorker(object):
         self.secrets = {name: self.load_secret(name) for name in required_secrets}
 
         # Load worker run configuration when available
-        if os.environ.get("ARKINDEX_WORKER_RUN_ID"):
-            worker_run = self.request(
-                "RetrieveWorkerRun", id=os.environ["ARKINDEX_WORKER_RUN_ID"]
-            )
-            configuration_id = worker_run.get("configuration_id")
-            if configuration_id:
-                worker_configuration = self.request(
-                    "RetrieveWorkerConfiguration", id=configuration_id
-                )
-                self.user_configuration = worker_configuration.get("configuration")
-                if self.user_configuration:
-                    logger.info("Loaded user configuration from WorkerRun")
+        worker_configuration = worker_run.get("configuration")
+        self.user_configuration = worker_configuration.get("configuration")
+        if self.user_configuration:
+            logger.info("Loaded user configuration from WorkerRun")
 
         # if debug mode is set to true activate debug mode in logger
         if self.user_configuration and self.user_configuration.get("debug"):