From 61b3119b3d1750b8ea6329e80d6d320b77311eea Mon Sep 17 00:00:00 2001
From: Eva Bardou <ebardou@teklia.com>
Date: Thu, 4 Mar 2021 17:51:33 +0100
Subject: [PATCH] Add create_elements function
---
arkindex_worker/worker.py | 59 +++++++++++++++++++++++++++++++++++++++
1 file changed, 59 insertions(+)
diff --git a/arkindex_worker/worker.py b/arkindex_worker/worker.py
index e00e4952..cdccfeef 100644
--- a/arkindex_worker/worker.py
+++ b/arkindex_worker/worker.py
@@ -394,6 +394,65 @@ class ElementsWorker(BaseWorker):
return sub_element["id"]
+ def create_elements(self, parent, elements):
+ """
+ Create children elements on the given element through API
+ Return the IDs of created elements
+ """
+ assert parent and isinstance(
+ parent, Element
+ ), "element shouldn't be null and should be of type Element"
+ assert elements and isinstance(
+ elements, list
+ ), "elements shouldn't be null and should be of type list"
+
+ for index, element in enumerate(elements):
+ assert isinstance(
+ element, dict
+ ), f"Element at index {index} in elements: Should be of type dict"
+
+ name = element.get("name")
+ assert name and isinstance(
+ name, str
+ ), f"Element at index {index} in elements: name shouldn't be null and should be of type str"
+
+ type = element.get("type")
+ assert type and isinstance(
+ type, str
+ ), f"Element at index {index} in elements: type shouldn't be null and should be of type str"
+
+ polygon = element.get("polygon")
+ assert polygon and isinstance(
+ polygon, list
+ ), f"Element at index {index} in elements: polygon shouldn't be null and should be of type list"
+ assert (
+ len(polygon) >= 3
+ ), f"Element at index {index} in elements: polygon should have at least three points"
+ assert all(
+ isinstance(point, list) and len(point) == 2 for point in polygon
+ ), f"Element at index {index} in elements: polygon points should be lists of two items"
+ assert all(
+ isinstance(coord, (int, float)) for point in polygon for coord in point
+ ), f"Element at index {index} in elements: polygon points should be lists of two numbers"
+
+ if self.is_read_only:
+ logger.warning("Cannot create elements as this worker is in read-only mode")
+ return
+
+ created_ids = self.api_client.request(
+ "CreateElements",
+ id=self.parent.id,
+ body={
+ "worker_version": self.worker_version_id,
+ "elements": elements,
+ },
+ )
+
+ for element in elements:
+ self.report.add_element(parent.id, element["type"])
+
+ return created_ids
+
def create_transcription(self, element, text, type=None, score=None):
"""
Create a transcription on the given element through the API.
--
GitLab