handle get single not found

This commit is contained in:
simon 2023-03-08 13:27:37 +07:00
parent 16440a4170
commit 2d2431364d
No known key found for this signature in database
GPG Key ID: 2C15AA5E89985DD4
1 changed files with 9 additions and 2 deletions

View File

@ -173,8 +173,11 @@ class TaskRedis(RedisBase):
def get_single(self, task_id):
"""return content of single task"""
result = self.conn.execute_command("GET", self.BASE + task_id).decode()
return json.loads(result)
result = self.conn.execute_command("GET", self.BASE + task_id)
if not result:
return False
return json.loads(result.decode())
def set_key(self, task_id, message, expire=False):
"""set value for lock, initial or update"""
@ -191,6 +194,10 @@ class TaskRedis(RedisBase):
raise ValueError
message = self.get_single(task_id)
if not message:
print(f"{task_id} not found")
raise KeyError
message.update({"command": command})
self.set_key(task_id, message)