feat(release)!: upgrade stac-fastapi (#492)

Author: smohiudd

database/runtime/handler.py

@@ -129,21 +129,6 @@ def create_permissions(cursor, db_name: str, username: str) -> None:
     )
 
 
-def enable_context(cursor) -> None:
-    """Enable context extension for actual and estimated matches in item search and associated optimizations."""
-    cursor.execute(
-        sql.SQL(
-            "INSERT INTO pgstac.pgstac_settings (name, value) "
-            "   VALUES "
-            "       ('context', 'auto'),"
-            "       ('context_estimated_count', '100000'),"
-            "       ('context_estimated_cost', '100000'),"
-            "       ('context_stats_ttl', '1 day')"
-            "   ON CONFLICT ON CONSTRAINT pgstac_settings_pkey DO UPDATE SET value = excluded.value;"
-        )
-    )
-
-
 def register_extensions(cursor) -> None:
     """Add PostGIS extension."""
     cursor.execute(sql.SQL("CREATE EXTENSION IF NOT EXISTS postgis;"))
@@ -454,15 +439,6 @@ def handler(event, context):
                     username=user_params["username"],
                 )
 
-        with psycopg.connect(stac_db_conninfo, autocommit=True) as conn:
-            with conn.cursor() as cur:
-                print(
-                    "Enabling and configuring item search context in pgstac_settings..."
-                )
-                enable_context(
-                    cursor=cur,
-                )
-
         print("Adding mosaic index...")
         with psycopg.connect(
             stac_db_admin_dsn,

stac_api/runtime/setup.py

@@ -9,10 +9,10 @@
 
 inst_reqs = [
     "boto3",
-    "stac-fastapi.api~=3.0",
-    "stac-fastapi.types~=3.0",
-    "stac-fastapi.extensions~=3.0",
-    "stac-fastapi.pgstac~=3.0",
+    "stac-fastapi.api~=5.0",
+    "stac-fastapi.types~=5.0",
+    "stac-fastapi.extensions~=5.0",
+    "stac-fastapi.pgstac~=5.0",
     "jinja2>=2.11.2,<4.0.0",
     "starlette-cramjam>=0.3.2,<0.4",
     "importlib_resources>=1.1.0;python_version<='3.11'",  # https://github.com/cogeotiff/rio-tiler/pull/379

stac_api/runtime/src/app.py

@@ -45,7 +45,7 @@
 @asynccontextmanager
 async def lifespan(app: FastAPI):
     """Get a database connection on startup, close it on shutdown."""
-    await connect_to_db(app)
+    await connect_to_db(app, postgres_settings=api_settings.postgres_settings)
     yield
     await close_db_connection(app)
 
@@ -70,9 +70,9 @@ async def lifespan(app: FastAPI):
     ),
     title=f"{api_settings.project_name} STAC API",
     description=api_settings.project_description,
-    settings=api_settings.load_postgres_settings(),
+    settings=api_settings,
     extensions=PgStacExtensions,
-    client=VedaCrudClient(post_request_model=POSTModel),
+    client=VedaCrudClient(pgstac_search_model=POSTModel),
     search_get_request_model=GETModel,
     search_post_request_model=POSTModel,
     items_get_request_model=items_get_request_model,

stac_api/runtime/src/config.py

@@ -28,7 +28,7 @@
     TransactionExtension,
 )
 from stac_fastapi.extensions.third_party import BulkTransactionExtension
-from stac_fastapi.pgstac.config import Settings
+from stac_fastapi.pgstac.config import PostgresSettings, Settings
 from stac_fastapi.pgstac.transactions import BulkTransactionsClient, TransactionsClient
 from stac_fastapi.pgstac.types.search import PgstacSearch
 
@@ -57,7 +57,7 @@ def get_secret_dict(secret_name: str):
         return json.loads(base64.b64decode(get_secret_value_response["SecretBinary"]))
 
 
-class _ApiSettings(BaseSettings):
+class _ApiSettings(Settings):
     """API settings"""
 
     project_name: Optional[str] = "veda"
@@ -82,22 +82,20 @@ def parse_cors_origin(cls, v):
         """Parse CORS origins."""
         return [origin.strip() for origin in v.split(",")]
 
-    def load_postgres_settings(self) -> "Settings":
-        """Load postgres connection params from AWS secret"""
-
+    @property
+    def postgres_settings(self) -> PostgresSettings:
+        """Load postgres connection params from AWS secret."""
         if self.pgstac_secret_arn:
             secret = get_secret_dict(self.pgstac_secret_arn)
-
-            return Settings(
+            return PostgresSettings(
                 postgres_host_reader=secret["host"],
                 postgres_host_writer=secret["host"],
                 postgres_dbname=secret["dbname"],
                 postgres_user=secret["username"],
                 postgres_pass=secret["password"],
                 postgres_port=secret["port"],
             )
-        else:
-            return Settings()
+        return PostgresSettings()
 
     model_config = SettingsConfigDict(
         env_file=".env", env_prefix="VEDA_STAC_", extra="ignore"
@@ -159,7 +157,7 @@ def TilesApiSettings() -> _TilesApiSettings:
             BulkTransactionExtension(client=BulkTransactionsClient()),
             TransactionExtension(
                 client=TransactionsClient(),
-                settings=ApiSettings().load_postgres_settings(),
+                settings=api_settings,
                 response_class=ORJSONResponse,
             ),
         ]