from databricks.sdk import WorkspaceClient
from databricks.sdk.service import sql

w = WorkspaceClient()

# Create SQL Warehouse
warehouse = w.sql_warehouses.create(
    name="Analytics Warehouse",
    cluster_size="2X-Small",
    max_num_clusters=4,
    min_num_clusters=1,
    auto_stop_mins=10,
    enable_serverless_compute=True,
    channel="CHANNEL_10_4"
)

# Execute query
result = w.statement_execution.execute_statement(
    warehouse_id=warehouse.id,
    statement="""
        SELECT 
            sale_date,
            SUM(total_amount) AS revenue,
            COUNT(*) AS transactions
        FROM sales.fact_sales
        WHERE sale_date >= '2024-01-01'
        GROUP BY sale_date
        ORDER BY sale_date
    """
)

for row in result.result.data_array:
    print(row)

# Create alert
alert = w.alerts.create(
    name="Revenue Drop Alert",
    query="""
        SELECT 
            SUM(total_amount) as daily_revenue
        FROM sales.fact_sales
        WHERE sale_date = current_date()
    """,
    condition="daily_revenue < 100000",
    state="ACTIVE",
    subscribers=[
        {"user_id": "user@company.com", "alert_type": "EMAIL"}
    ]
)

-- Dashboard query with parameters
SELECT
    sale_date,
    product_category,
    SUM(total_amount) AS revenue,
    COUNT(*) AS transactions
FROM sales.fact_sales
WHERE sale_date BETWEEN :start_date AND :end_date
  AND product_category = :category
GROUP BY sale_date, product_category
ORDER BY sale_date;