Merge pull request #53 from alukach/feature/customize-settings

Support providing postgres settings via argument

CHANGES.md

@@ -3,6 +3,8 @@
 ## Unreleased
 
 * Insert mosaic metadata `min/max zoom` and `bounds` in tilejson
+* allow users the ability to optionally provide `PostgresSettings` to `connect_to_db()` function in the event that they want to customize how their DB credentials are populated (author @alukach, https://github.com/stac-utils/titiler-pgstac/pull/53)
+
 
 ## 0.1.0.a7 (2022-04-05) Pre-Release
 

tests/conftest.py

@@ -124,7 +124,7 @@ async def app():
 
             await conn.close()
 
-            await connect_to_db(app)
+            await connect_to_db(app, settings)
             async with AsyncClient(app=app, base_url="http://test") as client:
                 yield client
             await close_db_connection(app)

titiler/pgstac/db.py

@@ -1,16 +1,21 @@
 """Database connection handling."""
 
+from typing import Optional
+
 from psycopg_pool import ConnectionPool
 
 from titiler.pgstac.settings import PostgresSettings
 
 from fastapi import FastAPI
 
-settings = PostgresSettings()
-
 
-async def connect_to_db(app: FastAPI) -> None:
+async def connect_to_db(
+    app: FastAPI, settings: Optional[PostgresSettings] = None
+) -> None:
     """Connect to Database."""
+    if not settings:
+        settings = PostgresSettings()
+
     app.state.dbpool = ConnectionPool(
         conninfo=settings.connection_string,
         min_size=settings.db_min_conn_size,