tracer-cloud--opensre/integrations/datadog/__init__.py

"""Datadog integration classifier."""

from __future__ import annotations

import logging
from typing import Any

from integrations._validation_helpers import report_classify_failure
from integrations.config_models import DatadogIntegrationConfig

logger = logging.getLogger(__name__)


def classify(
    credentials: dict[str, Any], record_id: str
) -> tuple[DatadogIntegrationConfig | None, str | None]:
    try:
        cfg = DatadogIntegrationConfig.model_validate(
            {
                "api_key": credentials.get("api_key", ""),
                "app_key": credentials.get("app_key", ""),
                "site": credentials.get("site", "datadoghq.com"),
                "integration_id": record_id,
            }
        )
    except Exception as exc:
        report_classify_failure(exc, logger=logger, integration="datadog", record_id=record_id)
        return None, None
    if cfg.api_key and cfg.app_key:
        return cfg, "datadog"
    return None, None