def patch_airflow_context_vars(): """ used for tracking bash operators """ import airflow from airflow.utils import operator_helpers from dbnd_airflow.airflow_override.operator_helpers import context_to_airflow_vars from dbnd._core.utils.object_utils import patch_models if hasattr(airflow.utils.operator_helpers, "context_to_airflow_vars"): patches = [( airflow.utils.operator_helpers, "context_to_airflow_vars", context_to_airflow_vars, )] patch_models(patches)
def patch_airflow_context_vars(): """Used for tracking bash operators""" import airflow from dbnd._core.utils.object_utils import patch_models from dbnd_airflow.airflow_override.operator_helpers import context_to_airflow_vars import airflow.models.taskinstance # isort:skip modules_to_patch = [ airflow.utils.operator_helpers, airflow.models.taskinstance ] patches = [] for module in modules_to_patch: if hasattr(module, "context_to_airflow_vars"): patches.append( (module, "context_to_airflow_vars", context_to_airflow_vars)) patch_models(patches)
def patch_airflow_windows_support(): from dbnd._core.utils.object_utils import patch_models patch_models(get_windows_compatible_patches())
def patch_airflow_modules(): patch_models(get_airflow_patches())