fixing bug preventing usage of connection uri

CityofSantaMonica · Nov 14, 2018 · dc5b7e0 · dc5b7e0
1 parent a7045e9
commit dc5b7e0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mds/db/load.py b/mds/db/load.py
@@ -29,8 +29,8 @@ def data_engine(uri=None, **kwargs):
         backend = kwargs["backend"] if "backend" in kwargs else "postgresql"
         user, password, host, port, db = kwargs["user"], kwargs["password"], kwargs["host"], kwargs["port"], kwargs["db"]
         uri = f"{backend}://{user}:{password}@{host}:{port}/{db}"
-    else:
-        raise KeyError()
+    elif uri is None:
+        raise KeyError("Provide either `uri` or `user`, `password`, `host`, `port`, and `db`.")
 
     return sqlalchemy.create_engine(uri)