From ce2e5b2af6abe7ce88418333772b871c50b700da Mon Sep 17 00:00:00 2001 From: simon987 Date: Thu, 25 Feb 2021 21:35:18 -0500 Subject: [PATCH] load redis from env if not specified --- hexlib/db.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/hexlib/db.py b/hexlib/db.py index 075cb09..040d59c 100644 --- a/hexlib/db.py +++ b/hexlib/db.py @@ -25,7 +25,9 @@ class PersistentState: class VolatileState: """Quick and dirty volatile dict-like redis wrapper""" - def __init__(self, prefix, redis_db): + def __init__(self, prefix, redis_db=None): + if redis_db is None: + redis_db = get_redis() self.rdb = redis_db self.prefix = prefix @@ -54,7 +56,9 @@ class VolatileQueue: class VolatileBooleanState: """Quick and dirty volatile dict-like redis wrapper for boolean values""" - def __init__(self, prefix, redis_db): + def __init__(self, prefix, redis_db=None): + if redis_db is None: + redis_db = get_redis() self.rdb = redis_db self.prefix = prefix