diff --git a/abcd/__init__.py b/abcd/__init__.py index 081e868c..54358239 100644 --- a/abcd/__init__.py +++ b/abcd/__init__.py @@ -24,7 +24,7 @@ def from_url(cls, url, **kwargs): r = parse.urlparse(url) logger.info(r) - if r.scheme == "mongodb": + if ConnectionType[r.scheme] is ConnectionType.mongodb: conn_settings = { "host": r.hostname, "port": r.port, @@ -40,8 +40,7 @@ def from_url(cls, url, **kwargs): return MongoDatabase(db_name=db, **conn_settings, **kwargs) - r.scheme = ConnectionType[r.scheme] - if r.scheme is ConnectionType.opensearch: + if ConnectionType[r.scheme] is ConnectionType.opensearch: conn_settings = { "host": r.hostname, "port": r.port,