airflow - 从 Airflow 中获取 DAG

Question

我正在尝试将 DAG 呈现为文档树，是否有直接的方法来获取它？

现在我正在使用(部分代码)手动生成 DOT 文件:

for task in dag.tasks:
    print("\t%s;" % task.task_id)
    relatives = [r.task_id for r in task.get_direct_relatives()]
    for r in relatives:
        print("\t%s -- %s;" % (task.task_id, r))

这可行，但我需要从外部动态导入所有 DAG..

Answer 1

您可以使用 airflow.models.DagBag 对象来枚举 DAG 对象。

from airflow.models import DagBag

for dag in DagBag().dags.values():
  for task in dag.tasks:
    [...]