Attributes

Classes

brickflow_plugins.airflow.operators.native_operators.BashOperatorModifier

Bases: OperatorModifier

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

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

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