Sirope can receive a customized redis object.

Baltasarq · Mar 24, 2022 · 5afb4c2 · 5afb4c2
1 parent 2646a07
commit 5afb4c2
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/sirope/sirope_main.py b/sirope/sirope_main.py
@@ -15,8 +15,14 @@
 class Sirope:
     OID_ID = "__oid__"
 
-    def __init__(self):
-        self._redis = redis.Redis()
+    def __init__(self, redis_obj: redis.Redis=None):
+        """Creates a Sirope object from a given Redis.
+            :param redis: A Redis object or None.
+        """
+        if not redis_obj:
+            self._redis = redis.Redis()
+        else:
+            self._redis = redis_obj
         self._indexes = SafeIndex.get(self._redis)
 
     def save(self, obj: object) -> OID: