Archives/tubearchivist
2
0
Fork 0
mirror of https://github.com/tubearchivist/tubearchivist synced 2024-11-02 09:41:07 +00:00
Code Issues Projects Releases Wiki Activity

variable expire value and simpler REDIS_PORT declaration

Browse Source
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
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
tubearchivist/home/src/helper.py
View File

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