simon987/hexlib
1
0
Fork 0
mirror of https://github.com/simon987/hexlib.git synced 2025-04-19 17:56:43 +00:00
Code Issues Projects Releases Wiki Activity

load redis from env if not specified

Browse Source
This commit is contained in:
simon987 2021-02-25 21:35:18 -05:00
parent 9cadce62ac
commit ce2e5b2af6
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
hexlib/db.py
View File

@ -25,7 +25,9 @@ class PersistentState:
class VolatileState: class VolatileState:
"""Quick and dirty volatile dict-like redis wrapper""" """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.rdb = redis_db
self.prefix = prefix self.prefix = prefix
@ -54,7 +56,9 @@ class VolatileQueue:
class VolatileBooleanState: class VolatileBooleanState:
"""Quick and dirty volatile dict-like redis wrapper for boolean values""" """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.rdb = redis_db
self.prefix = prefix self.prefix = prefix