prep update collection

Author: jonhealy1

stac_fastapi/elasticsearch/stac_fastapi/elasticsearch/database_logic.py

@@ -283,3 +283,9 @@ def create_collection(self, collection: stac_types.Collection):
             document=collection,
         )
 
+    def prep_update_collection(self, collection_id: str):
+        try:
+            _ = self.client.get(index=COLLECTIONS_INDEX, id=collection_id)
+        except elasticsearch.exceptions.NotFoundError:
+            raise NotFoundError(f"Collection {collection_id} not found")
+

stac_fastapi/elasticsearch/stac_fastapi/elasticsearch/transactions.py

@@ -101,10 +101,8 @@ def update_collection(
     ) -> stac_types.Collection:
         """Update collection."""
         base_url = str(kwargs["request"].base_url)
-        try:
-            _ = self.client.get(index=COLLECTIONS_INDEX, id=collection["id"])
-        except elasticsearch.exceptions.NotFoundError:
-            raise NotFoundError(f"Collection {collection['id']} not found")
+        
+        self.database.prep_update_collection(collection_id=collection["id"])
         self.delete_collection(collection["id"])
         self.create_collection(collection, **kwargs)