Skip to content
Snippets Groups Projects
Commit 9710ea5e authored by Manon Blanco's avatar Manon Blanco
Browse files

Use list_corpus_types

parent 783178ed
No related tags found
1 merge request!15Draft: Migrate cache to use the Arkindex export structure
Pipeline #193472 failed
......@@ -10,6 +10,7 @@ from time import sleep
from arkindex_export import Image as CachedImage
from arkindex_export import ImageServer as CachedImageServer
from arkindex_worker.worker.element import ElementMixin
from arkindex_export import create_database
from arkindex_worker.cache import CachedElement
......@@ -58,7 +59,7 @@ class ActivityState(Enum):
"""
class InitElementsWorker(BaseWorker):
class InitElementsWorker(BaseWorker, ElementMixin):
def configure(self) -> None:
super().configure()
......@@ -182,13 +183,9 @@ class InitElementsWorker(BaseWorker):
"corpus"
), "This worker only supports processes on corpora."
# TODO: Can't use `from arkindex_worker.worker.element.ElementMixin.list_corpus_types`
# Because `self.corpus_id` is not set?
corpus = self.api_client.request(
"RetrieveCorpus", id=self.process_information["corpus"]
)
self.list_corpus_types()
type_slugs = {
element_type["id"]: element_type["slug"] for element_type in corpus["types"]
element_type["id"]: element_type["slug"] for element_type in self.corpus_types.values()
}
elements = [
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment