From 5a2d7c07bfb8b51959ebfcf1a204b72546bf5667 Mon Sep 17 00:00:00 2001 From: simon Date: Fri, 3 Dec 2021 13:51:34 +0700 Subject: [PATCH] variable expire value and simpler REDIS_PORT declaration --- tubearchivist/home/src/helper.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) 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):