def _get_dep_statuses(self, ti, session, dep_context): """ Determines whether a task is ready to be rescheduled. Only tasks in NONE state with at least one row in task_reschedule table are handled by this dependency class, otherwise this dependency is considered as passed. This dependency fails if the latest reschedule request's reschedule date is still in future. """ is_mapped = ti.task.is_mapped if not is_mapped and not getattr(ti.task, "reschedule", False): # Mapped sensors don't have the reschedule property (it can only # be calculated after unmapping), so we don't check them here. # They are handled below by checking TaskReschedule instead. yield self._passing_status( reason="Task is not in reschedule mode.") return if dep_context.ignore_in_reschedule_period: yield self._passing_status( reason= "The context specified that being in a reschedule period was permitted." ) return if ti.state not in self.RESCHEDULEABLE_STATES: yield self._passing_status( reason= "The task instance is not in State_UP_FOR_RESCHEDULE or NONE state." ) return task_reschedule = (TaskReschedule.query_for_task_instance( task_instance=ti, descending=True, session=session).with_entities( TaskReschedule.reschedule_date).first()) if not task_reschedule: # Because mapped sensors don't have the reschedule property, here's the last resort # and we need a slightly different passing reason if is_mapped: yield self._passing_status( reason="The task is mapped and not in reschedule mode") return yield self._passing_status( reason="There is no reschedule request for this task instance." ) return now = timezone.utcnow() next_reschedule_date = task_reschedule.reschedule_date if now >= next_reschedule_date: yield self._passing_status( reason="Task instance id ready for reschedule.") return yield self._failing_status(reason=( "Task is not ready for reschedule yet but will be rescheduled automatically. " f"Current date is {now.isoformat()} and task will be " f"rescheduled at {next_reschedule_date.isoformat()}."))
def _get_dep_statuses(self, ti, session, dep_context): """ Determines whether a task is ready to be rescheduled. Only tasks in NONE state with at least one row in task_reschedule table are handled by this dependency class, otherwise this dependency is considered as passed. This dependency fails if the latest reschedule request's reschedule date is still in future. """ if not getattr(ti.task, "reschedule", False): yield self._passing_status(reason="Task is not in reschedule mode.") return if dep_context.ignore_in_reschedule_period: yield self._passing_status( reason="The context specified that being in a reschedule period was permitted." ) return if ti.state not in self.RESCHEDULEABLE_STATES: yield self._passing_status( reason="The task instance is not in State_UP_FOR_RESCHEDULE or NONE state." ) return task_reschedule = ( TaskReschedule.query_for_task_instance(task_instance=ti, descending=True, session=session) .with_entities(TaskReschedule.reschedule_date) .first() ) if not task_reschedule: yield self._passing_status(reason="There is no reschedule request for this task instance.") return now = timezone.utcnow() next_reschedule_date = task_reschedule.reschedule_date if now >= next_reschedule_date: yield self._passing_status(reason="Task instance id ready for reschedule.") return yield self._failing_status( reason=( "Task is not ready for reschedule yet but will be rescheduled automatically. " f"Current date is {now.isoformat()} and task will be " f"rescheduled at {next_reschedule_date.isoformat()}." ) )