From 5afb4c23d7576ddd6081ecc52dd958981b9643d7 Mon Sep 17 00:00:00 2001 From: baltasarq Date: Thu, 24 Mar 2022 19:13:36 +0100 Subject: [PATCH] Sirope can receive a customized redis object. --- sirope/sirope_main.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/sirope/sirope_main.py b/sirope/sirope_main.py index 77a7561..7d87440 100644 --- 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: