Skip to content

Attributes

Classes

brickflow_plugins.airflow.operators.native_operators.BashOperatorModifier()

Bases: OperatorModifier

Source code in brickflow_plugins/airflow/operators/__init__.py
def __init__(self):
    self._next_handler: Optional[AbstractOperatorModifier] = None

Functions

modify(operator: BashOperator, task: Task, workflow: Workflow) -> Optional[BashOperator]

Source code in brickflow_plugins/airflow/operators/native_operators.py
@check_if(BashOperator)
def modify(
    self, operator: BashOperator, task: Task, workflow: Workflow
) -> Optional["BashOperator"]:
    f = types.MethodType(_bash_execute, operator)
    operator.execute = f
    operator.on_kill = _bash_empty_on_kill
    return operator

brickflow_plugins.airflow.operators.native_operators.BranchPythonOperatorModifier()

Bases: OperatorModifier

Source code in brickflow_plugins/airflow/operators/__init__.py
def __init__(self):
    self._next_handler: Optional[AbstractOperatorModifier] = None

Functions

modify(operator: BranchPythonOperator, task: Task, workflow: Workflow) -> Optional[BranchPythonOperator]

Source code in brickflow_plugins/airflow/operators/native_operators.py
@check_if(BranchPythonOperator)
def modify(
    self, operator: BranchPythonOperator, task: Task, workflow: Workflow
) -> Optional["BranchPythonOperator"]:
    f = types.MethodType(_skip_all_except, operator)
    operator.skip_all_except = f
    return operator

brickflow_plugins.airflow.operators.native_operators.ShortCircuitOperatorModifier()

Bases: OperatorModifier

Source code in brickflow_plugins/airflow/operators/__init__.py
def __init__(self):
    self._next_handler: Optional[AbstractOperatorModifier] = None

Functions

modify(operator: ShortCircuitOperator, task: Task, workflow: Workflow) -> Optional[ShortCircuitOperator]

Source code in brickflow_plugins/airflow/operators/native_operators.py
@check_if(ShortCircuitOperator)
def modify(
    self, operator: ShortCircuitOperator, task: Task, workflow: Workflow
) -> Optional["ShortCircuitOperator"]:
    f = types.MethodType(_short_circuit_execute, operator)
    operator.execute = f
    return operator

Functions