handle get single not found
This commit is contained in:
parent
16440a4170
commit
2d2431364d
|
@ -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…
Reference in New Issue