airflow-config


Nameairflow-config JSON
Version 1.10.10 PyPI version JSON
download
home_pageNone
SummaryAirflow utilities for configuration of many DAGs and DAG environments
upload_time2025-07-11 23:18:35
maintainerNone
docs_urlNone
authorNone
requires_python>=3.9
licenseApache-2.0
keywords airflow config scheduler
VCS
bugtrack_url
requirements No requirements were recorded.
Travis-CI No Travis.
coveralls test coverage No coveralls.
            # airflow-config

[Apache Airflow](https://airflow.apache.org) utilities for configuration of DAGs and DAG environments

[![Build Status](https://github.com/airflow-laminar/airflow-config/actions/workflows/build.yaml/badge.svg?branch=main&event=push)](https://github.com/airflow-laminar/airflow-config/actions/workflows/build.yaml)
[![codecov](https://codecov.io/gh/airflow-laminar/airflow-config/branch/main/graph/badge.svg)](https://codecov.io/gh/airflow-laminar/airflow-config)
[![License](https://img.shields.io/github/license/airflow-laminar/airflow-config)](https://github.com/airflow-laminar/airflow-config)
[![PyPI](https://img.shields.io/pypi/v/airflow-config.svg)](https://pypi.python.org/pypi/airflow-config)

## Overview

This library allows for `YAML`-driven configuration of Airflow, including DAGs, Operators, and declaratively defined DAGs (à la [dag-factory](https://github.com/astronomer/dag-factory)).
It is built with [Pydantic](https://pydantic.dev), [Hydra](https://hydra.cc), and [OmegaConf](https://omegaconf.readthedocs.io/).

Consider the following basic DAG:

```python
from airflow import DAG
from airflow.providers.standard.operators.bash import BashOperator
from datetime import datetime, timedelta

with DAG(
    dag_id="test-dag",
    default_args={
        "depends_on_past": False,
        "email": ["my.email@myemail.com"],
        "email_on_failure": False,
        "email_on_retry": False,
        "retries": 0,
    },
    description="test that dag is working properly",
    schedule=timedelta(minutes=1),
    start_date=datetime(2024, 1, 1),
    catchup=False,
    tags=["utility", "test"],
):
    BashOperator(
        task_id="test-task",
        bash_command="echo 'test'",
    )
```

We can already see many options that we might want to drive centrally via config, perhaps based on some notion of environment (e.g. `dev`, `prod`, etc).

- `"email": ["my.email@myemail.com"]`
- `"email_on_failure": False`
- `"email_on_retry": False`
- `"retries": 0`
- `schedule=timedelta(minutes=1)`
- `tags=["utility", "test"]`

If we want to change these in our DAG, we need to modify code. Now imagine we have hundreds of DAGs, this can quickly get out of hand, especially since Airflow DAGs are Python code, and we might easily inject a syntax error or a trailing comma or other common problem.

Now consider the alternative, config-driven approach:

`config/dev.yaml`

```yaml
# @package _global_
_target_: airflow_config.Configuration
default_args:
  _target_: airflow_config.TaskArgs
  owner: test
  email: [myemail@myemail.com]
  email_on_failure: false
  email_on_retry: false
  retries: 0
  depends_on_past: false
default_dag_args:
  _target_: airflow_config.DagArgs
  schedule: "01:00"
  start_date: "2024-01-01"
  catchup: false
  tags: ["utility", "test"]
```

```python
from airflow.providers.standard.operators.bash import BashOperator
from airflow_config import DAG, load_config

config = load_config(config_name="dev")

with DAG(
    dag_id="test-dag",
    description="test that dag is working properly",
    schedule=timedelta(minutes=1),
    config=config
):
    BashOperator(
        task_id="test-task",
        bash_command="echo 'test'",
    )
```

This has a number of benefits:

- Make changes without code changes, with static type validation
- Make changes across any number of DAGs without having to copy-paste
- Organize collections of DAGs into groups, e.g. via enviroment like `dev`, `prod`, etc

## Features

- Configure DAGs from a central config file or...
- from multiple env-specific config files (e.g. `dev`, `uat`, `prod`)
- Specialize DAGs by `dag_id` from a single file (e.g. set each DAG's `schedule` from a single shared file)
- Generate entire DAGs declaratively, like [astronomer/dag-factory](https://github.com/astronomer/dag-factory)
- Configure other extensions like:
  - [airflow-priority](https://github.com/airflow-laminar/airflow-priority)
  - [airflow-balancer](https://github.com/airflow-laminar/airflow-balancer)
  - [airflow-supervisor](https://github.com/airflow-laminar/airflow-supervisor)
  - or write your own pydantic-based model and get yaml-based configuration for free


## Configuration

```python
class Configuration(BaseModel):
    # default task args
    # https://airflow.apache.org/docs/apache-airflow/stable/_api/airflow/models/baseoperator/index.html#airflow.models.baseoperator.BaseOperator
    default_task_args: TaskArgs

    # default dag args
    # https://airflow.apache.org/docs/apache-airflow/stable/_api/airflow/models/dag/index.html#airflow.models.dag.DAG
    default_dag_args: DagArgs

    # string (dag id) to Dag mapping
    dags: Optional[Dict[str, Dag]]

    # string (dag id) to Task mapping
    tasks: Optional[Dict[str, Task]]

    # used for extensions to inject arbitrary configuration.
    # See e.g.: https://github.com/airflow-laminar/airflow-supervisor?tab=readme-ov-file#example-dag-airflow-config
    extensions: Optional[Dict[str, BaseModel]]
```

Here is an example configuration defined via yaml:

```yaml
# @package _global_
_target_: airflow_config.Configuration
default_task_args:
  _target_: airflow_config.TaskArgs
  owner: blerg
  email: []
  email_on_failure: false
  email_on_retry: false
  retries: 0
  depends_on_past: false

default_dag_args:
  _target_: airflow_config.DagArgs
  start_date: ["2025-01-01", "America/New_York"]
  catchup: false
  max_active_runs: 1

dags:
  reboot:
    tags: ["reboot", "utility"]
    description: "Reboot machines"
    schedule: "0 0 * * *"
    max_active_tasks: 1
  clean-logs:
    tags: ["celery", "utility"]
    description: "Clean worker logs"
    schedule: "0 4 * * *"
```


## Examples

- [Basic 1](https://airflow-laminar.github.io/airflow-config/docs/src/examples.html#load-defaults-from-config)
- [Basic 2](https://airflow-laminar.github.io/airflow-config/docs/src/examples.html#load-more-defaults-from-config)
- [Specialize DAGs](https://airflow-laminar.github.io/airflow-config/docs/src/examples.html#specialize-individual-dags)
- [Declarative DAGs (DAG Factory)](https://airflow-laminar.github.io/airflow-config/docs/src/examples.html#declarative-dags-dag-factory)
- [Test Suite Setups](https://github.com/airflow-laminar/airflow-config/tree/main/airflow_config/tests/setups)


## License

This software is licensed under the Apache 2.0 license. See the [LICENSE](LICENSE) file for details.

            

Raw data

            {
    "_id": null,
    "home_page": null,
    "name": "airflow-config",
    "maintainer": null,
    "docs_url": null,
    "requires_python": ">=3.9",
    "maintainer_email": null,
    "keywords": "airflow, config, scheduler",
    "author": null,
    "author_email": "the airflow-config authors <t.paine154@gmail.com>",
    "download_url": "https://files.pythonhosted.org/packages/17/3d/02884ff294d64bcc63d58864064ebc81bffdc27173b150fbe70454f98c9c/airflow_config-1.10.10.tar.gz",
    "platform": null,
    "description": "# airflow-config\n\n[Apache Airflow](https://airflow.apache.org) utilities for configuration of DAGs and DAG environments\n\n[![Build Status](https://github.com/airflow-laminar/airflow-config/actions/workflows/build.yaml/badge.svg?branch=main&event=push)](https://github.com/airflow-laminar/airflow-config/actions/workflows/build.yaml)\n[![codecov](https://codecov.io/gh/airflow-laminar/airflow-config/branch/main/graph/badge.svg)](https://codecov.io/gh/airflow-laminar/airflow-config)\n[![License](https://img.shields.io/github/license/airflow-laminar/airflow-config)](https://github.com/airflow-laminar/airflow-config)\n[![PyPI](https://img.shields.io/pypi/v/airflow-config.svg)](https://pypi.python.org/pypi/airflow-config)\n\n## Overview\n\nThis library allows for `YAML`-driven configuration of Airflow, including DAGs, Operators, and declaratively defined DAGs (\u00e0 la [dag-factory](https://github.com/astronomer/dag-factory)).\nIt is built with [Pydantic](https://pydantic.dev), [Hydra](https://hydra.cc), and [OmegaConf](https://omegaconf.readthedocs.io/).\n\nConsider the following basic DAG:\n\n```python\nfrom airflow import DAG\nfrom airflow.providers.standard.operators.bash import BashOperator\nfrom datetime import datetime, timedelta\n\nwith DAG(\n    dag_id=\"test-dag\",\n    default_args={\n        \"depends_on_past\": False,\n        \"email\": [\"my.email@myemail.com\"],\n        \"email_on_failure\": False,\n        \"email_on_retry\": False,\n        \"retries\": 0,\n    },\n    description=\"test that dag is working properly\",\n    schedule=timedelta(minutes=1),\n    start_date=datetime(2024, 1, 1),\n    catchup=False,\n    tags=[\"utility\", \"test\"],\n):\n    BashOperator(\n        task_id=\"test-task\",\n        bash_command=\"echo 'test'\",\n    )\n```\n\nWe can already see many options that we might want to drive centrally via config, perhaps based on some notion of environment (e.g. `dev`, `prod`, etc).\n\n- `\"email\": [\"my.email@myemail.com\"]`\n- `\"email_on_failure\": False`\n- `\"email_on_retry\": False`\n- `\"retries\": 0`\n- `schedule=timedelta(minutes=1)`\n- `tags=[\"utility\", \"test\"]`\n\nIf we want to change these in our DAG, we need to modify code. Now imagine we have hundreds of DAGs, this can quickly get out of hand, especially since Airflow DAGs are Python code, and we might easily inject a syntax error or a trailing comma or other common problem.\n\nNow consider the alternative, config-driven approach:\n\n`config/dev.yaml`\n\n```yaml\n# @package _global_\n_target_: airflow_config.Configuration\ndefault_args:\n  _target_: airflow_config.TaskArgs\n  owner: test\n  email: [myemail@myemail.com]\n  email_on_failure: false\n  email_on_retry: false\n  retries: 0\n  depends_on_past: false\ndefault_dag_args:\n  _target_: airflow_config.DagArgs\n  schedule: \"01:00\"\n  start_date: \"2024-01-01\"\n  catchup: false\n  tags: [\"utility\", \"test\"]\n```\n\n```python\nfrom airflow.providers.standard.operators.bash import BashOperator\nfrom airflow_config import DAG, load_config\n\nconfig = load_config(config_name=\"dev\")\n\nwith DAG(\n    dag_id=\"test-dag\",\n    description=\"test that dag is working properly\",\n    schedule=timedelta(minutes=1),\n    config=config\n):\n    BashOperator(\n        task_id=\"test-task\",\n        bash_command=\"echo 'test'\",\n    )\n```\n\nThis has a number of benefits:\n\n- Make changes without code changes, with static type validation\n- Make changes across any number of DAGs without having to copy-paste\n- Organize collections of DAGs into groups, e.g. via enviroment like `dev`, `prod`, etc\n\n## Features\n\n- Configure DAGs from a central config file or...\n- from multiple env-specific config files (e.g. `dev`, `uat`, `prod`)\n- Specialize DAGs by `dag_id` from a single file (e.g. set each DAG's `schedule` from a single shared file)\n- Generate entire DAGs declaratively, like [astronomer/dag-factory](https://github.com/astronomer/dag-factory)\n- Configure other extensions like:\n  - [airflow-priority](https://github.com/airflow-laminar/airflow-priority)\n  - [airflow-balancer](https://github.com/airflow-laminar/airflow-balancer)\n  - [airflow-supervisor](https://github.com/airflow-laminar/airflow-supervisor)\n  - or write your own pydantic-based model and get yaml-based configuration for free\n\n\n## Configuration\n\n```python\nclass Configuration(BaseModel):\n    # default task args\n    # https://airflow.apache.org/docs/apache-airflow/stable/_api/airflow/models/baseoperator/index.html#airflow.models.baseoperator.BaseOperator\n    default_task_args: TaskArgs\n\n    # default dag args\n    # https://airflow.apache.org/docs/apache-airflow/stable/_api/airflow/models/dag/index.html#airflow.models.dag.DAG\n    default_dag_args: DagArgs\n\n    # string (dag id) to Dag mapping\n    dags: Optional[Dict[str, Dag]]\n\n    # string (dag id) to Task mapping\n    tasks: Optional[Dict[str, Task]]\n\n    # used for extensions to inject arbitrary configuration.\n    # See e.g.: https://github.com/airflow-laminar/airflow-supervisor?tab=readme-ov-file#example-dag-airflow-config\n    extensions: Optional[Dict[str, BaseModel]]\n```\n\nHere is an example configuration defined via yaml:\n\n```yaml\n# @package _global_\n_target_: airflow_config.Configuration\ndefault_task_args:\n  _target_: airflow_config.TaskArgs\n  owner: blerg\n  email: []\n  email_on_failure: false\n  email_on_retry: false\n  retries: 0\n  depends_on_past: false\n\ndefault_dag_args:\n  _target_: airflow_config.DagArgs\n  start_date: [\"2025-01-01\", \"America/New_York\"]\n  catchup: false\n  max_active_runs: 1\n\ndags:\n  reboot:\n    tags: [\"reboot\", \"utility\"]\n    description: \"Reboot machines\"\n    schedule: \"0 0 * * *\"\n    max_active_tasks: 1\n  clean-logs:\n    tags: [\"celery\", \"utility\"]\n    description: \"Clean worker logs\"\n    schedule: \"0 4 * * *\"\n```\n\n\n## Examples\n\n- [Basic 1](https://airflow-laminar.github.io/airflow-config/docs/src/examples.html#load-defaults-from-config)\n- [Basic 2](https://airflow-laminar.github.io/airflow-config/docs/src/examples.html#load-more-defaults-from-config)\n- [Specialize DAGs](https://airflow-laminar.github.io/airflow-config/docs/src/examples.html#specialize-individual-dags)\n- [Declarative DAGs (DAG Factory)](https://airflow-laminar.github.io/airflow-config/docs/src/examples.html#declarative-dags-dag-factory)\n- [Test Suite Setups](https://github.com/airflow-laminar/airflow-config/tree/main/airflow_config/tests/setups)\n\n\n## License\n\nThis software is licensed under the Apache 2.0 license. See the [LICENSE](LICENSE) file for details.\n",
    "bugtrack_url": null,
    "license": "Apache-2.0",
    "summary": "Airflow utilities for configuration of many DAGs and DAG environments",
    "version": "1.10.10",
    "project_urls": {
        "Homepage": "https://github.com/airflow-laminar/airflow-config",
        "Repository": "https://github.com/airflow-laminar/airflow-config"
    },
    "split_keywords": [
        "airflow",
        " config",
        " scheduler"
    ],
    "urls": [
        {
            "comment_text": null,
            "digests": {
                "blake2b_256": "b5629130b16b9694e074fe499dd7bde9914993ba1c86d135ec43609ccac5cc56",
                "md5": "c289526f73ada67277788f87178575f1",
                "sha256": "dfc9beed5b6741e6247596299c1faaf50a080ce055f2b8a29b948c8d227870df"
            },
            "downloads": -1,
            "filename": "airflow_config-1.10.10-py3-none-any.whl",
            "has_sig": false,
            "md5_digest": "c289526f73ada67277788f87178575f1",
            "packagetype": "bdist_wheel",
            "python_version": "py3",
            "requires_python": ">=3.9",
            "size": 186129,
            "upload_time": "2025-07-11T23:18:33",
            "upload_time_iso_8601": "2025-07-11T23:18:33.894530Z",
            "url": "https://files.pythonhosted.org/packages/b5/62/9130b16b9694e074fe499dd7bde9914993ba1c86d135ec43609ccac5cc56/airflow_config-1.10.10-py3-none-any.whl",
            "yanked": false,
            "yanked_reason": null
        },
        {
            "comment_text": null,
            "digests": {
                "blake2b_256": "173d02884ff294d64bcc63d58864064ebc81bffdc27173b150fbe70454f98c9c",
                "md5": "629ae6e2b86aeb4ed09a4c3cacf0ef39",
                "sha256": "250869af21db1714879f95a32d6978e51adc3270069c6f8c4ba92013352aece9"
            },
            "downloads": -1,
            "filename": "airflow_config-1.10.10.tar.gz",
            "has_sig": false,
            "md5_digest": "629ae6e2b86aeb4ed09a4c3cacf0ef39",
            "packagetype": "sdist",
            "python_version": "source",
            "requires_python": ">=3.9",
            "size": 191062,
            "upload_time": "2025-07-11T23:18:35",
            "upload_time_iso_8601": "2025-07-11T23:18:35.165646Z",
            "url": "https://files.pythonhosted.org/packages/17/3d/02884ff294d64bcc63d58864064ebc81bffdc27173b150fbe70454f98c9c/airflow_config-1.10.10.tar.gz",
            "yanked": false,
            "yanked_reason": null
        }
    ],
    "upload_time": "2025-07-11 23:18:35",
    "github": true,
    "gitlab": false,
    "bitbucket": false,
    "codeberg": false,
    "github_user": "airflow-laminar",
    "github_project": "airflow-config",
    "travis_ci": false,
    "coveralls": false,
    "github_actions": true,
    "lcname": "airflow-config"
}
        
Elapsed time: 1.46271s