diff --git a/abcd/__init__.py b/abcd/__init__.py index 54358239..70f8cdc9 100644 --- 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)