Fix collection item retrieval (#765)

Author: matheuspinheirosantos

CHANGELOG.md

@@ -30,6 +30,10 @@ You can check your current version with the following command:
 
 For more information, see [UP42 Python package description](https://pypi.org/project/up42-py/).
 
+### 2.5.0a5
+**August 15, 2025**
+- Fix how collection items are retrieved in `BulkDeletion`.
+
 ### 2.5.0a4
 **August 6, 2025**
 - Added bulk deletion of items: `BulkDeletion`.

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "up42-py"
-version = "2.5.0a4"
+version = "2.5.0a5"
 description = "Python SDK for UP42, the geospatial marketplace and developer platform."
 authors = ["UP42 GmbH <[email protected]>"]
 license = "https://github.com/up42/up42-py/blob/master/LICENSE"

tests/test_stac.py

@@ -172,6 +172,7 @@ class TestBulkDeletion:
     def test_should_raise_and_not_submit_when_missing_items(self):
         mock_stac_client = mock.Mock()
         mock_stac_client.get_items.return_value = iter([self.items[0]])
+        mock_stac_client.get_collection.return_value = self.collection_with_all_items
         bulk_deletion = stac.BulkDeletion(self.items[0].id)
         bulk_deletion.stac_client = mock_stac_client
         with pytest.raises(stac.IncompleteCollectionDeletionError):
@@ -184,6 +185,7 @@ def test_should_delete_staged_items(self, requests_mock: req_mock.Mocker):
         )
         mock_stac_client = mock.Mock()
         mock_stac_client.get_items.return_value = iter(self.items)
+        mock_stac_client.get_collection.return_value = self.collection_with_all_items
 
         bulk_deletion = stac.BulkDeletion(self.items[0].id, self.items[1].id)
         bulk_deletion.stac_client = mock_stac_client

up42/stac.py

@@ -113,23 +113,23 @@ def __init__(self, *item_ids: str):
 
     def delete(self):
         items = self.stac_client.get_items(*self._item_ids)
-        collections: set[pystac.Collection] = set()
+        collection_ids = set()
         for item in items:
-            collection = item.get_parent()
-            collections.add(collection)  # type: ignore
+            collection_ids.add(item.collection_id)
 
-        for collection in collections:
-            collection_item_ids = {item_in_collection.id for item_in_collection in collection.get_items()}
+        for collection_id in collection_ids:
+            collection_items = self.stac_client.get_collection(collection_id).get_items()
+            collection_item_ids = {item_in_collection.id for item_in_collection in collection_items}
             missing_items = collection_item_ids - self._item_ids
             if missing_items:
                 error_msg = (
                     f"The deletion request failed because the submitted items are part of a group that must be deleted "
-                    f"together. The submitted items belong to the following collection: '{collection.id}'. All items "
+                    f"together. The submitted items belong to the following collection: '{collection_id}'. All items "
                     f"from this collection must be deleted at once. To proceed, please add these missing item IDs "
                     f"to your request: {list(missing_items)}."
                 )
                 raise IncompleteCollectionDeletionError(error_msg)
 
-        for collection in collections:
-            url = host.endpoint(f"/v2/assets/stac/collections/{collection.id}")
+        for collection_id in collection_ids:
+            url = host.endpoint(f"/v2/assets/stac/collections/{collection_id}")
             self.session.delete(url=url)