:mod:`airflow.operators.branch_operator`
========================================

.. py:module:: airflow.operators.branch_operator

.. autoapi-nested-parse::

   Branching operators


Module Contents
---------------

.. py:class:: BaseBranchOperator

   Bases: :class:`airflow.models.BaseOperator`, :class:`airflow.models.SkipMixin`

   This is a base class for creating operators with branching functionality,
   similarly to BranchPythonOperator.

   Users should subclass this operator and implement the function
   `choose_branch(self, context)`. This should run whatever business logic
   is needed to determine the branch, and return either the task_id for
   a single task (as a str) or a list of task_ids.

   The operator will continue with the returned task_id(s), and all other
   tasks directly downstream of this operator will be skipped.

   
   .. method:: choose_branch(self, context)

      Subclasses should implement this, running whatever logic is
      necessary to choose a branch and returning a task_id or list of
      task_ids.

      :param context: Context dictionary as passed to execute()
      :type context: dict


   .. method:: execute(self, context)