pre-commit.

Author: rhysrevans3

stac_fastapi/opensearch/stac_fastapi/opensearch/database_logic.py

@@ -273,7 +273,9 @@ async def create_item_index(collection_id: str):
     }
 
     try:
-        await client.indices.create(index=f"{index_by_collection_id(collection_id)}-000001", body=search_body)
+        await client.indices.create(
+            index=f"{index_by_collection_id(collection_id)}-000001", body=search_body
+        )
     except TransportError as e:
         if e.status_code == 400:
             pass  # Ignore 400 status codes
@@ -354,8 +356,12 @@ class DatabaseLogic:
     client = AsyncSearchSettings().create_client
     sync_client = SyncSearchSettings().create_client
 
-    item_serializer: Type[serializers.ItemSerializer] = attr.ib(default=serializers.ItemSerializer)
-    collection_serializer: Type[serializers.CollectionSerializer] = attr.ib(default=serializers.CollectionSerializer)
+    item_serializer: Type[serializers.ItemSerializer] = attr.ib(
+        default=serializers.ItemSerializer
+    )
+    collection_serializer: Type[serializers.CollectionSerializer] = attr.ib(
+        default=serializers.CollectionSerializer
+    )
 
     extensions: List[str] = attr.ib(default=attr.Factory(list))
 
@@ -389,9 +395,15 @@ class DatabaseLogic:
                 "size": 10000,
             }
         },
-        "sun_elevation_frequency": {"histogram": {"field": "properties.view:sun_elevation", "interval": 5}},
-        "sun_azimuth_frequency": {"histogram": {"field": "properties.view:sun_azimuth", "interval": 5}},
-        "off_nadir_frequency": {"histogram": {"field": "properties.view:off_nadir", "interval": 5}},
+        "sun_elevation_frequency": {
+            "histogram": {"field": "properties.view:sun_elevation", "interval": 5}
+        },
+        "sun_azimuth_frequency": {
+            "histogram": {"field": "properties.view:sun_azimuth", "interval": 5}
+        },
+        "off_nadir_frequency": {
+            "histogram": {"field": "properties.view:off_nadir", "interval": 5}
+        },
         "centroid_geohash_grid_frequency": {
             "geohash_grid": {
                 "field": "properties.proj:centroid",
@@ -494,7 +506,9 @@ async def get_one_item(self, collection_id: str, item_id: str) -> Dict:
                 id=mk_item_id(item_id, collection_id),
             )
         except exceptions.NotFoundError:
-            raise NotFoundError(f"Item {item_id} does not exist in Collection {collection_id}")
+            raise NotFoundError(
+                f"Item {item_id} does not exist in Collection {collection_id}"
+            )
         return item["_source"]
 
     @staticmethod
@@ -517,7 +531,9 @@ def apply_free_text_filter(search: Search, free_text_queries: Optional[List[str]
         """Database logic to perform query for search endpoint."""
         if free_text_queries is not None:
             free_text_query_string = '" OR properties.\\*:"'.join(free_text_queries)
-            search = search.query("query_string", query=f'properties.\\*:"{free_text_query_string}"')
+            search = search.query(
+                "query_string", query=f'properties.\\*:"{free_text_query_string}"'
+            )
 
         return search
 
@@ -533,10 +549,16 @@ def apply_datetime_filter(search: Search, datetime_search):
             Search: The filtered search object.
         """
         if "eq" in datetime_search:
-            search = search.filter("term", **{"properties__datetime": datetime_search["eq"]})
+            search = search.filter(
+                "term", **{"properties__datetime": datetime_search["eq"]}
+            )
         else:
-            search = search.filter("range", properties__datetime={"lte": datetime_search["lte"]})
-            search = search.filter("range", properties__datetime={"gte": datetime_search["gte"]})
+            search = search.filter(
+                "range", properties__datetime={"lte": datetime_search["lte"]}
+            )
+            search = search.filter(
+                "range", properties__datetime={"gte": datetime_search["gte"]}
+            )
         return search
 
     @staticmethod
@@ -739,7 +761,11 @@ async def execute_search(
             if hits and (sort_array := hits[limit - 1].get("sort")):
                 next_token = urlsafe_b64encode(json.dumps(sort_array).encode()).decode()
 
-        matched = es_response["hits"]["total"]["value"] if es_response["hits"]["total"]["relation"] == "eq" else None
+        matched = (
+            es_response["hits"]["total"]["value"]
+            if es_response["hits"]["total"]["relation"] == "eq"
+            else None
+        )
         if count_task.done():
             try:
                 matched = count_task.result().get("count")
@@ -817,7 +843,9 @@ async def check_collection_exists(self, collection_id: str):
         if not await self.client.exists(index=COLLECTIONS_INDEX, id=collection_id):
             raise NotFoundError(f"Collection {collection_id} does not exist")
 
-    async def prep_create_item(self, item: Item, base_url: str, exist_ok: bool = False) -> Item:
+    async def prep_create_item(
+        self, item: Item, base_url: str, exist_ok: bool = False
+    ) -> Item:
         """
         Preps an item for insertion into the database.
 
@@ -839,11 +867,15 @@ async def prep_create_item(self, item: Item, base_url: str, exist_ok: bool = Fal
             index=index_alias_by_collection_id(item["collection"]),
             id=mk_item_id(item["id"], item["collection"]),
         ):
-            raise ConflictError(f"Item {item['id']} in collection {item['collection']} already exists")
+            raise ConflictError(
+                f"Item {item['id']} in collection {item['collection']} already exists"
+            )
 
         return self.item_serializer.stac_to_db(item, base_url)
 
-    def sync_prep_create_item(self, item: Item, base_url: str, exist_ok: bool = False) -> Item:
+    def sync_prep_create_item(
+        self, item: Item, base_url: str, exist_ok: bool = False
+    ) -> Item:
         """
         Prepare an item for insertion into the database.
 
@@ -872,7 +904,9 @@ def sync_prep_create_item(self, item: Item, base_url: str, exist_ok: bool = Fals
             index=index_alias_by_collection_id(collection_id),
             id=mk_item_id(item_id, collection_id),
         ):
-            raise ConflictError(f"Item {item_id} in collection {collection_id} already exists")
+            raise ConflictError(
+                f"Item {item_id} in collection {collection_id} already exists"
+            )
 
         return self.item_serializer.stac_to_db(item, base_url)
 
@@ -900,7 +934,9 @@ async def create_item(self, item: Item, refresh: bool = False):
         )
 
         if (meta := es_resp.get("meta")) and meta.get("status") == 409:
-            raise ConflictError(f"Item {item_id} in collection {collection_id} already exists")
+            raise ConflictError(
+                f"Item {item_id} in collection {collection_id} already exists"
+            )
 
     async def merge_patch_item(
         self,
@@ -1023,7 +1059,9 @@ async def json_patch_item(
 
         return item
 
-    async def delete_item(self, item_id: str, collection_id: str, refresh: bool = False):
+    async def delete_item(
+        self, item_id: str, collection_id: str, refresh: bool = False
+    ):
         """Delete a single item from the database.
 
         Args:
@@ -1041,7 +1079,9 @@ async def delete_item(self, item_id: str, collection_id: str, refresh: bool = Fa
                 refresh=refresh,
             )
         except exceptions.NotFoundError:
-            raise NotFoundError(f"Item {item_id} in collection {collection_id} not found")
+            raise NotFoundError(
+                f"Item {item_id} in collection {collection_id} not found"
+            )
 
     async def create_collection(self, collection: Collection, refresh: bool = False):
         """Create a single collection in the database.
@@ -1088,13 +1128,17 @@ async def find_collection(self, collection_id: str) -> Collection:
             collection as a `Collection` object. If the collection is not found, a `NotFoundError` is raised.
         """
         try:
-            collection = await self.client.get(index=COLLECTIONS_INDEX, id=collection_id)
+            collection = await self.client.get(
+                index=COLLECTIONS_INDEX, id=collection_id
+            )
         except exceptions.NotFoundError:
             raise NotFoundError(f"Collection {collection_id} not found")
 
         return collection["_source"]
 
-    async def update_collection(self, collection_id: str, collection: Collection, refresh: bool = False):
+    async def update_collection(
+        self, collection_id: str, collection: Collection, refresh: bool = False
+    ):
         """Update a collection from the database.
 
         Args:
@@ -1214,7 +1258,9 @@ async def json_patch_collection(
         if new_collection_id:
             collection["id"] = new_collection_id
             collection["links"] = resolve_links([], base_url)
-            await self.update_collection(collection_id=collection_id, collection=collection, refresh=False)
+            await self.update_collection(
+                collection_id=collection_id, collection=collection, refresh=False
+            )
 
         return collection
 
@@ -1235,10 +1281,14 @@ async def delete_collection(self, collection_id: str, refresh: bool = False):
             function also calls `delete_item_index` to delete the index for the items in the collection.
         """
         await self.find_collection(collection_id=collection_id)
-        await self.client.delete(index=COLLECTIONS_INDEX, id=collection_id, refresh=refresh)
+        await self.client.delete(
+            index=COLLECTIONS_INDEX, id=collection_id, refresh=refresh
+        )
         await delete_item_index(collection_id)
 
-    async def bulk_async(self, collection_id: str, processed_items: List[Item], refresh: bool = False) -> None:
+    async def bulk_async(
+        self, collection_id: str, processed_items: List[Item], refresh: bool = False
+    ) -> None:
         """Perform a bulk insert of items into the database asynchronously.
 
         Args:
@@ -1260,7 +1310,9 @@ async def bulk_async(self, collection_id: str, processed_items: List[Item], refr
             raise_on_error=False,
         )
 
-    def bulk_sync(self, collection_id: str, processed_items: List[Item], refresh: bool = False) -> None:
+    def bulk_sync(
+        self, collection_id: str, processed_items: List[Item], refresh: bool = False
+    ) -> None:
         """Perform a bulk insert of items into the database synchronously.
 
         Args: