refactor(ingestor-api): rename file & class to make project-agnostic (#24)

Author: alukach

lib/ingestor-api/runtime/src/collection.py

@@ -5,7 +5,7 @@
 
 from .schemas import StacCollection
 from .utils import get_db_credentials
-from .vedaloader import VEDALoader
+from .loader import Loader
 
 
 def ingest(collection: StacCollection):
@@ -17,7 +17,7 @@ def ingest(collection: StacCollection):
     try:
         creds = get_db_credentials(os.environ["DB_SECRET_ARN"])
         with PgstacDB(dsn=creds.dsn_string, debug=True) as db:
-            loader = VEDALoader(db=db)
+            loader = Loader(db=db)
             loader.load_collection(file=collection, insert_mode=Methods.upsert)
     except Exception as e:
         print(f"Encountered failure loading collection into pgSTAC: {e}")
@@ -29,5 +29,5 @@ def delete(collection_id: str):
     """
     creds = get_db_credentials(os.environ["DB_SECRET_ARN"])
     with PgstacDB(dsn=creds.dsn_string, debug=True) as db:
-        loader = VEDALoader(db=db)
+        loader = Loader(db=db)
         loader.delete_collection(collection_id)

lib/ingestor-api/runtime/src/vedaloader.py renamed to lib/ingestor-api/runtime/src/loader.py

@@ -1,12 +1,12 @@
 """Utilities to bulk load data into pgstac from json/ndjson."""
 import logging
 
-from pypgstac.load import Loader
+from pypgstac.load import Loader as BaseLoader
 
 logger = logging.getLogger(__name__)
 
 
-class VEDALoader(Loader):
+class Loader(BaseLoader):
     """Utilities for loading data and updating collection summaries/extents."""
 
     def __init__(self, db) -> None:

lib/ingestor-api/runtime/src/utils.py

@@ -9,7 +9,7 @@
 from pypgstac.load import Methods
 
 from .schemas import Ingestion
-from .vedaloader import VEDALoader
+from .loader import Loader
 
 
 class DbCreds(pydantic.BaseModel):
@@ -61,7 +61,7 @@ def load_items(creds: DbCreds, ingestions: Sequence[Ingestion]):
     Bulk insert STAC records into pgSTAC.
     """
     with PgstacDB(dsn=creds.dsn_string, debug=True) as db:
-        loader = VEDALoader(db=db)
+        loader = Loader(db=db)
 
         items = [
             # NOTE: Important to deserialize values to convert decimals to floats