def _set_dep()

in src/pydolphinscheduler/tasks/condition.py [0:0]

• 8 lines of code
• 4 McCabe index (conditional complexity)

    def _set_dep(self) -> None:
        """Set upstream according to parameter `condition`."""
        upstream = []
        for cond in self.condition.args:
            if isinstance(cond, ConditionOperator):
                for status in cond.args:
                    upstream.extend(list(status.tasks))
        self.set_upstream(upstream)
        self.set_downstream([self.success_task, self.failed_task])