Fix connection type

ElliottKasoar · Jun 11, 2024 · 8411277 · 8411277
1 parent f17f950
commit 8411277
Showing 1 changed file with 2 additions and 3 deletions.
diff --git 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,