Render 3rd party workflows in Airflow
Project description
Run your dbt Core projects as Apache Airflow DAGs and Task Groups with a few lines of code. Benefits include:
Run dbt projects against Airflow connections instead of dbt profiles
Native support for installing and running dbt in a virtual environment to avoid dependency conflicts with Airflow
Run tests immediately after a model is done to catch issues early
Utilize Airflow’s data-aware scheduling to run models immediately after upstream ingestion
Turn each dbt model into a task/task group complete with retries, alerting, etc.
Quickstart
Check out the Quickstart guide on our docs.
Example Usage
You can render an Airflow Task Group using the DbtTaskGroup class. Here’s an example with the jaffle_shop project:
from pendulum import datetime
from airflow import DAG
from airflow.operators.empty import EmptyOperator
from cosmos import DbtTaskGroup, ProfileConfig, ProjectConfig
from cosmos.profiles import PostgresUserPasswordProfileMapping
profile_config = ProfileConfig(
profile_name="default",
target_name="dev",
profile_mapping=PostgresUserPasswordProfileMapping(
conn_id="airflow_db",
profile_args={"schema": "public"},
),
)
with DAG(
dag_id="extract_dag",
start_date=datetime(2022, 11, 27),
schedule="@daily",
):
e1 = EmptyOperator(task_id="pre_dbt")
dbt_tg = DbtTaskGroup(
project_config=ProjectConfig("jaffle_shop"),
profile_config=profile_config,
)
e2 = EmptyOperator(task_id="post_dbt")
e1 >> dbt_tg >> e2This will generate an Airflow Task Group that looks like this:
Community
Join us on the Airflow Slack at #airflow-dbt
Changelog
We follow Semantic Versioning for releases. Check CHANGELOG.rst for the latest changes.
Contributing Guide
All contributions, bug reports, bug fixes, documentation improvements, enhancements are welcome.
A detailed overview an how to contribute can be found in the Contributing Guide.
As contributors and maintainers to this project, you are expected to abide by the Contributor Code of Conduct.
License
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
File details
Details for the file astronomer_cosmos-1.1.1.tar.gz.
File metadata
- Download URL: astronomer_cosmos-1.1.1.tar.gz
- Upload date:
- Size: 44.7 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/4.0.1 CPython/3.11.5
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | f39898e2a6a90dc42f7226bc442a9564adfdad3d4d4fdc301a90a3ff51dfa022 |
|
| MD5 | 57806762003772bab7aeff014146f276 |
|
| BLAKE2b-256 | cd56d2e09a5420fdf1c10316da1da34356a5ff7c3dffc3c6369706d1e945da8d |
File details
Details for the file astronomer_cosmos-1.1.1-py3-none-any.whl.
File metadata
- Download URL: astronomer_cosmos-1.1.1-py3-none-any.whl
- Upload date:
- Size: 67.3 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/4.0.1 CPython/3.11.5
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 745359d71f0017140d59ff41e127d339f8ac5b2d2345d3a0479a441b3cd80eb8 |
|
| MD5 | 30a9976ba0621b334db0daa8e1eef50b |
|
| BLAKE2b-256 | b16bf0af24b4d48842e4e780f4fb4cbfab5a9b46d5c4cd15e4d2439a02b98bf1 |
