Tidy setting db

ElliottKasoar · Jun 11, 2024 · d70fc3a · d70fc3a
1 parent f16ba00
commit d70fc3a
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/abcd/__init__.py b/abcd/__init__.py
@@ -24,6 +24,9 @@ def from_url(cls, url, **kwargs):
         r = parse.urlparse(url)
         logger.info(r)
 
+        db = r.path.split("/")[1] if r.path else None
+        db = db if db else "abcd"
+
         if ConnectionType[r.scheme] is ConnectionType.mongodb:
             conn_settings = {
                 "host": r.hostname,
@@ -33,9 +36,6 @@ def from_url(cls, url, **kwargs):
                 "authSource": "admin",
             }
 
-            db = r.path.split("/")[1] if r.path else None
-            db = db if db else "abcd"
-
             from abcd.backends.atoms_pymongo import MongoDatabase
 
             return MongoDatabase(db_name=db, **conn_settings, **kwargs)
@@ -48,9 +48,6 @@ def from_url(cls, url, **kwargs):
                 "password": r.password,
             }
 
-            db = r.path.split("/")[1] if r.path else None
-            db = db if db else "abcd"
-
             from abcd.backends.atoms_opensearch import OpenSearchDatabase
 
             return OpenSearchDatabase(db=db, **conn_settings, **kwargs)