Patch endpoints

CHANGES.md

@@ -2,6 +2,8 @@
 
 ## [Unreleased]
 
+* Add Item and Collection `PATCH` endpoints with support for [RFC 6902](https://tools.ietf.org/html/rfc6902) and [RFC 7396](https://tools.ietf.org/html/rfc7386)
+
 ## [5.1.1] - 2025-03-17
 
 ### Fixed
@@ -139,8 +141,7 @@
 
 ## [3.0.0] - 2024-07-29
 
-Full changelog: https://stac-utils.github.io/stac-fastapi/migrations/v3.0.0/#changelog
-
+Full changelog: https://stac-utils.github.io/stac-fastapi/migrations/v3.0.0/#changelog 
 **Changes since 3.0.0b3:**
 
 ### Changed

stac_fastapi/api/tests/test_api.py

@@ -3,10 +3,7 @@
 
 from stac_fastapi.api.app import StacApi
 from stac_fastapi.api.models import ItemCollectionUri, create_request_model
-from stac_fastapi.extensions.core import (
-    TokenPaginationExtension,
-    TransactionExtension,
-)
+from stac_fastapi.extensions.core import TokenPaginationExtension, TransactionExtension
 from stac_fastapi.types import config, core
 
 
@@ -430,6 +427,9 @@ def create_item(self, *args, **kwargs):
     def update_item(self, *args, **kwargs):
         return "dummy response"
 
+    def patch_item(self, *args, **kwargs):
+        return "dummy response"
+
     def delete_item(self, *args, **kwargs):
         return "dummy response"
 
@@ -439,6 +439,9 @@ def create_collection(self, *args, **kwargs):
     def update_collection(self, *args, **kwargs):
         return "dummy response"
 
+    def patch_collection(self, *args, **kwargs):
+        return "dummy response"
+
     def delete_collection(self, *args, **kwargs):
         return "dummy response"
 

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

@@ -5,6 +5,7 @@
 
 import attr
 from fastapi import APIRouter, Body, FastAPI
+from pydantic import TypeAdapter
 from stac_pydantic import Collection, Item, ItemCollection
 from stac_pydantic.shared import MimeTypes
 from starlette.responses import JSONResponse, Response
@@ -15,6 +16,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):
@@ -42,13 +44,33 @@ class PutItem(ItemUri):
     item: Annotated[Item, Body()] = attr.ib(default=None)
 
 
+@attr.s
+class PatchItem(ItemUri):
+    """Patch Item."""
+
+    patch: Annotated[
+        Union[PartialItem, List[PatchOperation]],
+        Body(),
+    ] = attr.ib(default=None)
+
+
 @attr.s
 class PutCollection(CollectionUri):
     """Update Collection."""
 
     collection: Annotated[Collection, Body()] = attr.ib(default=None)
 
 
+@attr.s
+class PatchCollection(CollectionUri):
+    """Patch Collection."""
+
+    patch: Annotated[
+        Union[PartialCollection, List[PatchOperation]],
+        Body(),
+    ] = attr.ib(default=None)
+
+
 @attr.s
 class TransactionExtension(ApiExtension):
     """Transaction Extension.
@@ -126,6 +148,82 @@ def register_update_item(self):
             endpoint=create_async_endpoint(self.client.update_item, PutItem),
         )
 
+    def register_patch_item(self):
+        """Register patch item endpoint (PATCH
+        /collections/{collection_id}/items/{item_id})."""
+        self.router.add_api_route(
+            name="Patch Item",
+            path="/collections/{collection_id}/items/{item_id}",
+            response_model=Item if self.settings.enable_response_models else None,
+            responses={
+                200: {
+                    "content": {
+                        MimeTypes.geojson.value: {},
+                    },
+                    "model": Item,
+                }
+            },
+            openapi_extra={
+                "requestBody": {
+                    "content": {
+                        "application/json-patch+json": {
+                            "schema": TypeAdapter(List[PatchOperation]).json_schema()
+                            | {
+                                "examples": [
+                                    [
+                                        {
+                                            "op": "add",
+                                            "path": "/properties/foo",
+                                            "value": "bar",
+                                        },
+                                        {
+                                            "op": "replace",
+                                            "path": "/properties/foo",
+                                            "value": "bar",
+                                        },
+                                        {
+                                            "op": "test",
+                                            "path": "/properties/foo",
+                                            "value": "bar",
+                                        },
+                                        {
+                                            "op": "copy",
+                                            "path": "/properties/foo",
+                                            "from": "/properties/bar",
+                                        },
+                                        {
+                                            "op": "move",
+                                            "path": "/properties/foo",
+                                            "from": "/properties/bar",
+                                        },
+                                        {
+                                            "op": "remove",
+                                            "path": "/properties/foo",
+                                        },
+                                    ]
+                                ]
+                            },
+                        },
+                        "application/merge-patch+json": {
+                            "schema": TypeAdapter(PartialItem).json_schema(),
+                        },
+                        "application/json": {
+                            "schema": TypeAdapter(PartialItem).json_schema(),
+                        },
+                    },
+                    "required": True,
+                },
+            },
+            response_class=self.response_class,
+            response_model_exclude_unset=True,
+            response_model_exclude_none=True,
+            methods=["PATCH"],
+            endpoint=create_async_endpoint(
+                self.client.patch_item,
+                PatchItem,
+            ),
+        )
+
     def register_delete_item(self):
         """Register delete item endpoint (DELETE
         /collections/{collection_id}/items/{item_id})."""
@@ -148,11 +246,6 @@ def register_delete_item(self):
             endpoint=create_async_endpoint(self.client.delete_item, ItemUri),
         )
 
-    def register_patch_item(self):
-        """Register patch item endpoint (PATCH
-        /collections/{collection_id}/items/{item_id})."""
-        raise NotImplementedError
-
     def register_create_collection(self):
         """Register create collection endpoint (POST /collections)."""
         self.router.add_api_route(
@@ -196,6 +289,81 @@ def register_update_collection(self):
             endpoint=create_async_endpoint(self.client.update_collection, PutCollection),
         )
 
+    def register_patch_collection(self):
+        """Register patch collection endpoint (PATCH /collections/{collection_id})."""
+        self.router.add_api_route(
+            name="Patch Collection",
+            path="/collections/{collection_id}",
+            response_model=Collection if self.settings.enable_response_models else None,
+            responses={
+                200: {
+                    "content": {
+                        MimeTypes.geojson.value: {},
+                    },
+                    "model": Collection,
+                }
+            },
+            openapi_extra={
+                "requestBody": {
+                    "content": {
+                        "application/json-patch+json": {
+                            "schema": TypeAdapter(List[PatchOperation]).json_schema()
+                            | {
+                                "examples": [
+                                    [
+                                        {
+                                            "op": "add",
+                                            "path": "/summeries/foo",
+                                            "value": "bar",
+                                        },
+                                        {
+                                            "op": "replace",
+                                            "path": "/summeries/foo",
+                                            "value": "bar",
+                                        },
+                                        {
+                                            "op": "test",
+                                            "path": "/summeries/foo",
+                                            "value": "bar",
+                                        },
+                                        {
+                                            "op": "copy",
+                                            "path": "/summeries/foo",
+                                            "from": "/summeries/bar",
+                                        },
+                                        {
+                                            "op": "move",
+                                            "path": "/summeries/foo",
+                                            "from": "/summeries/bar",
+                                        },
+                                        {
+                                            "op": "remove",
+                                            "path": "/summeries/foo",
+                                        },
+                                    ]
+                                ]
+                            },
+                        },
+                        "application/merge-patch+json": {
+                            "schema": TypeAdapter(PartialCollection).json_schema(),
+                        },
+                        "application/json": {
+                            "schema": TypeAdapter(PartialCollection).json_schema(),
+                        },
+                    },
+                    "required": True,
+                },
+            },
+            response_class=self.response_class,
+            response_model_exclude_unset=True,
+            response_model_exclude_none=True,
+            methods=["PATCH"],
+            endpoint=create_async_endpoint(
+                self.client.patch_collection,
+                PatchCollection,
+            ),
+        )
+
     def register_delete_collection(self):
         """Register delete collection endpoint (DELETE /collections/{collection_id})."""
         self.router.add_api_route(
@@ -217,10 +385,6 @@ def register_delete_collection(self):
             endpoint=create_async_endpoint(self.client.delete_collection, CollectionUri),
         )
 
-    def register_patch_collection(self):
-        """Register patch collection endpoint (PATCH /collections/{collection_id})."""
-        raise NotImplementedError
-
     def register(self, app: FastAPI) -> None:
         """Register the extension with a FastAPI application.
 
@@ -233,8 +397,10 @@ def register(self, app: FastAPI) -> None:
         self.router.prefix = app.state.router_prefix
         self.register_create_item()
         self.register_update_item()
+        self.register_patch_item()
         self.register_delete_item()
         self.register_create_collection()
         self.register_update_collection()
+        self.register_patch_collection()
         self.register_delete_collection()
         app.include_router(self.router, tags=["Transaction Extension"])

stac_fastapi/extensions/tests/test_transaction.py

@@ -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):
@@ -46,6 +47,19 @@ def update_item(self, collection_id: str, item_id: str, item: Item, **kwargs):
             "type": item.type,
         }
 
+    def patch_item(
+        self,
+        collection_id: str,
+        item_id: str,
+        patch: Union[PartialItem, PatchOperation],
+        **kwargs,
+    ):
+        return {
+            "path_collection_id": collection_id,
+            "path_item_id": item_id,
+            "patch": patch,
+        }
+
     def delete_item(self, item_id: str, collection_id: str, **kwargs):
         return {
             "path_collection_id": collection_id,
@@ -58,6 +72,17 @@ def create_collection(self, collection: Collection, **kwargs):
     def update_collection(self, collection_id: str, collection: Collection, **kwargs):
         return {"path_collection_id": collection_id, "type": collection.type}
 
+    def patch_collection(
+        self,
+        collection_id: str,
+        patch: Union[PartialCollection, PatchOperation],
+        **kwargs,
+    ):
+        return {
+            "path_collection_id": collection_id,
+            "patch": patch,
+        }
+
     def delete_collection(self, collection_id: str, **kwargs):
         return {"path_collection_id": collection_id}
 
@@ -88,6 +113,19 @@ def test_update_item(client: TestClient, item: Item) -> None:
     assert response.json()["type"] == "Feature"
 
 
+def test_patch_item(client: TestClient) -> None:
+    response = client.patch(
+        "/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"] == [
+        {"op": "add", "path": "/properties/foo", "value": "bar"}
+    ]
+
+
 def test_delete_item(client: TestClient) -> None:
     response = client.delete("/collections/a-collection/items/an-item")
     assert response.is_success, response.text
@@ -108,6 +146,18 @@ def test_update_collection(client: TestClient, collection: Collection) -> None:
     assert response.json()["type"] == "Collection"
 
 
+def test_patch_collection(client: TestClient) -> None:
+    response = client.patch(
+        "/collections/a-collection",
+        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"] == [
+        {"op": "add", "path": "/properties/foo", "value": "bar"}
+    ]
+
+
 def test_delete_collection(client: TestClient, collection: Collection) -> None:
     response = client.delete("/collections/a-collection")
     assert response.is_success, response.text