Sfoglia il codice sorgente

FIX badly written check if cluster_state is None

Thierry RAMORASOAVINA 8 anni fa
parent
commit
949a70d00f
1 ha cambiato i file con 8 aggiunte e 5 eliminazioni
  1. 8 5
      celery/events/state.py

+ 8 - 5
celery/events/state.py

@@ -293,11 +293,14 @@ class Task(object):
     def __init__(self, uuid=None, cluster_state=None, children=None, **kwargs):
         self.uuid = uuid
         self.cluster_state = cluster_state
-        self.children = WeakSet(
-            self.cluster_state.tasks.get(task_id)
-            for task_id in children or ()
-            if self.cluster_state is not None and task_id in self.cluster_state.tasks
-        )
+        if self.cluster_state is not None:
+            self.children = WeakSet(
+                self.cluster_state.tasks.get(task_id)
+                for task_id in children or ()
+                if task_id in self.cluster_state.tasks
+            )
+        else:
+            self.children = WeakSet()
         self._serializer_handlers = {
             'children': self._serializable_children,
             'root': self._serializable_root,