从气流中获取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:
    [...]