Auto-generated Diagrams from Airflow DAGs.
This project aims to easily visualise your Airflow DAGs on service level from providers like AWS, GCP, Azure, etc. via diagrams.
To install it from PyPI run:
pip install airflow-diagrams
To use this auto-generator just add the following two lines to your Airflow DAG (and run it):
from airflow_diagrams import generate_diagram_from_dag
generate_diagram_from_dag(dag=dag, diagram_file="diagram.py")
This will create a file called diagram.py
which contains the definition to create a diagram.
Run this file and you will get a rendered diagram.
A working example can be found in examples with the example_dag generating the diagram (rendered version).
It iterates through all operators in the DAG and decides based on a mapping which diagram node to display for each type of operator.
Contributions are welcome <3. Please go ahead and raise an issue if you have one or open a PR. Thank you. :)