feat: add POSTGRES_PORT to get_default_postgres_url (#181)

Update utils.py
datavisyn · Feb 16, 2024 · 6f315ba · 6f315ba
1 parent 953c74b
commit 6f315ba
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/visyn_core/settings/utils.py b/visyn_core/settings/utils.py
@@ -37,10 +37,11 @@ def get_default_postgres_url(
     password: str = "admin",
     host: str | None = os.getenv("POSTGRES_HOSTNAME"),
     host_fallback: str = "localhost",
-    port: int = 5432,
+    port: int | str | None = os.getenv("POSTGRES_PORT"),
+    port_fallback: int | str = 5432,
     database: str = "db",
 ) -> str:
     """
     Returns a default postgres url, including the default values for `driver`, `user`, `password`, `host`, `port` and `database`.
     """
-    return f"{driver}://{user}:{password}@{host or host_fallback}:{port}/{database}"
+    return f"{driver}://{user}:{password}@{host or host_fallback}:{port or port_fallback}/{database}"