return command to task

2023-03-24 10:41:33 +07:00 · 2023-03-24 10:41:33 +07:00 · 2b8012b5d4
parent 81701df1c7
commit 2b8012b5d4
1 changed files with 4 additions and 0 deletions
--- 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