tracer-cloud--opensre/tests/utils/alert_factory/factory.py

import uuid
from typing import Any

from .formatters.grafana import format_as_grafana
from .intent import AlertIntent


def from_pipeline_run(
    pipeline_name: str,
    run_name: str,
    status: str,
    timestamp: str,
    severity: str = "critical",
    alert_name: str = "PipelineFailure",
    environment: str = "production",
    trace_id: str | None = None,
    run_url: str | None = None,
    external_url: str = "",
    alert_id: str | None = None,
    annotations: dict[str, Any] | None = None,
) -> dict[str, Any]:
    """Pure constructor: creates a pipeline run alert payload (Grafana format by default)."""
    intent = AlertIntent(
        pipeline_name=pipeline_name,
        run_name=run_name,
        status=status,
        timestamp=timestamp,
        severity=severity,
        alert_name=alert_name,
        environment=environment,
        trace_id=trace_id,
        run_url=run_url,
        external_url=external_url,
        alert_id=alert_id or str(uuid.uuid4()),
        annotations=annotations or {},
    )

    return format_as_grafana(intent)


def create_alert(
    pipeline_name: str,
    run_name: str,
    status: str,
    timestamp: str,
    annotations: dict[str, Any] | None = None,
    trace_id: str | None = None,
    run_url: str | None = None,
    external_url: str = "",
    alert_id: str | None = None,
    severity: str = "critical",
    alert_name: str = "PipelineFailure",
    environment: str = "production",
) -> dict[str, Any]:
    """
    Source-agnostic constructor for standardized alerts.

    Assembles an AlertIntent and renders it via the default formatter (Grafana).
    """
    return from_pipeline_run(
        pipeline_name=pipeline_name,
        run_name=run_name,
        status=status,
        timestamp=timestamp,
        annotations=annotations,
        trace_id=trace_id,
        run_url=run_url,
        external_url=external_url,
        alert_id=alert_id,
        severity=severity,
        alert_name=alert_name,
        environment=environment,
    )