Merge pull request #111 from Skrattoune/patch-8

Delegating set_parent_task to ProcessInfo.__init__

Author: jedie

huey_monitor/tqdm.py

@@ -68,6 +68,12 @@ def __init__(self,
             cumulate_progress=cumulate2parents,
         )
 
+        if parent_task_id:
+            TaskModel.objects.set_parent_task(
+                main_task_id=parent_task_id,
+                sub_task_id=task.id,
+            )
+            
         self.total_progress = 0
 
         logger.info('Init TaskModel %s', self)

huey_monitor_tests/test_app/tasks.py

@@ -87,15 +87,10 @@ def parallel_sub_task(task, parent_task_id, item_chunk, **info_kwargs):
     """
     Useless example: Just calculate the SHA256 hash from all files
     """
-    # Save relationship between the main and sub tasks:
-    TaskModel.objects.set_parent_task(
-        main_task_id=parent_task_id,
-        sub_task_id=task.id
-    )
-
     total_items = len(item_chunk)
 
-    # Init progress information of this sub task:
+    # Init progress information of this sub task
+    # and save relationship between the main and sub tasks:
     process_info = ProcessInfo(
         task, total=total_items, parent_task_id=parent_task_id, **info_kwargs
     )