Generate Apache Airflow DAGs directly inside Visual Studio Code using an intuitive interactive GUI.
This extension helps you quickly create well-structured DAG Python files with support for 30+ Airflow operators, inline/file-based inputs, automatic downstream dependencies, and clean code formatting.
📦 Installation
From VS Code Marketplace
Open Visual Studio Code.
Go to the Extensions Marketplace (Ctrl+Shift+X or Cmd+Shift+X on macOS).
Search for Airflow DAG Generator.
Click Install.
From Command Line
code --install-extension AirflowDAGGenerator-KP.airflowDagGenerator
Usage
Open Visual Studio Code.
Press Ctrl + Shift + P (or Cmd + Shift + P on macOS) to open the Command Palette.
Search for: "Generate Airflow DAGs interactively from VS Code"
Search Or goto search bar in vs code type "> " and search " Generate Airflow DAGs interactively from VS Code"
Select the command to launch the DAG Generator.
Add tasks, configure operators, and export the DAG as a Python file.
## Features
Features
- Add tasks dynamically and configure multiple operators:
- BigQueryOperator: Inline SQL or external SQL file.
- BashOperator: Inline command or script file.
- SSHOperator: Remote command execution with timeouts.
- PythonOperator: Run Python functions.
- BranchPythonOperator: Conditional branching logic.
- GCSToGCSOperator: Transfer objects between GCS buckets.
- GCSToBigQueryOperator: Load GCS data to BigQuery.
- BigQueryToGCSOperator: Export BigQuery tables to GCS.
- EmailOperator: Send emails from the DAG.
- DagRunOperator: Trigger another DAG.
- DataflowPythonOperator: Run GCP Dataflow Python pipelines.
- DataflowTemplateOperator: Launch Dataflow jobs from templates.
- SqliteOperator, MySqlOperator, PostgresOperator, MsSqlOperator: Execute SQL queries.
- HiveOperator, PigOperator: Run Hive or Pig scripts.
- SparkSubmitOperator: Submit Spark jobs with config options.
- BashPythonOperator: Combine bash commands and Python code.
- SlackAPIPostOperator: Send messages to Slack channels.
- GCSFileSensor: Wait for files in GCS buckets.
- PythonSensor: Use Python callables as sensors.
- GCSToGCSComposeOperator: Compose multiple GCS files into one.
- S3ToGCSOperator: Transfer files from AWS S3 to GCS.
- DummyOperator: Placeholder task for DAG structure.
- Supports downstream dependencies automatically for tasks in order.
- Generates well-formatted DAG Python files with dag=dag included in all operators.
- Supports **downstream dependencies** automatically for tasks in order.
- Generates well-formatted DAG Python files with `dag=dag` included in all operators.
---
## Requirements
If you have any requirements or dependencies, add a section describing those and how to install and configure them.
## Extension Settings
Include if your extension adds any VS Code settings through the `contributes.configuration` extension point.
For example:
This extension contributes the following settings:
* `myExtension.enable`: Enable/disable this extension.
* `myExtension.thing`: Set to `blah` to do something.
## Known Issues
Calling out known issues can help limit users opening duplicate issues against your extension.
## Release Notes
Users appreciate release notes as you update your extension.
### 1.0.0
Initial release of ...
### 1.0.1
Fixed issue #.
### 1.1.0
Added features X, Y, and Z.
---