From f904431ede2ab1b29ce90910d9e33e665552efc2 Mon Sep 17 00:00:00 2001
From: Erwan Rouchet <rouchet@teklia.com>
Date: Fri, 2 Apr 2021 15:58:46 +0200
Subject: [PATCH] Restore get_worker_version_slug
---
arkindex_worker/worker/version.py | 10 ++++++++--
1 file changed, 8 insertions(+), 2 deletions(-)
diff --git a/arkindex_worker/worker/version.py b/arkindex_worker/worker/version.py
index 39d75aab..df12660a 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)
--
GitLab