diff --git a/arkindex_worker/worker/version.py b/arkindex_worker/worker/version.py index 39d75aabe2bd22624cdcf161fa82c9155b309e50..df12660a9dab88babfebabcd0fc5ba62ec9103c8 100644 --- a/arkindex_worker/worker/version.py +++ b/arkindex_worker/worker/version.py @@ -14,6 +14,13 @@ class WorkerVersionMixin(object): return worker_version + def get_worker_version_slug(self, worker_version_id: str) -> str: + """ + Get worker version slug from cache if possible, otherwise make API request + """ + worker_version = self.get_worker_version(worker_version_id) + return worker_version["worker"]["slug"] + def get_ml_result_slug(self, ml_result) -> str: """ Helper function to get the worker slug from element, classification or transcription. @@ -36,5 +43,4 @@ class WorkerVersionMixin(object): if worker_version_id is None: return - worker_version = self.get_worker_version(worker_version_id) - return worker_version["worker"]["slug"] + return self.get_worker_version_slug(worker_version_id)