variable expire value and simpler REDIS_PORT declaration

This commit is contained in:
simon 2021-12-03 13:51:34 +07:00
parent 3d0bb40da6
commit 5a2d7c07bf
No known key found for this signature in database
GPG Key ID: 2C15AA5E89985DD4

View File

@ -153,12 +153,9 @@ class RedisArchivist:
"""collection of methods to interact with redis"""
REDIS_HOST = os.environ.get("REDIS_HOST")
REDIS_PORT = os.environ.get("REDIS_PORT")
REDIS_PORT = os.environ.get("REDIS_PORT") or 6379
NAME_SPACE = "ta:"
if not REDIS_PORT:
REDIS_PORT = 6379
def __init__(self):
self.redis_connection = redis.Redis(
host=self.REDIS_HOST, port=self.REDIS_PORT
@ -171,8 +168,12 @@ class RedisArchivist:
)
if expire:
if isinstance(expire, bool):
secs = 20
else:
secs = expire
self.redis_connection.execute_command(
"EXPIRE", self.NAME_SPACE + key, 20
"EXPIRE", self.NAME_SPACE + key, secs
)
def get_message(self, key):