From a1a0a52e4d1e1392ab247037f194d65208dd8aa0 Mon Sep 17 00:00:00 2001
From: Siarhei Siniak <siarheisiniak@gmail.com>
Date: Sun, 7 Jul 2024 12:11:38 +0300
Subject: [PATCH] [~] Refactor

---
 python/tasks/tiktok/tasks.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/python/tasks/tiktok/tasks.py b/python/tasks/tiktok/tasks.py
index 6cac43d..58cfbf7 100644
--- a/python/tasks/tiktok/tasks.py
+++ b/python/tasks/tiktok/tasks.py
@@ -469,6 +469,11 @@ def process_graph(
     return dict(
         ordered_nodes=ordered_nodes,
         done_ids=done_ids,
+        done_tasks={
+            k : celery.result.AsyncResult(task_ids[k])
+            for k in nodes
+            if task_ids.get(k) in done_ids
+        },
         task_ids=task_ids,
         error_ids=error_ids,
         pending_ids=pending_ids,