diff --git a/arkindex_worker/worker/__init__.py b/arkindex_worker/worker/__init__.py index 25696d60e558117d28b1bc8d9ed28159fdf9571d..37998b10dd520410adda7cba78f3f06e6e66bce5 100644 --- a/arkindex_worker/worker/__init__.py +++ b/arkindex_worker/worker/__init__.py @@ -137,6 +137,8 @@ class ElementsWorker( def configure(self): super().configure() + super().configure_cache() + # Add report concerning elements self.report = Reporter( **self.worker_details, version=getattr(self, "worker_version_id", None) diff --git a/arkindex_worker/worker/base.py b/arkindex_worker/worker/base.py index 56079e7a75c46ca8893799bfe17d3996b9d794b4..94a0da013235065b4aff1f3c9794a425b9ccf2e6 100644 --- a/arkindex_worker/worker/base.py +++ b/arkindex_worker/worker/base.py @@ -211,6 +211,7 @@ class BaseWorker(object): logger.setLevel(logging.DEBUG) logger.debug("Debug output enabled") + def configure_cache(self): task_id = os.environ.get("PONOS_TASK") paths = None if self.support_cache and self.args.database is not None: diff --git a/tests/conftest.py b/tests/conftest.py index b82138f82da3e616958a6a884e39b3a78057ae50..64af8eb13a10e4299e3b4dcf71841ab821da3cbb 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -286,6 +286,7 @@ def mock_elements_worker_with_cache(monkeypatch, mock_config_api, tmp_path): worker = ElementsWorker(support_cache=True) worker.configure() + worker.configure_cache() return worker diff --git a/tests/test_merge.py b/tests/test_merge.py index 687b7792526a9973e31f33da80a8651a1a87e730..764287a9ca80c3d666709bf4ae27c143a6ad003d 100644 --- a/tests/test_merge.py +++ b/tests/test_merge.py @@ -182,6 +182,7 @@ def test_merge_from_worker( # Create the task's output dir, so that it can create its own database (tmpdir / "my_task").mkdir() mock_base_worker_with_cache.configure() + mock_base_worker_with_cache.configure_cache() # Then we have 2 elements and a transcription assert CachedImage.select().count() == 0