Skip to content

tsvetkov-ii/airflow-declarative

 
 

Repository files navigation

airflow-declarative

Airflow declarative DAGs via YAML.

Compatibility:

  • Python 2.7 / 3.5+
  • Airflow 1.9+

Key Features

  • Declarative DAGs in plain text YAML helps a lot to understand how DAG will looks like. Made for humans, not programmers.
  • It makes extremely hard to turn your DAGs into code mess. Even if you make complicated YAMLs generator the result would be readable for humans.
  • No more guilty about coupling business logic with task management system (Airflow). They now could coexists separated.
  • Static analysis becomes a trivial task.
  • It's a good abstraction to create your own scheduler/worker compatible with original Airflow one.

Examples

Check tests/dags directory for example of DAGs which will works and which won't. Use src/airflow_declarative/schema.py module for the reference about YAML file schema. It should be self descriptive.

Don't be shy to experiment: trafaret-config will help you to understand what had gone wrong and why and where.

Usage

Upstream Airflow

To use with current (up to 1.8.2 release) upstream Airflow, you need to provide DAGs via Python file anyway. That should looks something like this:

import os

import airflow_declarative

# Assuming that the yaml dags are located in the same directory
# as this Python dag module:
ROOT = os.path.dirname(__file__)

DAGS = [
    airflow_declarative.from_path(os.path.join(ROOT, item))
    for item in os.listdir(ROOT)
    if item.endswith((".yml", ".yaml"))
]

globals().update({dag.dag_id: dag for dags in DAGS for dag in dags})

And place such file to AIRFLOW_HOME directory. Airflow will load dags in old fashion way.

Patched Airflow

Checkout patches directory for patches against Airflow release to have native declarative dags support on it. In this case no Python files are need on AIRFLOW_HOME path - just put there your YAMLs, they'll get loaded automagically.

Packages

No packages published

Languages

  • Python 95.0%
  • Makefile 5.0%