Skip to content
Snippets Groups Projects

Use WORKER_VERSION_ID env var in helper methods

Merged Eva Bardou requested to merge use-workerversion-id into master
All threads resolved!
3 files
+ 27
0
Compare changes
  • Side-by-side
  • Inline
Files
3
+ 7
0
@@ -30,6 +30,12 @@ class BaseWorker(object):
self.work_dir = os.path.join(xdg_data_home, "arkindex")
os.makedirs(self.work_dir, exist_ok=True)
self.worker_version_id = os.environ.get("WORKER_VERSION_ID")
if not self.worker_version_id:
raise Exception(
"Missing WORKER_VERSION_ID environment variable to start the Worker"
)
logger.info(f"Worker will use {self.work_dir} as working directory")
def configure(self):
@@ -192,6 +198,7 @@ class ElementsWorker(BaseWorker):
"corpus": element.corpus.id,
"polygon": polygon,
"parent": element.id,
"worker_version": self.worker_version_id,
},
)
self.report.add_element(element.id, type)
Loading