Enable asset indexing.

Author: rhysrevans3

stac_fastapi/elasticsearch/stac_fastapi/elasticsearch/database_logic.py

@@ -113,7 +113,7 @@
         "id": {"type": "keyword"},
         "collection": {"type": "keyword"},
         "geometry": {"type": "geo_shape"},
-        "assets": {"type": "object", "enabled": False},
+        "assets": {"type": "object"},
         "links": {"type": "object", "enabled": False},
         "properties": {
             "type": "object",
@@ -324,9 +324,7 @@ class DatabaseLogic:
     sync_client = SyncElasticsearchSettings().create_client
 
     item_serializer: Type[ItemSerializer] = attr.ib(default=ItemSerializer)
-    collection_serializer: Type[CollectionSerializer] = attr.ib(
-        default=CollectionSerializer
-    )
+    collection_serializer: Type[CollectionSerializer] = attr.ib(default=CollectionSerializer)
 
     extensions: List[str] = attr.ib(default=attr.Factory(list))
 
@@ -360,15 +358,9 @@ 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",
@@ -465,9 +457,7 @@ 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
@@ -497,16 +487,10 @@ 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
@@ -600,9 +584,7 @@ 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
 
@@ -715,11 +697,7 @@ 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")
@@ -799,9 +777,7 @@ 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.
 
@@ -823,15 +799,11 @@ async def prep_create_item(
             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.
 
@@ -860,9 +832,7 @@ def sync_prep_create_item(
             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)
 
@@ -890,13 +860,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 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:
@@ -914,9 +880,7 @@ async def delete_item(
                 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.
@@ -963,17 +927,13 @@ 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:
@@ -1035,14 +995,10 @@ 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:
@@ -1064,9 +1020,7 @@ async def bulk_async(
             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: