Merge branch 'main' into feat/object-oriented-api

Author: Lucashsmello

datamint/api/dto/__init__.py

@@ -13,4 +13,6 @@
     "BoxGeometry",
     "LineGeometry",
     "CoordinateSystem"
+    "LineGeometry",
+    "CoordinateSystem"
 ]

datamint/api/endpoints/annotations_api.py

@@ -383,7 +383,7 @@ def create(self,
 
         annotations = [annotation_dto] if isinstance(annotation_dto, CreateAnnotationDto) else annotation_dto
         annotations = [ann.to_dict() if isinstance(ann, CreateAnnotationDto) else ann for ann in annotations]
-        resource_id = resource.id if isinstance(resource, Resource) else resource
+        resource_id = self._entid(resource)
         respdata = self._make_request('POST',
                                       f'{self.endpoint_base}/{resource_id}/annotations',
                                       json=annotations).json()
@@ -786,6 +786,37 @@ def _numpy_to_bytesio_png(seg_imgs: np.ndarray) -> Generator[BinaryIO, None, Non
             img_bytes.seek(0)
             yield img_bytes
 
+    def create_image_classification(self,
+                                    resource: str | Resource,
+                                    identifier: str,
+                                    value: str,
+                                    imported_from: str | None = None,
+                                    model_id: str | None = None,
+                                    ) -> str:
+        """
+        Create an image-level classification annotation.
+
+        Args:
+            resource: The resource unique id or Resource instance.
+            identifier: The annotation identifier/label.
+            value: The classification value.
+            imported_from: The imported from source value.
+            model_id: The model unique id.
+
+        Returns:
+            The id of the created annotation.
+        """
+        annotation_dto = CreateAnnotationDto(
+            type=AnnotationType.CATEGORY,
+            identifier=identifier,
+            scope='image',
+            value=value,
+            imported_from=imported_from,
+            model_id=model_id
+        )
+
+        return self.create(resource, annotation_dto)            
+
     def add_line_annotation(self,
                             point1: tuple[int, int] | tuple[float, float, float],
                             point2: tuple[int, int] | tuple[float, float, float],

datamint/api/endpoints/resources_api.py

@@ -988,6 +988,12 @@ def set_tags(self,
                  resource: str | Resource,
                  tags: Sequence[str],
                  ):
+        """
+        Set tags for a resource, IMPORTANT: This replaces all existing tags.
+        Args:
+            resource: The resource unique id or Resource object.
+            tags: The tags to set.
+        """
         data = {'tags': tags}
         resource_id = self._entid(resource)
 
@@ -1010,3 +1016,39 @@ def set_tags(self,
     #     resource._ensure_attr('projects')
     #     proj_ids = [p['id'] for p in resource.projects]
     #     return [proj for proj in self.projects_api.get_all() if proj.id in proj_ids]
+
+    def add_tags(self,
+                 resource: str | Resource,
+                 tags: Sequence[str],
+                 ):
+        """
+        Add tags to a resource, IMPORTANT: This appends to existing tags.
+        Args:
+            resource: The resource unique id or Resource object.
+            tags: The tags to add.
+        """
+        if isinstance(resource, str):
+            resource = self.get_by_id(resource)
+        old_tags = resource.tags if resource.tags is not None else []
+        return self.set_tags(resource, old_tags + list(tags))
+
+    def bulk_delete(self, entities: Sequence[str | Resource]) -> None:
+        """Delete multiple entities. Faster than deleting them one by one.
+
+        Args:
+            entities: Sequence of unique identifiers for the entities to delete or the entity instances themselves.
+
+        Raises:
+            httpx.HTTPStatusError: If deletion fails or any entity not found
+        """
+        from math import ceil
+
+        resources_ids = [self._entid(ent) for ent in entities]
+        if len(resources_ids) == 0:
+            return
+        batch_size = 200
+        for i in range(0, ceil(len(resources_ids)/batch_size)):
+            batch_ids = resources_ids[i*batch_size:(i+1)*batch_size]
+            self._make_request('DELETE',
+                               f'{self.endpoint_base}',
+                               params={'resource_ids': ','.join(batch_ids)})

datamint/api/entity_base_api.py

@@ -189,7 +189,7 @@ class DeletableEntityApi(EntityBaseApi[T]):
     retrieval and restoration of such entities.
     """
 
-    def delete(self, entity: str | BaseEntity) -> None:
+    def delete(self, entity: str | T) -> None:
         """Delete an entity.
 
         Args:
@@ -200,7 +200,7 @@ def delete(self, entity: str | BaseEntity) -> None:
         """
         self._make_entity_request('DELETE', entity)
 
-    def bulk_delete(self, entities: Sequence[str | BaseEntity]) -> None:
+    def bulk_delete(self, entities: Sequence[str | T]) -> None:
         """Delete multiple entities.
 
         Args: