diff --git a/tubearchivist/home/src/helper.py b/tubearchivist/home/src/helper.py index d45e934..faa5495 100644 --- a/tubearchivist/home/src/helper.py +++ b/tubearchivist/home/src/helper.py @@ -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):