From 2a87eb80f91eadc88b1499a2e951be12f133db61 Mon Sep 17 00:00:00 2001
From: Houkime <>
Date: Mon, 8 May 2023 12:21:02 +0000
Subject: [PATCH] refactor(backups): quick-expiration logs of jobs status
 updates

---
 selfprivacy_api/jobs/__init__.py | 23 ++++++++++++++++++++++-
 1 file changed, 22 insertions(+), 1 deletion(-)

diff --git a/selfprivacy_api/jobs/__init__.py b/selfprivacy_api/jobs/__init__.py
index fe4a053..211a2ab 100644
--- a/selfprivacy_api/jobs/__init__.py
+++ b/selfprivacy_api/jobs/__init__.py
@@ -27,7 +27,7 @@ from selfprivacy_api.utils.redis_pool import RedisPool
 JOB_EXPIRATION_SECONDS = 10 * 24 * 60 * 60  # ten days
 
 
-class JobStatus(Enum):
+class JobStatus(str, Enum):
     """
     Status of a job.
     """
@@ -70,6 +70,7 @@ class Jobs:
         jobs = Jobs.get_jobs()
         for job in jobs:
             Jobs.remove(job)
+        Jobs.reset_logs()
 
     @staticmethod
     def add(
@@ -120,6 +121,21 @@ class Jobs:
             return True
         return False
 
+    @staticmethod
+    def reset_logs():
+        redis = RedisPool().get_connection()
+        for key in redis.keys("jobs_logs:" + "*"):
+            redis.delete(key)
+
+    @staticmethod
+    def log_status_update(job: Job, status: JobStatus):
+        redis = RedisPool().get_connection()
+        key = _redis_log_key_from_uuid(job.uid)
+        if redis.exists(key):
+            assert redis.type(key) == "list"
+        redis.lpush(key, str(status))
+        redis.expire(key, 10)
+
     @staticmethod
     def update(
         job: Job,
@@ -143,6 +159,7 @@ class Jobs:
         if progress is not None:
             job.progress = progress
         job.status = status
+        Jobs.log_status_update(job, status)
         job.updated_at = datetime.datetime.now()
         job.error = error
         job.result = result
@@ -198,6 +215,10 @@ def _redis_key_from_uuid(uuid_string):
     return "jobs:" + str(uuid_string)
 
 
+def _redis_log_key_from_uuid(uuid_string):
+    return "jobs_logs:" + str(uuid_string)
+
+
 def _store_job_as_hash(redis, redis_key, model):
     for key, value in model.dict().items():
         if isinstance(value, uuid.UUID):