truthound-dashboard 1.3.1__py3-none-any.whl → 1.4.0__py3-none-any.whl

truthound_dashboard/core/notifications/deduplication/strategies.py

@@ -0,0 +1,422 @@
+"""Window strategies for deduplication.
+
+This module provides different windowing strategies that determine
+how the deduplication time window is calculated.
+
+Strategies:
+    - SlidingWindowStrategy: Rolling time window from last occurrence
+    - TumblingWindowStrategy: Fixed non-overlapping time windows
+    - SessionWindowStrategy: Gap-based windows for event bursts
+    - AdaptiveWindowStrategy: Dynamic window sizing based on load
+
+Each strategy can be configured with different parameters to
+suit various use cases.
+"""
+
+from __future__ import annotations
+
+import math
+import time
+from abc import ABC, abstractmethod
+from dataclasses import dataclass, field
+from datetime import datetime
+from typing import Any, ClassVar
+
+from ...validation_limits import (
+    get_deduplication_limits,
+    validate_positive_int,
+)
+
+
+class StrategyRegistry:
+    """Registry for window strategies.
+
+    Provides plugin architecture for custom strategy implementations.
+    """
+
+    _strategies: ClassVar[dict[str, type["BaseWindowStrategy"]]] = {}
+
+    @classmethod
+    def register(cls, strategy_type: str):
+        """Decorator to register a strategy type."""
+
+        def decorator(strategy_class: type["BaseWindowStrategy"]) -> type["BaseWindowStrategy"]:
+            strategy_class.strategy_type = strategy_type
+            cls._strategies[strategy_type] = strategy_class
+            return strategy_class
+
+        return decorator
+
+    @classmethod
+    def get(cls, strategy_type: str) -> type["BaseWindowStrategy"] | None:
+        """Get a registered strategy class by type."""
+        return cls._strategies.get(strategy_type)
+
+    @classmethod
+    def create(cls, strategy_type: str, **params: Any) -> "BaseWindowStrategy | None":
+        """Create a strategy instance by type."""
+        strategy_class = cls.get(strategy_type)
+        if strategy_class is None:
+            return None
+        return strategy_class(**params)
+
+    @classmethod
+    def list_types(cls) -> list[str]:
+        """Get list of registered strategy types."""
+        return list(cls._strategies.keys())
+
+
+@dataclass
+class BaseWindowStrategy(ABC):
+    """Abstract base class for window strategies.
+
+    Strategies determine how the deduplication window is calculated
+    for a given fingerprint and context.
+    """
+
+    strategy_type: ClassVar[str] = "base"
+
+    @abstractmethod
+    def get_window_seconds(
+        self,
+        fingerprint: str,
+        context: dict[str, Any] | None = None,
+    ) -> int:
+        """Calculate the window duration in seconds.
+
+        Args:
+            fingerprint: The notification fingerprint.
+            context: Optional context for window calculation.
+
+        Returns:
+            Window duration in seconds.
+        """
+        ...
+
+    @abstractmethod
+    def get_window_key(
+        self,
+        fingerprint: str,
+        timestamp: datetime | None = None,
+    ) -> str:
+        """Generate a window-specific key for the fingerprint.
+
+        Used by tumbling windows to align entries to window boundaries.
+
+        Args:
+            fingerprint: The notification fingerprint.
+            timestamp: Optional timestamp (defaults to now).
+
+        Returns:
+            Window-aligned key.
+        """
+        ...
+
+
+@StrategyRegistry.register("sliding")
+@dataclass
+class SlidingWindowStrategy(BaseWindowStrategy):
+    """Sliding window strategy with validation.
+
+    The window slides with each occurrence - duplicates are suppressed
+    if they occur within `window_seconds` of the last occurrence.
+
+    This is the most common strategy for real-time deduplication.
+
+    Validation:
+        - window_seconds must be between 1 and 86400 (configurable).
+
+    Attributes:
+        window_seconds: Base window duration in seconds.
+    """
+
+    window_seconds: int = 300
+
+    def __post_init__(self) -> None:
+        """Validate window_seconds after initialization."""
+        limits = get_deduplication_limits()
+        valid, error = limits.validate_window_seconds(self.window_seconds)
+        if not valid:
+            raise ValueError(error)
+
+    def get_window_seconds(
+        self,
+        fingerprint: str,
+        context: dict[str, Any] | None = None,
+    ) -> int:
+        """Return the configured window duration."""
+        return self.window_seconds
+
+    def get_window_key(
+        self,
+        fingerprint: str,
+        timestamp: datetime | None = None,
+    ) -> str:
+        """Return the fingerprint unchanged (no window alignment)."""
+        return fingerprint
+
+
+@StrategyRegistry.register("tumbling")
+@dataclass
+class TumblingWindowStrategy(BaseWindowStrategy):
+    """Tumbling window strategy with validation.
+
+    The window is divided into fixed, non-overlapping periods.
+    All events within the same period share the same window.
+
+    Useful for batch-style deduplication where you want at most
+    one notification per time period.
+
+    Validation:
+        - window_seconds must be between 1 and 86400 (configurable).
+        - align_to must be one of 'minute', 'hour', 'day'.
+
+    Attributes:
+        window_seconds: Duration of each tumbling window.
+        align_to: What to align windows to ('minute', 'hour', 'day').
+    """
+
+    window_seconds: int = 300
+    align_to: str = "minute"
+
+    def __post_init__(self) -> None:
+        """Validate configuration after initialization."""
+        limits = get_deduplication_limits()
+        valid, error = limits.validate_window_seconds(self.window_seconds)
+        if not valid:
+            raise ValueError(error)
+
+        valid_alignments = ("minute", "hour", "day")
+        if self.align_to not in valid_alignments:
+            raise ValueError(
+                f"align_to must be one of {valid_alignments}, got '{self.align_to}'"
+            )
+
+    def get_window_seconds(
+        self,
+        fingerprint: str,
+        context: dict[str, Any] | None = None,
+    ) -> int:
+        """Return the configured window duration."""
+        return self.window_seconds
+
+    def get_window_key(
+        self,
+        fingerprint: str,
+        timestamp: datetime | None = None,
+    ) -> str:
+        """Generate window-aligned key.
+
+        Aligns the fingerprint to the start of its tumbling window.
+        """
+        if timestamp is None:
+            timestamp = datetime.utcnow()
+
+        # Calculate window start
+        ts = timestamp.timestamp()
+        window_start = int(ts // self.window_seconds) * self.window_seconds
+
+        return f"{fingerprint}:{window_start}"
+
+
+@StrategyRegistry.register("session")
+@dataclass
+class SessionWindowStrategy(BaseWindowStrategy):
+    """Session window strategy with validation.
+
+    Groups events into sessions based on gaps between occurrences.
+    A new session starts if no event occurs within `gap_seconds`.
+
+    Useful for handling event bursts where you want to deduplicate
+    within a burst but allow new notifications after quiet periods.
+
+    Validation:
+        - gap_seconds must be between 1 and 86400 (configurable).
+        - max_session_seconds must be between 1 and 86400 (configurable).
+        - max_session_seconds must be >= gap_seconds.
+
+    Attributes:
+        gap_seconds: Maximum gap between events in same session.
+        max_session_seconds: Maximum session duration.
+    """
+
+    gap_seconds: int = 60
+    max_session_seconds: int = 3600
+
+    _session_starts: dict[str, float] = field(default_factory=dict)
+    _last_events: dict[str, float] = field(default_factory=dict)
+
+    def __post_init__(self) -> None:
+        """Validate configuration after initialization."""
+        limits = get_deduplication_limits()
+
+        # Validate gap_seconds
+        valid, error = limits.validate_window_seconds(self.gap_seconds)
+        if not valid:
+            raise ValueError(f"gap_seconds: {error}")
+
+        # Validate max_session_seconds
+        valid, error = limits.validate_window_seconds(self.max_session_seconds)
+        if not valid:
+            raise ValueError(f"max_session_seconds: {error}")
+
+        # max_session_seconds should be >= gap_seconds
+        if self.max_session_seconds < self.gap_seconds:
+            raise ValueError(
+                f"max_session_seconds ({self.max_session_seconds}) must be >= "
+                f"gap_seconds ({self.gap_seconds})"
+            )
+
+    def get_window_seconds(
+        self,
+        fingerprint: str,
+        context: dict[str, Any] | None = None,
+    ) -> int:
+        """Calculate remaining session window."""
+        now = time.time()
+        last_event = self._last_events.get(fingerprint, 0)
+
+        # Check if session is still active
+        if now - last_event > self.gap_seconds:
+            # New session
+            return 0
+
+        # Return remaining session time
+        session_start = self._session_starts.get(fingerprint, now)
+        elapsed = now - session_start
+        remaining = self.max_session_seconds - elapsed
+
+        return max(0, int(remaining))
+
+    def get_window_key(
+        self,
+        fingerprint: str,
+        timestamp: datetime | None = None,
+    ) -> str:
+        """Generate session-aligned key."""
+        now = time.time() if timestamp is None else timestamp.timestamp()
+        last_event = self._last_events.get(fingerprint, 0)
+
+        # Check if this is a new session
+        if now - last_event > self.gap_seconds:
+            self._session_starts[fingerprint] = now
+            self._last_events[fingerprint] = now
+            return f"{fingerprint}:session:{int(now)}"
+
+        # Same session
+        self._last_events[fingerprint] = now
+        session_start = self._session_starts.get(fingerprint, now)
+        return f"{fingerprint}:session:{int(session_start)}"
+
+
+@StrategyRegistry.register("adaptive")
+@dataclass
+class AdaptiveWindowStrategy(BaseWindowStrategy):
+    """Adaptive window strategy with validation.
+
+    Dynamically adjusts window size based on event frequency.
+    High-frequency events get longer windows, low-frequency get shorter.
+
+    This helps prevent notification fatigue during incidents while
+    maintaining responsiveness during normal operations.
+
+    Validation:
+        - min_window_seconds must be between 1 and 86400 (configurable).
+        - max_window_seconds must be between 1 and 86400 (configurable).
+        - max_window_seconds must be >= min_window_seconds.
+        - scale_factor must be >= 0.1.
+        - decay_seconds must be between 1 and 86400 (configurable).
+
+    Attributes:
+        min_window_seconds: Minimum window duration.
+        max_window_seconds: Maximum window duration.
+        scale_factor: How quickly window grows with frequency.
+        decay_seconds: Time for window to decay back to minimum.
+    """
+
+    min_window_seconds: int = 60
+    max_window_seconds: int = 3600
+    scale_factor: float = 2.0
+    decay_seconds: int = 1800
+
+    _event_counts: dict[str, int] = field(default_factory=dict)
+    _last_events: dict[str, float] = field(default_factory=dict)
+
+    def __post_init__(self) -> None:
+        """Validate configuration after initialization."""
+        limits = get_deduplication_limits()
+
+        # Validate min_window_seconds
+        valid, error = limits.validate_window_seconds(self.min_window_seconds)
+        if not valid:
+            raise ValueError(f"min_window_seconds: {error}")
+
+        # Validate max_window_seconds
+        valid, error = limits.validate_window_seconds(self.max_window_seconds)
+        if not valid:
+            raise ValueError(f"max_window_seconds: {error}")
+
+        # Validate decay_seconds
+        valid, error = limits.validate_window_seconds(self.decay_seconds)
+        if not valid:
+            raise ValueError(f"decay_seconds: {error}")
+
+        # max_window_seconds must be >= min_window_seconds
+        if self.max_window_seconds < self.min_window_seconds:
+            raise ValueError(
+                f"max_window_seconds ({self.max_window_seconds}) must be >= "
+                f"min_window_seconds ({self.min_window_seconds})"
+            )
+
+        # scale_factor must be positive
+        if self.scale_factor < 0.1:
+            raise ValueError(
+                f"scale_factor must be at least 0.1, got {self.scale_factor}"
+            )
+
+    def get_window_seconds(
+        self,
+        fingerprint: str,
+        context: dict[str, Any] | None = None,
+    ) -> int:
+        """Calculate adaptive window based on event frequency."""
+        now = time.time()
+        last_event = self._last_events.get(fingerprint, 0)
+        count = self._event_counts.get(fingerprint, 0)
+
+        # Decay count over time
+        if last_event > 0:
+            elapsed = now - last_event
+            decay_factor = max(0, 1 - (elapsed / self.decay_seconds))
+            count = int(count * decay_factor)
+
+        # Calculate window based on count
+        if count == 0:
+            window = self.min_window_seconds
+        else:
+            # Logarithmic scaling
+            scale = 1 + math.log(1 + count) * self.scale_factor
+            window = int(self.min_window_seconds * scale)
+
+        # Clamp to bounds
+        return min(self.max_window_seconds, max(self.min_window_seconds, window))
+
+    def get_window_key(
+        self,
+        fingerprint: str,
+        timestamp: datetime | None = None,
+    ) -> str:
+        """Return fingerprint and update counts."""
+        now = time.time() if timestamp is None else timestamp.timestamp()
+
+        # Update event tracking
+        count = self._event_counts.get(fingerprint, 0)
+        self._event_counts[fingerprint] = count + 1
+        self._last_events[fingerprint] = now
+
+        return fingerprint
+
+    def reset(self, fingerprint: str) -> None:
+        """Reset tracking for a fingerprint."""
+        self._event_counts.pop(fingerprint, None)
+        self._last_events.pop(fingerprint, None)

truthound_dashboard/core/notifications/dispatcher.py

@@ -44,6 +44,7 @@ from .base import (
 )
 from .events import (
     DriftDetectedEvent,
+    SchemaChangedEvent,
     ScheduleFailedEvent,
     TestNotificationEvent,
     ValidationFailedEvent,
@@ -159,6 +160,7 @@ class NotificationDispatcher:
             "validation_failed": ["validation_failed", "critical_issues", "high_issues"],
             "schedule_failed": ["schedule_failed", "validation_failed"],
             "drift_detected": ["drift_detected"],
+            "schema_changed": ["schema_changed", "breaking_schema_change"],
             "test": [],
         }
 
@@ -202,6 +204,11 @@ class NotificationDispatcher:
             # Could add threshold checks here from rule.condition_config
             pass
 
+        elif isinstance(event, SchemaChangedEvent):
+            # Check for breaking change condition
+            if rule.condition == "breaking_schema_change" and not event.has_breaking_changes:
+                return False
+
         return True
 
     async def _send_to_channel(
@@ -391,6 +398,42 @@ class NotificationDispatcher:
 
         return await self.dispatch(event)
 
+    async def notify_schema_changed(
+        self,
+        source_id: str,
+        source_name: str,
+        from_version: int | None,
+        to_version: int,
+        total_changes: int,
+        breaking_changes: int = 0,
+        changes: list[dict[str, Any]] | None = None,
+    ) -> list[NotificationResult]:
+        """Send notifications for schema changes.
+
+        Args:
+            source_id: ID of the source with schema changes.
+            source_name: Name of the source.
+            from_version: Previous schema version (None if first version).
+            to_version: New schema version.
+            total_changes: Total number of changes detected.
+            breaking_changes: Number of breaking changes.
+            changes: Optional list of change details.
+
+        Returns:
+            List of delivery results.
+        """
+        event = SchemaChangedEvent(
+            source_id=source_id,
+            source_name=source_name,
+            from_version=from_version,
+            to_version=to_version,
+            total_changes=total_changes,
+            breaking_changes=breaking_changes,
+            changes=changes or [],
+        )
+
+        return await self.dispatch(event)
+
     async def test_channel(self, channel_id: str) -> NotificationResult:
         """Send a test notification to a specific channel.
 

truthound_dashboard/core/notifications/escalation/__init__.py

@@ -0,0 +1,149 @@
+"""Escalation engine for notification management.
+
+This module provides a multi-level escalation system for managing
+alerts that require attention at different organizational levels.
+
+Features:
+    - Multi-level escalation policies
+    - State machine for incident tracking
+    - Configurable delay between escalation levels
+    - Support for user, group, and on-call targets
+    - Auto-resolution on success
+
+State Machine:
+    PENDING -> TRIGGERED -> ESCALATED -> RESOLVED
+                  |             ^
+                  v             |
+             ACKNOWLEDGED ------+
+
+Example:
+    from truthound_dashboard.core.notifications.escalation import (
+        EscalationEngine,
+        EscalationPolicy,
+        EscalationLevel,
+        EscalationTarget,
+        TargetType,
+    )
+
+    policy = EscalationPolicy(
+        name="critical_alerts",
+        levels=[
+            EscalationLevel(
+                level=1,
+                delay_minutes=0,
+                targets=[
+                    EscalationTarget(type=TargetType.USER, identifier="team-lead", channel="slack")
+                ],
+            ),
+            EscalationLevel(
+                level=2,
+                delay_minutes=15,
+                targets=[
+                    EscalationTarget(type=TargetType.GROUP, identifier="managers", channel="pagerduty")
+                ],
+            ),
+        ],
+    )
+
+    engine = EscalationEngine(policy=policy)
+    await engine.trigger("incident-123", context={"severity": "critical"})
+"""
+
+from .engine import EscalationEngine, EscalationEngineConfig
+from .models import (
+    EscalationEvent,
+    EscalationIncident,
+    EscalationLevel,
+    EscalationPolicy,
+    EscalationState,
+    EscalationTarget,
+    StateTransition,
+    TargetType,
+)
+from .backends import (
+    BackendType,
+    InMemorySchedulerBackend,
+    JobData,
+    JobExecutionResult,
+    JobState,
+    MisfirePolicy,
+    SchedulerBackend,
+    SchedulerBackendConfig,
+    SQLAlchemySchedulerBackend,
+    create_scheduler_backend,
+)
+from .scheduler import (
+    DefaultEscalationHandler,
+    EscalationHandler,
+    EscalationResult,
+    EscalationSchedulerConfig,
+    EscalationSchedulerService,
+    EscalationStrategy,
+    ImmediateEscalationStrategy,
+    LoggingEscalationHandler,
+    TimeBasedEscalationStrategy,
+    get_escalation_scheduler,
+    reset_escalation_scheduler,
+    start_escalation_scheduler,
+    stop_escalation_scheduler,
+)
+from .state_machine import EscalationStateMachine
+from .stores import (
+    BaseEscalationStore,
+    EscalationMetrics,
+    EscalationStoreType,
+    InMemoryEscalationStore,
+    RedisEscalationStore,
+    SQLiteEscalationStore,
+    create_escalation_store,
+)
+
+__all__ = [
+    # Models
+    "EscalationState",
+    "TargetType",
+    "EscalationTarget",
+    "EscalationLevel",
+    "EscalationPolicy",
+    "EscalationIncident",
+    "EscalationEvent",
+    "StateTransition",
+    # State Machine
+    "EscalationStateMachine",
+    # Stores
+    "BaseEscalationStore",
+    "EscalationMetrics",
+    "EscalationStoreType",
+    "InMemoryEscalationStore",
+    "RedisEscalationStore",
+    "SQLiteEscalationStore",
+    "create_escalation_store",
+    # Engine
+    "EscalationEngine",
+    "EscalationEngineConfig",
+    # Scheduler Backends
+    "BackendType",
+    "JobState",
+    "MisfirePolicy",
+    "SchedulerBackendConfig",
+    "JobData",
+    "JobExecutionResult",
+    "SchedulerBackend",
+    "InMemorySchedulerBackend",
+    "SQLAlchemySchedulerBackend",
+    "create_scheduler_backend",
+    # Scheduler
+    "EscalationSchedulerService",
+    "EscalationSchedulerConfig",
+    "EscalationHandler",
+    "EscalationResult",
+    "EscalationStrategy",
+    "DefaultEscalationHandler",
+    "LoggingEscalationHandler",
+    "TimeBasedEscalationStrategy",
+    "ImmediateEscalationStrategy",
+    "get_escalation_scheduler",
+    "reset_escalation_scheduler",
+    "start_escalation_scheduler",
+    "stop_escalation_scheduler",
+]