back to PartialItem & PatchOperation

Author: rhysrevans3

stac_fastapi/extensions/stac_fastapi/extensions/core/transaction.py

@@ -1,7 +1,7 @@
 """Transaction extension."""
 
 from enum import Enum
-from typing import Any, List, Optional, Type, Union
+from typing import List, Optional, Type, Union
 
 import attr
 from fastapi import APIRouter, Body, FastAPI, Header
@@ -15,6 +15,7 @@
 from stac_fastapi.types.config import ApiSettings
 from stac_fastapi.types.core import AsyncBaseTransactionsClient, BaseTransactionsClient
 from stac_fastapi.types.extension import ApiExtension
+from stac_fastapi.types.stac import PartialCollection, PartialItem, PatchOperation
 
 
 class TransactionConformanceClasses(str, Enum):
@@ -47,7 +48,7 @@ class PatchItem(ItemUri):
     """Patch Item."""
 
     patch: Annotated[
-        Any,
+        Union[PartialItem, List[PatchOperation]],
         Body(),
     ] = attr.ib(default=None)
     content_type: Annotated[
@@ -68,7 +69,7 @@ class PatchCollection(CollectionUri):
     """Patch Collection."""
 
     patch: Annotated[
-        Any,
+        Union[PartialCollection, List[PatchOperation]],
         Body(),
     ] = attr.ib(default=None)
     content_type: Annotated[

stac_fastapi/extensions/tests/test_transaction.py

@@ -1,5 +1,5 @@
 import json
-from typing import Any, Iterator, Union
+from typing import Iterator, Union
 
 import pytest
 from stac_pydantic import Collection
@@ -11,6 +11,7 @@
 from stac_fastapi.extensions.core import TransactionExtension
 from stac_fastapi.types.config import ApiSettings
 from stac_fastapi.types.core import BaseCoreClient, BaseTransactionsClient
+from stac_fastapi.types.stac import PartialCollection, PartialItem, PatchOperation
 
 
 class DummyCoreClient(BaseCoreClient):
@@ -50,7 +51,7 @@ def patch_item(
         self,
         collection_id: str,
         item_id: str,
-        patch: Any,
+        patch: Union[PartialItem, PatchOperation],
         **kwargs,
     ):
         return {
@@ -74,7 +75,7 @@ def update_collection(self, collection_id: str, collection: Collection, **kwargs
     def patch_collection(
         self,
         collection_id: str,
-        patch: Any,
+        patch: Union[PartialCollection, PatchOperation],
         **kwargs,
     ):
         return {
@@ -114,12 +115,15 @@ def test_update_item(client: TestClient, item: Item) -> None:
 
 def test_patch_item(client: TestClient) -> None:
     response = client.patch(
-        "/collections/a-collection/items/an-item", content='{"patch": true}'
+        "/collections/a-collection/items/an-item",
+        content='[{"op": "add", "path": "/properties/foo", "value": "bar"}]',
     )
     assert response.is_success, response.text
     assert response.json()["path_collection_id"] == "a-collection"
     assert response.json()["path_item_id"] == "an-item"
-    assert response.json()["patch"] == {"patch": True}
+    assert response.json()["patch"] == [
+        {"op": "add", "path": "/properties/foo", "value": "bar"}
+    ]
 
 
 def test_delete_item(client: TestClient) -> None:
@@ -145,11 +149,13 @@ def test_update_collection(client: TestClient, collection: Collection) -> None:
 def test_patch_collection(client: TestClient) -> None:
     response = client.patch(
         "/collections/a-collection",
-        content='{"patch": true}',
+        content='[{"op": "add", "path": "/properties/foo", "value": "bar"}]',
     )
     assert response.is_success, response.text
     assert response.json()["path_collection_id"] == "a-collection"
-    assert response.json()["patch"] == {"patch": True}
+    assert response.json()["patch"] == [
+        {"op": "add", "path": "/properties/foo", "value": "bar"}
+    ]
 
 
 def test_delete_collection(client: TestClient, collection: Collection) -> None:

stac_fastapi/types/stac_fastapi/types/core.py

@@ -85,7 +85,7 @@ def patch_item(
         self,
         collection_id: str,
         item_id: str,
-        patch: Any,
+        patch: Union[stac.PartialItem, List[stac.PatchOperation]],
         content_type: Optional[str] = None,
         **kwargs,
     ) -> Optional[Union[stac.Item, Response]]:
@@ -160,7 +160,7 @@ def update_collection(
     def patch_collection(
         self,
         collection_id: str,
-        patch: Any,
+        patch: Union[stac.PartialItem, List[stac.PatchOperation]],
         content_type: Optional[str] = None,
         **kwargs,
     ) -> Optional[Union[stac.Collection, Response]]:
@@ -242,7 +242,7 @@ async def patch_item(
         self,
         collection_id: str,
         item_id: str,
-        patch: Any,
+        patch: Union[stac.PartialItem, List[stac.PatchOperation]],
         content_type: Optional[str] = None,
         **kwargs,
     ) -> Optional[Union[stac.Item, Response]]:
@@ -317,7 +317,7 @@ async def update_collection(
     async def patch_collection(
         self,
         collection_id: str,
-        patch: Any,
+        patch: Union[stac.PartialItem, List[stac.PatchOperation]],
         content_type: Optional[str] = None,
         **kwargs,
     ) -> Optional[Union[stac.Collection, Response]]: