Construcg DB URL from parts

lsst-dm · Mar 2, 2024 · b14ecf2 · b14ecf2
1 parent 961da5e
commit b14ecf2
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/python/lsst/consdb/hinfo.py b/python/lsst/consdb/hinfo.py
@@ -251,9 +251,17 @@ def process_date(day_obs: str) -> None:
         translator = LsstCamTranslator
         instrument_mapping = LSSTCAM_MAPPING
 
-url = os.environ.get(
-    "POSTGRES_URL", "postgresql://usdf-butler.slac.stanford.edu:5432/lsstdb1"
-)
+host = os.environ.get("DB_HOST")
+passwd = os.environ.get("DB_PASS")
+user = os.environ.get("DB_USER")
+dbname = os.environ.get("DB_NAME")
+url = ""
+if host and passwd and user and dbname:
+    url = f"postgresql://{user}:{passwd}@{host}/{dbname}"
+else:
+    url = os.environ.get(
+        "POSTGRES_URL", "postgresql://usdf-butler.slac.stanford.edu:5432/lsstdb1"
+    )
 engine = create_engine(url)
 metadata_obj = MetaData(schema=f"cdb_{instrument.lower()}")
 exposure_table = Table("exposure", metadata_obj, autoload_with=engine)