airflow-toolkit 2.0.1__tar.gz → 2.2.0__tar.gz

{airflow_toolkit-2.0.1 → airflow_toolkit-2.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: airflow-toolkit
-Version: 2.0.1
+Version: 2.2.0
 Summary: A toolkit of operators, hooks and utilities for Apache Airflow 3
 Author-email: Biel Llobera <biel_llobera@dkl.digital>
 Requires-Python: <3.15,>=3.11
@@ -23,8 +23,6 @@ Provides-Extra: azure
 Requires-Dist: apache-airflow-providers-microsoft-azure>=8; extra == "azure"
 Provides-Extra: sftp
 Requires-Dist: apache-airflow-providers-sftp>=5.3; extra == "sftp"
-Provides-Extra: slack
-Requires-Dist: apache-airflow-providers-slack>=9.3; extra == "slack"
 Provides-Extra: http
 Requires-Dist: apache-airflow-providers-http>=5.6.4; extra == "http"
 Requires-Dist: requests>=2.31.0; extra == "http"
@@ -46,7 +44,6 @@ Requires-Dist: apache-airflow-providers-amazon>=9.15.0; extra == "airflow3-full"
 Requires-Dist: apache-airflow-providers-google>=18; extra == "airflow3-full"
 Requires-Dist: apache-airflow-providers-microsoft-azure>=8; extra == "airflow3-full"
 Requires-Dist: apache-airflow-providers-sftp>=5.3; extra == "airflow3-full"
-Requires-Dist: apache-airflow-providers-slack>=9.3; extra == "airflow3-full"
 Requires-Dist: apache-airflow-providers-http>=5.6.4; extra == "airflow3-full"
 Requires-Dist: requests>=2.31.0; extra == "airflow3-full"
 Requires-Dist: jmespath<2,>=1.0.1; extra == "airflow3-full"
@@ -437,74 +434,99 @@ Because `AzureDatabricksSqlHook` implements `DbApiHook`, it can be used as `sour
 
 ## Notifications
 
-### Slack (incoming webhook)
+Send rich failure notifications to Slack, email, Microsoft Teams, and Discord from a single call. The notification system is built around three ideas:
 
-Send DAG or task failure alerts to a Slack channel using `dag_failure_slack_notification_webhook`. Requires a Slack App with Incoming Webhooks enabled.
+1. **Context builder** — extracts DAG run metadata (run ID, logical date, schedule, interval, duration, environment) from the Airflow callback context once, and makes it available to all channels.
+2. **Channel formatters** — each channel (Slack Block Kit, HTML email, Teams Adaptive Card, Discord embed) formats the same context into the right payload for that platform.
+3. **Two usage patterns** — as an `on_failure_callback` (invisible to the graph) or as an explicit `@task` node in the DAG graph.
 
-Create an Airflow connection named `SLACK_WEBHOOK_NOTIFICATION_CONN` (or set `AIRFLOW_CONN_SLACK_WEBHOOK_NOTIFICATION_CONN`).
+### Pattern A — `on_failure_callback`
 
-#### DAG-level notification
+The callback fires automatically when any task in the DAG fails. Nothing appears in the task graph.
 
 ```python
-from datetime import datetime, timedelta
-from airflow.sdk import DAG
-from airflow.providers.standard.operators.bash import BashOperator
-from airflow_toolkit.notifications.slack.webhook import dag_failure_slack_notification_webhook
+from airflow_toolkit.notifications import dag_failure_notification
 
 with DAG(
     'my_pipeline',
-    schedule=timedelta(days=1),
-    start_date=datetime(2024, 1, 1),
-    catchup=False,
-    on_failure_callback=dag_failure_slack_notification_webhook(),
-) as dag:
-
-    t = BashOperator(task_id='run', bash_command='python my_script.py')
+    schedule='0 6 * * *',
+    on_failure_callback=dag_failure_notification(
+        channels=['slack', 'email'],
+        environment='PROD',
+        slack_webhook_url='https://hooks.slack.com/services/...',
+        email_to=['data-team@example.com'],
+    ),
+):
+    ...
 ```
 
-#### Task-level notification
+### Pattern B — explicit task in the graph
+
+`get_failure_notification_task` returns an Airflow task with `trigger_rule='one_failed'`. It fires when any upstream task fails and is **skipped** when all tasks succeed. The notification step is visible in the Airflow UI, has its own logs, and appears in the task history.
 
 ```python
-BashOperator(
-    task_id='run',
-    bash_command='python my_script.py',
-    on_failure_callback=dag_failure_slack_notification_webhook(source='TASK'),
-)
-```
+from airflow_toolkit.notifications import get_failure_notification_task
 
-#### Custom message
+with DAG('my_pipeline', schedule='0 6 * * *'):
+    extract = ...
+    load    = ...
 
-```python
-on_failure_callback=dag_failure_slack_notification_webhook(
-    text='Pipeline {{ dag.dag_id }} failed on {{ ds }}',
-    include_blocks=False,
-)
+    notify = get_failure_notification_task(
+        channels=['slack', 'email'],
+        environment='PROD',
+        slack_webhook_url='https://hooks.slack.com/services/...',
+        email_to=['data-team@example.com'],
+    )
+
+    [extract, load] >> notify
 ```
 
-#### Custom Slack blocks
+### Supported channels
+
+| Channel | Parameter | Requires |
+|---|---|---|
+| `slack` | `slack_webhook_url` | — |
+| `email` | `email_to: list[str]`, `email_from` (optional) | Airflow SMTP configured |
+| `teams` | `teams_webhook_url` | — |
+| `discord` | `discord_webhook_url` | — |
+
+Any combination of channels can be used in a single call. Channels are delivered sequentially in the order listed.
+
+### All parameters
 
 ```python
-on_failure_callback=dag_failure_slack_notification_webhook(
-    blocks={
-        'type': 'section',
-        'text': {'type': 'mrkdwn', 'text': '*Pipeline failed* — check the logs.'},
-    }
+dag_failure_notification(
+    channels=['slack', 'email', 'teams', 'discord'],
+
+    # Environment label shown in every notification (DEV / STG / PROD)
+    environment='PROD',
+
+    # Slack
+    slack_webhook_url='https://hooks.slack.com/services/...',
+
+    # Email
+    email_to=['ops@example.com'],
+    email_from=None,                 # uses Airflow SMTP default if omitted
+
+    # Teams
+    teams_webhook_url='https://outlook.office.com/webhook/...',
+
+    # Discord
+    discord_webhook_url='https://discord.com/api/webhooks/...',
 )
 ```
 
-Default notification format:
+`get_failure_notification_task` accepts the same parameters.
 
-![image](https://github.com/DeepKernelLabs/airflow-toolkit/assets/152852247/52a5bf95-21bc-4c3b-8093-79953c0c5d61)
+### Environment colours
 
-**Parameters:**
+Each environment maps to a distinct colour across all channels so alerts are recognisable at a glance:
 
-| Parameter | Type | Description |
-|---|---|---|
-| `text` | `str` (optional) | Plain-text message. Overrides blocks if provided. |
-| `blocks` | `dict` (optional) | Custom Slack Block Kit payload. |
-| `include_blocks` | `bool` (optional) | Whether to include the default formatted block. |
-| `source` | `'DAG'` \| `'TASK'` (optional) | Source of the failure. Default: `'DAG'`. |
-| `image_url` | `str` (optional) | Accessory image URL. Can also be set via `AIRFLOW_TOOLKIT__SLACK_NOTIFICATION_IMG_URL`. |
+| Environment | Slack | Teams | Discord |
+|---|---|---|---|
+| `PROD` | 🔴 red | Attention (red) | #ED4245 |
+| `STG` | 🟡 yellow | Warning (orange) | #FF8C00 |
+| `DEV` | 🟢 green | Good (green) | #57F287 |
 
 ---
 

{airflow_toolkit-2.0.1 → airflow_toolkit-2.2.0}/README.md

@@ -381,74 +381,99 @@ Because `AzureDatabricksSqlHook` implements `DbApiHook`, it can be used as `sour
 
 ## Notifications
 
-### Slack (incoming webhook)
+Send rich failure notifications to Slack, email, Microsoft Teams, and Discord from a single call. The notification system is built around three ideas:
 
-Send DAG or task failure alerts to a Slack channel using `dag_failure_slack_notification_webhook`. Requires a Slack App with Incoming Webhooks enabled.
+1. **Context builder** — extracts DAG run metadata (run ID, logical date, schedule, interval, duration, environment) from the Airflow callback context once, and makes it available to all channels.
+2. **Channel formatters** — each channel (Slack Block Kit, HTML email, Teams Adaptive Card, Discord embed) formats the same context into the right payload for that platform.
+3. **Two usage patterns** — as an `on_failure_callback` (invisible to the graph) or as an explicit `@task` node in the DAG graph.
 
-Create an Airflow connection named `SLACK_WEBHOOK_NOTIFICATION_CONN` (or set `AIRFLOW_CONN_SLACK_WEBHOOK_NOTIFICATION_CONN`).
+### Pattern A — `on_failure_callback`
 
-#### DAG-level notification
+The callback fires automatically when any task in the DAG fails. Nothing appears in the task graph.
 
 ```python
-from datetime import datetime, timedelta
-from airflow.sdk import DAG
-from airflow.providers.standard.operators.bash import BashOperator
-from airflow_toolkit.notifications.slack.webhook import dag_failure_slack_notification_webhook
+from airflow_toolkit.notifications import dag_failure_notification
 
 with DAG(
     'my_pipeline',
-    schedule=timedelta(days=1),
-    start_date=datetime(2024, 1, 1),
-    catchup=False,
-    on_failure_callback=dag_failure_slack_notification_webhook(),
-) as dag:
-
-    t = BashOperator(task_id='run', bash_command='python my_script.py')
+    schedule='0 6 * * *',
+    on_failure_callback=dag_failure_notification(
+        channels=['slack', 'email'],
+        environment='PROD',
+        slack_webhook_url='https://hooks.slack.com/services/...',
+        email_to=['data-team@example.com'],
+    ),
+):
+    ...
 ```
 
-#### Task-level notification
+### Pattern B — explicit task in the graph
+
+`get_failure_notification_task` returns an Airflow task with `trigger_rule='one_failed'`. It fires when any upstream task fails and is **skipped** when all tasks succeed. The notification step is visible in the Airflow UI, has its own logs, and appears in the task history.
 
 ```python
-BashOperator(
-    task_id='run',
-    bash_command='python my_script.py',
-    on_failure_callback=dag_failure_slack_notification_webhook(source='TASK'),
-)
-```
+from airflow_toolkit.notifications import get_failure_notification_task
 
-#### Custom message
+with DAG('my_pipeline', schedule='0 6 * * *'):
+    extract = ...
+    load    = ...
 
-```python
-on_failure_callback=dag_failure_slack_notification_webhook(
-    text='Pipeline {{ dag.dag_id }} failed on {{ ds }}',
-    include_blocks=False,
-)
+    notify = get_failure_notification_task(
+        channels=['slack', 'email'],
+        environment='PROD',
+        slack_webhook_url='https://hooks.slack.com/services/...',
+        email_to=['data-team@example.com'],
+    )
+
+    [extract, load] >> notify
 ```
 
-#### Custom Slack blocks
+### Supported channels
+
+| Channel | Parameter | Requires |
+|---|---|---|
+| `slack` | `slack_webhook_url` | — |
+| `email` | `email_to: list[str]`, `email_from` (optional) | Airflow SMTP configured |
+| `teams` | `teams_webhook_url` | — |
+| `discord` | `discord_webhook_url` | — |
+
+Any combination of channels can be used in a single call. Channels are delivered sequentially in the order listed.
+
+### All parameters
 
 ```python
-on_failure_callback=dag_failure_slack_notification_webhook(
-    blocks={
-        'type': 'section',
-        'text': {'type': 'mrkdwn', 'text': '*Pipeline failed* — check the logs.'},
-    }
+dag_failure_notification(
+    channels=['slack', 'email', 'teams', 'discord'],
+
+    # Environment label shown in every notification (DEV / STG / PROD)
+    environment='PROD',
+
+    # Slack
+    slack_webhook_url='https://hooks.slack.com/services/...',
+
+    # Email
+    email_to=['ops@example.com'],
+    email_from=None,                 # uses Airflow SMTP default if omitted
+
+    # Teams
+    teams_webhook_url='https://outlook.office.com/webhook/...',
+
+    # Discord
+    discord_webhook_url='https://discord.com/api/webhooks/...',
 )
 ```
 
-Default notification format:
+`get_failure_notification_task` accepts the same parameters.
 
-![image](https://github.com/DeepKernelLabs/airflow-toolkit/assets/152852247/52a5bf95-21bc-4c3b-8093-79953c0c5d61)
+### Environment colours
 
-**Parameters:**
+Each environment maps to a distinct colour across all channels so alerts are recognisable at a glance:
 
-| Parameter | Type | Description |
-|---|---|---|
-| `text` | `str` (optional) | Plain-text message. Overrides blocks if provided. |
-| `blocks` | `dict` (optional) | Custom Slack Block Kit payload. |
-| `include_blocks` | `bool` (optional) | Whether to include the default formatted block. |
-| `source` | `'DAG'` \| `'TASK'` (optional) | Source of the failure. Default: `'DAG'`. |
-| `image_url` | `str` (optional) | Accessory image URL. Can also be set via `AIRFLOW_TOOLKIT__SLACK_NOTIFICATION_IMG_URL`. |
+| Environment | Slack | Teams | Discord |
+|---|---|---|---|
+| `PROD` | 🔴 red | Attention (red) | #ED4245 |
+| `STG` | 🟡 yellow | Warning (orange) | #FF8C00 |
+| `DEV` | 🟢 green | Good (green) | #57F287 |
 
 ---
 

{airflow_toolkit-2.0.1 → airflow_toolkit-2.2.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "airflow-toolkit"
-version = "2.0.1"
+version = "2.2.0"
 description = "A toolkit of operators, hooks and utilities for Apache Airflow 3"
 authors = [{ name = "Biel Llobera", email = "biel_llobera@dkl.digital" }]
 requires-python = ">=3.11,<3.15"
@@ -37,9 +37,6 @@ azure = [
 sftp = [
     "apache-airflow-providers-sftp>=5.3",
 ]
-slack = [
-    "apache-airflow-providers-slack>=9.3",
-]
 http = [
     "apache-airflow-providers-http>=5.6.4",
     "requests>=2.31.0",
@@ -64,7 +61,6 @@ airflow3-full = [
     "apache-airflow-providers-google>=18",
     "apache-airflow-providers-microsoft-azure>=8",
     "apache-airflow-providers-sftp>=5.3",
-    "apache-airflow-providers-slack>=9.3",
     "apache-airflow-providers-http>=5.6.4",
     "requests>=2.31.0",
     "jmespath>=1.0.1,<2",

{airflow_toolkit-2.0.1 → airflow_toolkit-2.2.0}/src/airflow_toolkit/filesystems/filesystem_protocol.py

@@ -17,4 +17,10 @@ class FilesystemProtocol(Protocol):
 
     def check_prefix(self, prefix: str) -> bool: ...
 
-    def list_files(self, prefix: str) -> list[str]: ...
+    def list_files(self, prefix: str) -> list[str]:
+        """Return paths of all files under *prefix*, recursively.
+
+        Directories are never included. Returned paths must be usable
+        with ``read()``, ``delete_file()``, etc.
+        """
+        ...

{airflow_toolkit-2.0.1 → airflow_toolkit-2.2.0}/src/airflow_toolkit/filesystems/impl/azure_databricks_volume_filesystem.py

@@ -8,7 +8,7 @@ from airflow_toolkit.providers.azure.hooks.azure_databricks import (
     AzureDatabricksVolumeHook,
 )
 
-logger = logging.getLogger(__file__)
+logger = logging.getLogger(__name__)
 
 
 class AzureDatabricksVolumeFilesystem(FilesystemProtocol):
@@ -52,16 +52,9 @@ class AzureDatabricksVolumeFilesystem(FilesystemProtocol):
         conn.files.delete_directory(prefix)
 
     def check_file(self, path: str) -> bool:
-        prefix = path.rsplit("/", 1)[0]
-
         try:
-            file_list_path = [
-                entry.path == path
-                for entry in self.hook.get_conn().files.list_directory_contents(prefix)
-                if not entry.is_directory
-            ]
-            return any(file_list_path)
-
+            self.hook.get_conn().files.get_metadata(path)
+            return True
         except NotFound:
             return False
 
@@ -79,8 +72,15 @@ class AzureDatabricksVolumeFilesystem(FilesystemProtocol):
             return False
 
     def list_files(self, prefix: str) -> list[str]:
-        return [
-            entry.path
-            for entry in self.hook.get_conn().files.list_directory_contents(prefix)
-            if not entry.is_directory and entry.path is not None
-        ]
+        results: list[str] = []
+        try:
+            for entry in self.hook.get_conn().files.list_directory_contents(prefix):
+                if entry.path is None:
+                    continue
+                if entry.is_directory:
+                    results.extend(self.list_files(entry.path))
+                else:
+                    results.append(entry.path)
+        except NotFound:
+            return []
+        return results

{airflow_toolkit-2.0.1 → airflow_toolkit-2.2.0}/src/airflow_toolkit/filesystems/impl/azure_file_share_filesystem.py

@@ -8,7 +8,7 @@ from airflow_toolkit.providers.azure.hooks.azure_file_share import (
     AzureFileShareServicePrincipalHook,
 )
 
-logger = logging.getLogger(__file__)
+logger = logging.getLogger(__name__)
 
 
 class AzureFileShareFilesystem(FilesystemProtocol):
@@ -47,8 +47,11 @@ class AzureFileShareFilesystem(FilesystemProtocol):
         return self.hook.get_conn().get_directory_client(prefix).exists()
 
     def list_files(self, prefix: str) -> list[str]:
-        return [
-            f"{prefix}/{item.name}"
-            for item in self.hook.get_conn().list_directories_and_files(prefix)
-            if not item.is_directory
-        ]
+        results: list[str] = []
+        for item in self.hook.get_conn().list_directories_and_files(prefix):
+            full_path = f"{prefix}/{item.name}"
+            if isinstance(item, DirectoryProperties):
+                results.extend(self.list_files(full_path))
+            else:
+                results.append(full_path)
+        return results

{airflow_toolkit-2.0.1 → airflow_toolkit-2.2.0}/src/airflow_toolkit/filesystems/impl/blob_storage_filesystem.py

@@ -5,7 +5,7 @@ from airflow.providers.microsoft.azure.hooks.wasb import WasbHook
 
 from airflow_toolkit.filesystems.filesystem_protocol import FilesystemProtocol
 
-logger = logging.getLogger(__file__)
+logger = logging.getLogger(__name__)
 
 
 class BlobStorageFilesystem(FilesystemProtocol):
@@ -15,7 +15,7 @@ class BlobStorageFilesystem(FilesystemProtocol):
     def read(self, path: str) -> bytes:
         container_name, blob_name = _get_container_and_blob_name(path)
         stream = self.hook.download(container_name=container_name, blob_name=blob_name)
-        return stream.readall().encode()
+        return stream.readall()
 
     def write(self, data: str | bytes | BytesIO, path: str):
         container_name, blob_name = _get_container_and_blob_name(path)

{airflow_toolkit-2.0.1 → airflow_toolkit-2.2.0}/src/airflow_toolkit/filesystems/impl/local_filesystem.py

@@ -47,4 +47,4 @@ class LocalFilesystem(FilesystemProtocol):
 
     def list_files(self, prefix: str) -> list[str]:
         path_to_list = Path(self.hook.get_path()) / prefix.lstrip("/")
-        return [str(file) for file in path_to_list.glob("*") if file.is_file()]
+        return [str(file) for file in path_to_list.rglob("*") if file.is_file()]

{airflow_toolkit-2.0.1 → airflow_toolkit-2.2.0}/src/airflow_toolkit/filesystems/impl/sftp_filesystem.py

@@ -55,4 +55,5 @@ class SFTPFilesystem(FilesystemProtocol):
         return self.hook.isdir(prefix)
 
     def list_files(self, prefix: str) -> list[str]:
-        return self.hook.list_directory(prefix)
+        files, _, _ = self.hook.get_tree_map(prefix)
+        return files

airflow_toolkit-2.2.0/src/airflow_toolkit/notifications/__init__.py

@@ -0,0 +1,134 @@
+from __future__ import annotations
+
+import typing
+
+from airflow_toolkit.notifications.context import (
+    NotificationContext,
+    build_notification_context,
+)
+
+__all__ = [
+    "dag_failure_notification",
+    "get_failure_notification_task",
+    "build_notification_context",
+    "NotificationContext",
+]
+
+_VALID_CHANNELS = {"slack", "email", "teams", "discord"}
+
+
+def dag_failure_notification(
+    channels: list[str],
+    environment: str = "PROD",
+    slack_webhook_url: str | None = None,
+    email_to: list[str] | None = None,
+    email_from: str | None = None,
+    teams_webhook_url: str | None = None,
+    discord_webhook_url: str | None = None,
+) -> typing.Callable[[dict[str, typing.Any]], None]:
+    """Return an on_failure_callback that sends DAG failure notifications.
+
+    Usage::
+
+        with DAG(..., on_failure_callback=dag_failure_notification(
+            channels=["slack", "email"],
+            environment="PROD",
+            slack_webhook_url="https://hooks.slack.com/services/...",
+            email_to=["ops@example.com"],
+        )):
+            ...
+    """
+    unknown = set(channels) - _VALID_CHANNELS
+    if unknown:
+        raise ValueError(
+            f"Unknown notification channels: {unknown}. Valid: {_VALID_CHANNELS}"
+        )
+
+    def callback(context: dict[str, typing.Any]) -> None:
+        ctx = build_notification_context(context, environment=environment)
+
+        if "slack" in channels:
+            if not slack_webhook_url:
+                raise ValueError(
+                    "slack_webhook_url is required when 'slack' is in channels"
+                )
+            from airflow_toolkit.notifications.channels.slack import (
+                send_slack_notification,
+            )
+
+            send_slack_notification(ctx, webhook_url=slack_webhook_url)
+
+        if "email" in channels:
+            if not email_to:
+                raise ValueError("email_to is required when 'email' is in channels")
+            from airflow_toolkit.notifications.channels.email import (
+                send_email_notification,
+            )
+
+            send_email_notification(ctx, to=email_to, from_email=email_from)
+
+        if "teams" in channels:
+            if not teams_webhook_url:
+                raise ValueError(
+                    "teams_webhook_url is required when 'teams' is in channels"
+                )
+            from airflow_toolkit.notifications.channels.teams import (
+                send_teams_notification,
+            )
+
+            send_teams_notification(ctx, webhook_url=teams_webhook_url)
+
+        if "discord" in channels:
+            if not discord_webhook_url:
+                raise ValueError(
+                    "discord_webhook_url is required when 'discord' is in channels"
+                )
+            from airflow_toolkit.notifications.channels.discord import (
+                send_discord_notification,
+            )
+
+            send_discord_notification(ctx, webhook_url=discord_webhook_url)
+
+    return callback
+
+
+def get_failure_notification_task(
+    channels: list[str],
+    environment: str = "PROD",
+    slack_webhook_url: str | None = None,
+    email_to: list[str] | None = None,
+    email_from: str | None = None,
+    teams_webhook_url: str | None = None,
+    discord_webhook_url: str | None = None,
+) -> typing.Any:
+    """Return an @task with trigger_rule='one_failed' that sends failure notifications.
+
+    Usage::
+
+        notify = get_failure_notification_task(channels=["slack"], environment="PROD")
+        [task_a, task_b] >> notify
+    """
+    from airflow.sdk import task as airflow_task
+
+    _channels = channels
+    _environment = environment
+    _slack_webhook_url = slack_webhook_url
+    _email_to = email_to
+    _email_from = email_from
+    _teams_webhook_url = teams_webhook_url
+    _discord_webhook_url = discord_webhook_url
+
+    @airflow_task(task_id="notify_failure", trigger_rule="one_failed")
+    def notify_failure(**context: typing.Any) -> None:
+        callback = dag_failure_notification(
+            channels=_channels,
+            environment=_environment,
+            slack_webhook_url=_slack_webhook_url,
+            email_to=_email_to,
+            email_from=_email_from,
+            teams_webhook_url=_teams_webhook_url,
+            discord_webhook_url=_discord_webhook_url,
+        )
+        callback(context)
+
+    return notify_failure()

airflow_toolkit-2.2.0/src/airflow_toolkit/notifications/channels/discord.py

@@ -0,0 +1,57 @@
+from __future__ import annotations
+
+import typing
+
+from airflow_toolkit.notifications.context import NotificationContext
+
+# Discord embed colors (decimal RGB)
+_ENV_COLOR: dict[str, int] = {
+    "PROD": 15548997,  # red    #ED4245
+    "STG": 16750592,  # orange #FF8C00
+    "DEV": 5763719,  # green  #57F287
+}
+
+
+def build_discord_payload(ctx: NotificationContext) -> dict[str, typing.Any]:
+    env = ctx["environment"]
+    color = _ENV_COLOR.get(env, 15548997)
+
+    return {
+        "content": f"\U0001f534 DAG `{ctx['dag_id']}` failed",
+        "embeds": [
+            {
+                "title": f"[{env}] DAG Failure — {ctx['dag_id']}",
+                "url": ctx["dag_url"],
+                "color": color,
+                "fields": [
+                    {"name": "Run ID", "value": ctx["run_id"], "inline": False},
+                    {"name": "Environment", "value": env, "inline": True},
+                    {"name": "Logical Date", "value": ctx["ds"], "inline": True},
+                    {"name": "Schedule", "value": ctx["schedule"], "inline": True},
+                    {
+                        "name": "Interval Start",
+                        "value": ctx["data_interval_start"],
+                        "inline": True,
+                    },
+                    {
+                        "name": "Interval End",
+                        "value": ctx["data_interval_end"],
+                        "inline": True,
+                    },
+                    {
+                        "name": "Execution At",
+                        "value": ctx["execution_at"],
+                        "inline": True,
+                    },
+                    {"name": "Duration", "value": ctx["duration"], "inline": True},
+                ],
+                "footer": {"text": ctx["base_url"]},
+            }
+        ],
+    }
+
+
+def send_discord_notification(ctx: NotificationContext, webhook_url: str) -> None:
+    import requests
+
+    requests.post(webhook_url, json=build_discord_payload(ctx)).raise_for_status()