From 2b8012b5d4cbae6bdfd31fd224db9b6af6787592 Mon Sep 17 00:00:00 2001 From: simon Date: Fri, 24 Mar 2023 10:41:33 +0700 Subject: [PATCH] return command to task --- tubearchivist/home/tasks.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tubearchivist/home/tasks.py b/tubearchivist/home/tasks.py index 8a9fed83..8ef59251 100644 --- a/tubearchivist/home/tasks.py +++ b/tubearchivist/home/tasks.py @@ -147,6 +147,10 @@ class BaseTask(Task): task_id = self.request.id message = self.TASK_CONFIG.get(self.name) message.update({"level": level, "id": task_id}) + task_result = TaskManager().get_task(task_id) + if task_result: + message.update({"command": task_result.get("command", False)}) + key = f"message:{message.get('group')}:{task_id.split('-')[0]}" return message, key