Extract common structure into separate modules.

1 files changed, 13 insertions, 0 deletions

diff --git a/scripts/redis_logger.py b/scripts/redis_logger.py
index 97ffe26..76b8ba6 100644
--- a/scripts/redis_logger.py
+++ b/scripts/redis_logger.py
@@ -31,3 +31,16 @@ class RedisLogger(logging.Handler):
         self.redisconnection.rpush(
             self.messages_list_name(), self.format(record))
         self.redisconnection.expire(self.messages_list_name(), self.expiry)
+
+def setup_redis_logger(rconn: Redis,
+                       jobid: uuid.UUID,
+                       job_messagelist: str,
+                       expiry: int = 86400) -> RedisLogger:
+    """Setup a default RedisLogger logger."""
+    formatter = logging.Formatter(
+                "%(asctime)s - %(name)s - %(levelname)s: %(message)s")
+    # job_messagelist = f"{str(args.jobid)}:log-messages"
+    rconn.hset(name=str(jobid), key="log-messagelist", value=job_messagelist)
+    redislogger = RedisLogger(rconn, jobid, expiry=expiry)
+    redislogger.setFormatter(formatter)
+    return redislogger