Name the arguments to MongoWrapper and RedisWrapper

Signed-off-by: Ivan Kanakarakis <[email protected]>
IdentityPython · Jul 12, 2021 · 1c82c47 · 1c82c47
1 parent 0fc2e45
commit 1c82c47
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/pyop/storage.py b/src/pyop/storage.py
@@ -54,9 +54,11 @@ def pop(self, key, default=None):
     @classmethod
     def from_uri(cls, db_uri, collection, db_name=None, ttl=None):
         if db_uri.startswith("mongodb"):
-            return MongoWrapper(db_uri, db_name, collection, ttl)
-        if db_uri.startswith("redis") or db_uri.startswith("unix"):
-            return RedisWrapper(db_uri, collection, ttl)
+            return MongoWrapper(
+                db_uri=db_uri, db_name=db_name, collection=collection, ttl=ttl
+            )
+        elif db_uri.startswith("redis") or db_uri.startswith("unix"):
+            return RedisWrapper(db_uri=db_uri, collection=collection, ttl=ttl)
 
         return ValueError(f"Invalid DB URI: {db_uri}")