collection serializer

Author: jonhealy1

stac_fastapi/core/stac_fastapi/core/core.py

@@ -204,7 +204,7 @@ async def all_collections(self, **kwargs) -> Collections:
         token = request.query_params.get("token")
 
         collections, next_token = await self.database.get_all_collections(
-            token=token, limit=limit
+            token=token, limit=limit, base_url=base_url
         )
 
         links = [
@@ -238,7 +238,9 @@ async def get_collection(self, collection_id: str, **kwargs) -> Collection:
         """
         base_url = str(kwargs["request"].base_url)
         collection = await self.database.find_collection(collection_id=collection_id)
-        return self.collection_serializer.db_to_stac(collection, base_url)
+        return self.collection_serializer.db_to_stac(
+            collection=collection, base_url=base_url
+        )
 
     async def item_collection(
         self,

stac_fastapi/elasticsearch/stac_fastapi/elasticsearch/database_logic.py

@@ -290,7 +290,7 @@ class DatabaseLogic:
     """CORE LOGIC"""
 
     async def get_all_collections(
-        self, token: Optional[str], limit: int
+        self, token: Optional[str], limit: int, base_url: str
     ) -> Tuple[List[Dict[str, Any]], Optional[str]]:
         """Retrieve a list of all collections from Elasticsearch, supporting pagination.
 
@@ -315,7 +315,12 @@ async def get_all_collections(
         )
 
         hits = response["hits"]["hits"]
-        collections = [hit["_source"] for hit in hits]
+        collections = [
+            self.collection_serializer.db_to_stac(
+                collection=hit["_source"], base_url=base_url
+            )
+            for hit in hits
+        ]
 
         next_token = None
         if len(hits) == limit:

stac_fastapi/opensearch/stac_fastapi/opensearch/database_logic.py

@@ -311,7 +311,7 @@ class DatabaseLogic:
     """CORE LOGIC"""
 
     async def get_all_collections(
-        self, token: Optional[str], limit: int
+        self, token: Optional[str], limit: int, base_url: str
     ) -> Tuple[List[Dict[str, Any]], Optional[str]]:
         """
         Retrieve a list of all collections from Opensearch, supporting pagination.
@@ -339,7 +339,12 @@ async def get_all_collections(
         )
 
         hits = response["hits"]["hits"]
-        collections = [hit["_source"] for hit in hits]
+        collections = [
+            self.collection_serializer.db_to_stac(
+                collection=hit["_source"], base_url=base_url
+            )
+            for hit in hits
+        ]
 
         next_token = None
         if len(hits) == limit: