Add PyDABs support for SQL Alerts

acceptance/bundle/python/alerts-support/databricks.yml

@@ -0,0 +1,24 @@
+bundle:
+  name: my_project
+
+sync: {paths: []} # don't need to copy files
+
+python:
+  resources:
+    - "resources:load_resources"
+  mutators:
+    - "mutators:update_alert"
+
+resources:
+  alerts:
+    my_alert_1:
+      display_name: "My Alert"
+      query_text: "SELECT 1"
+      warehouse_id: "my_warehouse"
+      evaluation:
+        comparison_operator: "GREATER_THAN"
+        source:
+          name: "column_1"
+      schedule:
+        quartz_cron_schedule: "0 0 0 * * ?"
+        timezone_id: "UTC"

acceptance/bundle/python/alerts-support/mutators.py

@@ -0,0 +1,11 @@
+from dataclasses import replace
+
+from databricks.bundles.alerts import Alert
+from databricks.bundles.core import alert_mutator
+
+
+@alert_mutator
+def update_alert(alert: Alert) -> Alert:
+    assert isinstance(alert.display_name, str)
+
+    return replace(alert, display_name=f"{alert.display_name} (updated)")

acceptance/bundle/python/alerts-support/output.txt

@@ -0,0 +1,50 @@
+
+>>> uv run [UV_ARGS] -q [CLI] bundle validate --output json
+{
+  "experimental": {
+    "python": {
+      "mutators": [
+        "mutators:update_alert"
+      ],
+      "resources": [
+        "resources:load_resources"
+      ]
+    }
+  },
+  "resources": {
+    "alerts": {
+      "my_alert_1": {
+        "display_name": "My Alert (updated)",
+        "evaluation": {
+          "comparison_operator": "GREATER_THAN",
+          "source": {
+            "name": "column_1"
+          }
+        },
+        "parent_path": "/Workspace/Users/[USERNAME]/.bundle/my_project/default/resources",
+        "query_text": "SELECT 1",
+        "schedule": {
+          "quartz_cron_schedule": "0 0 0 * * ?",
+          "timezone_id": "UTC"
+        },
+        "warehouse_id": "my_warehouse"
+      },
+      "my_alert_2": {
+        "display_name": "My Alert (2) (updated)",
+        "evaluation": {
+          "comparison_operator": "LESS_THAN",
+          "source": {
+            "name": "column_2"
+          }
+        },
+        "parent_path": "/Workspace/Users/[USERNAME]/.bundle/my_project/default/resources",
+        "query_text": "SELECT 2",
+        "schedule": {
+          "quartz_cron_schedule": "0 0 12 * * ?",
+          "timezone_id": "UTC"
+        },
+        "warehouse_id": "my_warehouse_2"
+      }
+    }
+  }
+}

acceptance/bundle/python/alerts-support/resources.py

@@ -0,0 +1,24 @@
+from databricks.bundles.core import Resources
+
+
+def load_resources() -> Resources:
+    resources = Resources()
+
+    resources.add_alert(
+        "my_alert_2",
+        {
+            "display_name": "My Alert (2)",
+            "query_text": "SELECT 2",
+            "warehouse_id": "my_warehouse_2",
+            "evaluation": {
+                "comparison_operator": "LESS_THAN",
+                "source": {"name": "column_2"},
+            },
+            "schedule": {
+                "quartz_cron_schedule": "0 0 12 * * ?",
+                "timezone_id": "UTC",
+            },
+        },
+    )
+
+    return resources

acceptance/bundle/python/alerts-support/script

@@ -0,0 +1,5 @@
+
+trace uv run $UV_ARGS -q $CLI bundle validate --output json | \
+  jq "pick(.experimental.python, .resources)"
+
+rm -fr .databricks __pycache__

acceptance/bundle/python/alerts-support/test.toml

@@ -0,0 +1,4 @@
+Cloud = false # tests don't interact with APIs
+
+# alerts are only supported in the current version of the wheel
+EnvMatrix.PYDAB_VERSION = ["current"]

python/codegen/codegen/packages.py

@@ -8,6 +8,7 @@
     "resources.Catalog": "catalogs",
     "resources.Schema": "schemas",
     "resources.Volume": "volumes",
+    "resources.Alert": "alerts",
 }
 
 RESOURCE_TYPES = list(RESOURCE_NAMESPACE.keys())
@@ -21,6 +22,8 @@
     "pipelines",
     "resources",
     "catalog",
+    "sql",
+    "iam",
 ]
 
 RENAMES = {

python/databricks/bundles/alerts/__init__.py

@@ -0,0 +1,123 @@
+__all__ = [
+    "Aggregation",
+    "AggregationParam",
+    "Alert",
+    "AlertDict",
+    "AlertEvaluationState",
+    "AlertEvaluationStateParam",
+    "AlertParam",
+    "AlertStatementParameter",
+    "AlertStatementParameterDict",
+    "AlertStatementParameterParam",
+    "AlertV2Evaluation",
+    "AlertV2EvaluationDict",
+    "AlertV2EvaluationParam",
+    "AlertV2Notification",
+    "AlertV2NotificationDict",
+    "AlertV2NotificationParam",
+    "AlertV2Operand",
+    "AlertV2OperandColumn",
+    "AlertV2OperandColumnDict",
+    "AlertV2OperandColumnParam",
+    "AlertV2OperandDict",
+    "AlertV2OperandParam",
+    "AlertV2OperandValue",
+    "AlertV2OperandValueDict",
+    "AlertV2OperandValueParam",
+    "AlertV2RunAs",
+    "AlertV2RunAsDict",
+    "AlertV2RunAsParam",
+    "AlertV2Subscription",
+    "AlertV2SubscriptionDict",
+    "AlertV2SubscriptionParam",
+    "ComparisonOperator",
+    "ComparisonOperatorParam",
+    "CronSchedule",
+    "CronScheduleDict",
+    "CronScheduleParam",
+    "Lifecycle",
+    "LifecycleDict",
+    "LifecycleParam",
+    "Permission",
+    "PermissionDict",
+    "PermissionLevel",
+    "PermissionLevelParam",
+    "PermissionParam",
+    "SchedulePauseStatus",
+    "SchedulePauseStatusParam",
+]
+
+
+from databricks.bundles.alerts._models.aggregation import Aggregation, AggregationParam
+from databricks.bundles.alerts._models.alert import Alert, AlertDict, AlertParam
+from databricks.bundles.alerts._models.alert_evaluation_state import (
+    AlertEvaluationState,
+    AlertEvaluationStateParam,
+)
+from databricks.bundles.alerts._models.alert_statement_parameter import (
+    AlertStatementParameter,
+    AlertStatementParameterDict,
+    AlertStatementParameterParam,
+)
+from databricks.bundles.alerts._models.alert_v2_evaluation import (
+    AlertV2Evaluation,
+    AlertV2EvaluationDict,
+    AlertV2EvaluationParam,
+)
+from databricks.bundles.alerts._models.alert_v2_notification import (
+    AlertV2Notification,
+    AlertV2NotificationDict,
+    AlertV2NotificationParam,
+)
+from databricks.bundles.alerts._models.alert_v2_operand import (
+    AlertV2Operand,
+    AlertV2OperandDict,
+    AlertV2OperandParam,
+)
+from databricks.bundles.alerts._models.alert_v2_operand_column import (
+    AlertV2OperandColumn,
+    AlertV2OperandColumnDict,
+    AlertV2OperandColumnParam,
+)
+from databricks.bundles.alerts._models.alert_v2_operand_value import (
+    AlertV2OperandValue,
+    AlertV2OperandValueDict,
+    AlertV2OperandValueParam,
+)
+from databricks.bundles.alerts._models.alert_v2_run_as import (
+    AlertV2RunAs,
+    AlertV2RunAsDict,
+    AlertV2RunAsParam,
+)
+from databricks.bundles.alerts._models.alert_v2_subscription import (
+    AlertV2Subscription,
+    AlertV2SubscriptionDict,
+    AlertV2SubscriptionParam,
+)
+from databricks.bundles.alerts._models.comparison_operator import (
+    ComparisonOperator,
+    ComparisonOperatorParam,
+)
+from databricks.bundles.alerts._models.cron_schedule import (
+    CronSchedule,
+    CronScheduleDict,
+    CronScheduleParam,
+)
+from databricks.bundles.alerts._models.lifecycle import (
+    Lifecycle,
+    LifecycleDict,
+    LifecycleParam,
+)
+from databricks.bundles.alerts._models.permission import (
+    Permission,
+    PermissionDict,
+    PermissionParam,
+)
+from databricks.bundles.alerts._models.permission_level import (
+    PermissionLevel,
+    PermissionLevelParam,
+)
+from databricks.bundles.alerts._models.schedule_pause_status import (
+    SchedulePauseStatus,
+    SchedulePauseStatusParam,
+)

python/databricks/bundles/alerts/_models/aggregation.py

@@ -0,0 +1,19 @@
+from enum import Enum
+from typing import Literal
+
+
+class Aggregation(Enum):
+    SUM = "SUM"
+    COUNT = "COUNT"
+    COUNT_DISTINCT = "COUNT_DISTINCT"
+    AVG = "AVG"
+    MEDIAN = "MEDIAN"
+    MIN = "MIN"
+    MAX = "MAX"
+    STDDEV = "STDDEV"
+
+
+AggregationParam = (
+    Literal["SUM", "COUNT", "COUNT_DISTINCT", "AVG", "MEDIAN", "MIN", "MAX", "STDDEV"]
+    | Aggregation
+)