Skip to content

Commit 9d24983

Browse files
committed
feat: simplify password retrieval logic for pygeoapi configuration
1 parent 73f26bf commit 9d24983

1 file changed

Lines changed: 7 additions & 8 deletions

File tree

core/pygeoapi.py

Lines changed: 7 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -288,18 +288,17 @@ def _pygeoapi_db_settings() -> tuple[str, str, str, str, str]:
288288
"PYGEOAPI_POSTGRES_USER or POSTGRES_USER must be set and non-empty "
289289
"to generate the pygeoapi configuration."
290290
)
291-
password_value = os.environ.get("PYGEOAPI_POSTGRES_PASSWORD")
292-
if password_value is None:
293-
password_value = os.environ.get("POSTGRES_PASSWORD")
294-
if password_value is None:
291+
has_pygeoapi_password = os.environ.get("PYGEOAPI_POSTGRES_PASSWORD") is not None
292+
has_postgres_password = os.environ.get("POSTGRES_PASSWORD") is not None
293+
if not has_pygeoapi_password and not has_postgres_password:
295294
raise RuntimeError(
296295
"PYGEOAPI_POSTGRES_PASSWORD or POSTGRES_PASSWORD must be set to "
297296
"generate the pygeoapi configuration."
298297
)
299-
# Normalize to a dedicated runtime env var used by generated pygeoapi config.
300-
runtime_password_env_var = "PYGEOAPI_DB_AUTH_TOKEN"
301-
os.environ[runtime_password_env_var] = password_value
302-
return host, port, dbname, user, f"${{{runtime_password_env_var}}}"
298+
password_env_var = (
299+
"PYGEOAPI_POSTGRES_PASSWORD" if has_pygeoapi_password else "POSTGRES_PASSWORD"
300+
)
301+
return host, port, dbname, user, f"${{{password_env_var}}}"
303302

304303

305304
def _write_config(path: Path) -> None:

0 commit comments

Comments
 (0)