datamint 2.4.0__py3-none-any.whl → 2.4.1__py3-none-any.whl

datamint/api/endpoints/resources_api.py

@@ -25,6 +25,7 @@ import nest_asyncio  # For running asyncio in jupyter notebooks
 from PIL import Image
 import io
 from datamint.types import ImagingData
+from collections import defaultdict
 
 
 _LOGGER = logging.getLogger(__name__)
@@ -279,7 +280,6 @@ class ResourcesApi(CreatableEntityApi[Resource], DeletableEntityApi[Resource]):
             _LOGGER.warning(msg)
             _USER_LOGGER.warning(msg)
 
-
         mimetype = standardize_mimetype(mimetype)
 
         if is_a_dicom_file == True or is_dicom(file_path):
@@ -440,12 +440,12 @@ class ResourcesApi(CreatableEntityApi[Resource], DeletableEntityApi[Resource]):
 
             try:
                 tasks = [__upload_single_resource(f, segfiles, metadata_file)
-                        for f, segfiles, metadata_file in zip(files_path, segmentation_files, metadata_files)]
+                         for f, segfiles, metadata_file in zip(files_path, segmentation_files, metadata_files)]
             except ValueError:
                 msg = f"Error preparing upload tasks. Try `assemble_dicom=False`."
                 _LOGGER.error(msg)
                 _USER_LOGGER.error(msg)
-                raise 
+                raise
             return await asyncio.gather(*tasks, return_exceptions=on_error == 'skip')
 
     def upload_resources(self,
@@ -996,22 +996,28 @@ class ResourcesApi(CreatableEntityApi[Resource], DeletableEntityApi[Resource]):
                     raise
 
     def set_tags(self,
-                 resource: str | Resource,
+                 resource: str | Resource | Sequence[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.
+            resource: The resource object or a list of resources.
             tags: The tags to set.
         """
         data = {'tags': tags}
-        resource_id = self._entid(resource)
-
-        response = self._make_entity_request('PUT',
-                                             resource_id,
-                                             add_path='tags',
-                                             json=data)
+        if isinstance(resource, Sequence):
+            resource_ids = [self._entid(res) for res in resource]
+            response = self._make_request('PUT',
+                                          f'{self.endpoint_base}/tags',
+                                          json={'resource_ids': resource_ids,
+                                                'tags': tags})
+        else:
+            resource_id = self._entid(resource)
+            response = self._make_entity_request('PUT',
+                                                 resource_id,
+                                                 add_path='tags',
+                                                 json=data)
         return response
 
     # def get_projects(self, resource: Resource) -> Sequence[Project]:
@@ -1029,7 +1035,7 @@ class ResourcesApi(CreatableEntityApi[Resource], DeletableEntityApi[Resource]):
     #     return [proj for proj in self.projects_api.get_all() if proj.id in proj_ids]
 
     def add_tags(self,
-                 resource: str | Resource,
+                 resource: str | Resource | Sequence[str | Resource],
                  tags: Sequence[str],
                  ):
         """
@@ -1040,8 +1046,26 @@ class ResourcesApi(CreatableEntityApi[Resource], DeletableEntityApi[Resource]):
         """
         if isinstance(resource, str):
             resource = self.get_by_id(resource)
+        elif isinstance(resource, Sequence):
+            # Transform every str to Resource first.
+            resources = [self.get_by_id(res) if isinstance(res, str) else res for res in resource]
+
+            # group resource having the exact same tags to minimize requests
+            tag_map: dict[tuple, list[Resource]] = defaultdict(list)
+            for res in resources:
+                old_tags = res.tags if res.tags is not None else []
+                # key = tuple(sorted(old_tags))
+                key = tuple(old_tags)  # keep order, assuming order matters for tags
+                tag_map[key].append(res)
+
+            # finally, set tags for each group
+            for old_tags_tuple, res_group in tag_map.items():
+                old_tags = list(old_tags_tuple)
+                self.set_tags(res_group, old_tags + list(tags))
+            return
+
         old_tags = resource.tags if resource.tags is not None else []
-        return self.set_tags(resource, old_tags + list(tags))
+        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.

{datamint-2.4.0.dist-info → datamint-2.4.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: datamint
-Version: 2.4.0
+Version: 2.4.1
 Summary: A library for interacting with the Datamint API, designed for efficient data management, processing and Deep Learning workflows.
 Requires-Python: >=3.10
 Classifier: Programming Language :: Python :: 3

{datamint-2.4.0.dist-info → datamint-2.4.1.dist-info}/RECORD

@@ -10,7 +10,7 @@ datamint/api/endpoints/channels_api.py,sha256=oQqxSw9DJzAqtVQI7-tc1llTdnsm-URx8j
 datamint/api/endpoints/datasetsinfo_api.py,sha256=WdzrUzK63w9gvAP6U--P65FbD-3X-jm9TPCcYnRNjas,597
 datamint/api/endpoints/models_api.py,sha256=tbVuajc-mCsIp5AKSCoq3uQRDWgKnJaIA6tf_ck8-XY,1502
 datamint/api/endpoints/projects_api.py,sha256=Pfr3fEiMw_aRzoGtcVXHJQ68leVoPAghw23L4ZIglno,8237
-datamint/api/endpoints/resources_api.py,sha256=ZsGy4xTdLHLkM2z8VyRRLEjB0Rv3kTZFQceYkFYpk7A,50601
+datamint/api/endpoints/resources_api.py,sha256=TwePvZJp0yBXYoWLzBe4BXucPWAQWjiokzWyQDsD29E,51950
 datamint/api/endpoints/users_api.py,sha256=pnkuTZ1B9Y0FtwwvXO8J64e02RSkRxnBmTl9UGSuC5I,1186
 datamint/api/entity_base_api.py,sha256=-8SIt4M8P9G2b8SQznuWpFuFE8zEQjQxkRkw0s_w0Y4,11692
 datamint/apihandler/annotation_api_handler.py,sha256=W3vV4z3BqX1OQe1r7zr8dI-IVu4zUDxED4QttdiWV-E,57098
@@ -61,7 +61,7 @@ datamint/types.py,sha256=2OaY5QJvQIJKxyMNJYzxBksKCa9ZS2gb_ayJrByvu2Y,410
 datamint/utils/logging_utils.py,sha256=9pRoaPrWu2jOdDCiAoUsjEdP5ZwaealWL3hjUqFvx9g,4022
 datamint/utils/torchmetrics.py,sha256=lwU0nOtsSWfebyp7dvjlAggaqXtj5ohSEUXOg3L0hJE,2837
 datamint/utils/visualization.py,sha256=yaUVAOHar59VrGUjpAWv5eVvQSfztFG0eP9p5Vt3l-M,4470
-datamint-2.4.0.dist-info/METADATA,sha256=g4WkKWDq3mBETvQibcoISGXTtQCRdebfrw4leyurg3E,8974
-datamint-2.4.0.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
-datamint-2.4.0.dist-info/entry_points.txt,sha256=wTr9DXteCfAFqLUstN1Qx_SqC_rFLrAUIcDx9q1eXZw,845
-datamint-2.4.0.dist-info/RECORD,,
+datamint-2.4.1.dist-info/METADATA,sha256=aViMZ43PB9dY38iwu36keNIcnp__cIpV16NRNOKcT7g,8974
+datamint-2.4.1.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
+datamint-2.4.1.dist-info/entry_points.txt,sha256=wTr9DXteCfAFqLUstN1Qx_SqC_rFLrAUIcDx9q1eXZw,845
+datamint-2.4.1.dist-info/RECORD,,