airflow.models.skipmixin

Module Contents

airflow.models.skipmixin.XCOM_SKIPMIXIN_KEY = skipmixin_key[source]
airflow.models.skipmixin.XCOM_SKIPMIXIN_SKIPPED = skipped[source]
airflow.models.skipmixin.XCOM_SKIPMIXIN_FOLLOWED = followed[source]
class airflow.models.skipmixin.SkipMixin[source]

Bases: airflow.utils.log.logging_mixin.LoggingMixin

_set_state_to_skipped(self, dag_run, execution_date, tasks, session)[source]

Used internally to set state of task instances to skipped from the same dag run.

skip(self, dag_run, execution_date, tasks, session=None)[source]

Sets tasks instances to skipped from the same dag run.

If this instance has a task_id attribute, store the list of skipped task IDs to XCom so that NotPreviouslySkippedDep knows these tasks should be skipped when they are cleared.

Parameters
  • dag_run – the DagRun for which to set the tasks to skipped

  • execution_date – execution_date

  • tasks – tasks to skip (not task_ids)

  • session – db session to use

skip_all_except(self, ti, branch_task_ids)[source]

This method implements the logic for a branching operator; given a single task ID or list of task IDs to follow, this skips all other tasks immediately downstream of this operator.

branch_task_ids is stored to XCom so that NotPreviouslySkippedDep knows skipped tasks or newly added tasks should be skipped when they are cleared.