handle get single not found

pull/474/head
simon 2 years ago
parent 16440a4170
commit 2d2431364d
No known key found for this signature in database
GPG Key ID: 2C15AA5E89985DD4

@ -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)

Loading…
Cancel
Save