Beispiel #1
0
def operator_to_dbnd_task(operator):
    from dbnd._core.current import get_task_by_task_id

    dbnd_task_id = operator_to_to_dbnd_task_id(operator)
    if dbnd_task_id:
        return get_task_by_task_id(dbnd_task_id)
    return None
Beispiel #2
0
    def source_task(self):
        # type: ()->Optional[Task]
        if self.source:
            from dbnd._core.current import get_task_by_task_id

            return get_task_by_task_id(self.source.task_id)
        return None
Beispiel #3
0
def try_operator_to_dbnd_task(operator):
    from airflow.models import BaseOperator
    from dbnd._core.current import get_task_by_task_id

    if isinstance(
            operator,
            BaseOperator):  # we are trying to support native Airflow Operators
        dbnd_task_id = operator_to_to_dbnd_task_id(operator)
        if dbnd_task_id:
            return get_task_by_task_id(dbnd_task_id)
    return None