Patch endpoints

stac_fastapi/types/stac_fastapi/types/core.py

@@ -21,11 +21,7 @@
 from stac_fastapi.types.requests import get_base_url
 from stac_fastapi.types.rfc3339 import DateTimeType
 from stac_fastapi.types.search import BaseSearchPostRequest
-from stac_fastapi.types.transaction import (
-    PartialCollection,
-    PartialItem,
-    PatchOperation,
-)
+from stac_fastapi.types.transaction import PartialCollection, PartialItem, PatchOperation
 
 __all__ = [
     "NumType",
@@ -92,6 +88,7 @@ def patch_item(
         collection_id: str,
         item_id: str,
         patch: Union[PartialItem, List[PatchOperation]],
+        content_type: Optional[str],
         **kwargs,
     ) -> Optional[Union[stac.Item, Response]]:
         """Update an item from a collection.
@@ -106,7 +103,6 @@ def patch_item(
         Returns:
             The patched item.
         """
-        content_type = kwargs["request"].headers.get("content-type", "application/json")
         if isinstance(patch, list) and content_type == "application/json-patch+json":
             return self.json_patch_item(
                 collection_id,
@@ -233,6 +229,7 @@ def patch_collection(
         self,
         collection_id: str,
         patch: Union[PartialCollection, List[PatchOperation]],
+        content_type: Optional[str],
         **kwargs,
     ) -> Optional[Union[stac.Collection, Response]]:
         """Update a collection.
@@ -246,7 +243,6 @@ def patch_collection(
         Returns:
             The patched collection.
         """
-        content_type = kwargs["request"].headers.get("content-type", "application/json")
         if isinstance(patch, list) and content_type == "application/json-patch+json":
             return self.json_patch_collection(collection_id, patch, **kwargs)
 
@@ -371,6 +367,7 @@ async def patch_item(
         collection_id: str,
         item_id: str,
         patch: Union[PartialItem, List[PatchOperation]],
+        content_type: Optional[str],
         **kwargs,
     ) -> Optional[Union[stac.Item, Response]]:
         """Update an item from a collection.
@@ -385,7 +382,7 @@ async def patch_item(
         Returns:
             The patched item.
         """
-        content_type = kwargs["request"].headers.get("content-type", "application/json")
+        print("content_type", content_type)
         if isinstance(patch, list) and content_type == "application/json-patch+json":
             return await self.json_patch_item(
                 collection_id,
@@ -512,6 +509,7 @@ async def patch_collection(
         self,
         collection_id: str,
         patch: Union[PartialCollection, List[PatchOperation]],
+        content_type: Optional[str],
         **kwargs,
     ) -> Optional[Union[stac.Collection, Response]]:
         """Update a collection.
@@ -525,7 +523,6 @@ async def patch_collection(
         Returns:
             The patched collection.
         """
-        content_type = kwargs["request"].headers.get("content-type", "application/json")
         if isinstance(patch, list) and content_type == "application/json-patch+json":
             partialCollectionValidator = TypeAdapter(PartialCollection)
 

stac_fastapi/types/stac_fastapi/types/transaction.py

@@ -4,7 +4,7 @@
 from typing import Any, Dict, List, Literal, Optional, Union
 
 import attr
-from fastapi import Body
+from fastapi import Body, Header
 from pydantic import BaseModel
 from stac_pydantic import Collection, Item, ItemCollection
 from stac_pydantic.shared import BBox
@@ -106,6 +106,10 @@ class PatchItem(ItemUri):
         Union[PartialItem, List[PatchOperation]],
         Body(),
     ] = attr.ib(default=None)
+    content_type: Annotated[
+        Optional[str],
+        Header(),
+    ] = attr.ib(default="application/json")
 
 
 @attr.s
@@ -123,3 +127,7 @@ class PatchCollection(CollectionUri):
         Union[PartialCollection, List[PatchOperation]],
         Body(),
     ] = attr.ib(default=None)
+    content_type: Annotated[
+        Optional[str],
+        Header(),
+    ] = attr.ib(default="application/json")