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

truthound_dashboard/main.py

@@ -46,7 +46,11 @@ from truthound_dashboard.config import get_settings
 from truthound_dashboard.core.cache import get_cache
 from truthound_dashboard.core.logging import setup_logging
 from truthound_dashboard.core.maintenance import get_maintenance_manager
+from truthound_dashboard.core.notifications.escalation.scheduler import (
+    get_escalation_scheduler,
+)
 from truthound_dashboard.core.scheduler import get_scheduler
+from truthound_dashboard.core.websocket import get_websocket_manager
 from truthound_dashboard.db import init_db
 
 logger = logging.getLogger(__name__)
@@ -94,6 +98,16 @@ async def lifespan(_app: FastAPI) -> AsyncGenerator[None, None]:
     await scheduler.start()
     logger.info("Validation scheduler started")
 
+    # Start escalation scheduler
+    escalation_scheduler = get_escalation_scheduler()
+    await escalation_scheduler.start()
+    logger.info("Escalation scheduler started")
+
+    # Start WebSocket manager
+    ws_manager = get_websocket_manager()
+    await ws_manager.start()
+    logger.info("WebSocket manager started")
+
     # Register maintenance tasks with scheduler
     _register_maintenance_tasks()
 
@@ -102,6 +116,14 @@ async def lifespan(_app: FastAPI) -> AsyncGenerator[None, None]:
     # Shutdown
     logger.info("Shutting down Truthound Dashboard")
 
+    # Stop WebSocket manager
+    await ws_manager.stop()
+    logger.info("WebSocket manager stopped")
+
+    # Stop escalation scheduler
+    await escalation_scheduler.stop()
+    logger.info("Escalation scheduler stopped")
+
     # Stop scheduler
     await scheduler.stop()
     logger.info("Scheduler stopped")

truthound_dashboard/schemas/__init__.py

@@ -41,7 +41,19 @@ from .history import (
     RecentValidation,
     TrendDataPoint,
 )
-from .profile import ColumnProfile, ProfileRequest, ProfileResponse
+from .profile import (
+    ColumnProfile,
+    DetectedPattern,
+    HistogramBucket,
+    PatternDetectionConfig,
+    PatternType,
+    ProfileRequest,
+    ProfileResponse,
+    SamplingConfig,
+    SamplingMetadata,
+    SamplingStrategy,
+    SamplingStrategyType,
+)
 from .rule import (
     RuleBase,
     RuleCreate,
@@ -148,13 +160,22 @@ from .schema import (
     SchemaUpdateRequest,
 )
 from .source import (
+    FieldDefinitionSchema,
+    FieldOption,
+    FieldType,
     SourceBase,
+    SourceCategory,
     SourceCreate,
     SourceListResponse,
     SourceResponse,
     SourceSummary,
     SourceType,
+    SourceTypeCategorySchema,
+    SourceTypeDefinitionSchema,
+    SourceTypesResponse,
     SourceUpdate,
+    TestConnectionRequest,
+    TestConnectionResponse,
 )
 from .validation import (
     IssueSeverity,
@@ -166,6 +187,194 @@ from .validation import (
     ValidationStatus,
     ValidationSummary,
 )
+# Reports
+from .reports import (
+    AvailableFormatsResponse,
+    ReportFormatType,
+    ReportGenerateRequest,
+    ReportMetadataResponse,
+    ReportResponse,
+    ReportThemeType,
+)
+# Maintenance
+from .maintenance import (
+    CacheClearRequest,
+    CacheStatsResponse,
+    CleanupResultItem,
+    CleanupTriggerRequest,
+    MaintenanceReportResponse,
+    MaintenanceStatusResponse,
+    RetentionPolicyConfig,
+    RetentionPolicyResponse,
+)
+# Schema Evolution
+from .schema_evolution import (
+    SchemaChangeListResponse,
+    SchemaChangeResponse,
+    SchemaChangeSeverity,
+    SchemaChangeType,
+    SchemaEvolutionRequest,
+    SchemaEvolutionResponse,
+    SchemaEvolutionSummary,
+    SchemaVersionListResponse,
+    SchemaVersionResponse,
+    SchemaVersionSummary,
+)
+# Rule Suggestion
+from .rule_suggestion import (
+    ApplyRulesRequest,
+    ApplyRulesResponse,
+    CrossColumnRuleSuggestion,
+    CrossColumnRuleType,
+    ExportRulesRequest,
+    ExportRulesResponse,
+    PresetInfo,
+    PresetsResponse,
+    RuleCategory,
+    RuleExportFormat,
+    RulePreset,
+    RuleSuggestionRequest,
+    RuleSuggestionResponse,
+    StrictnessLevel,
+    SuggestedRule,
+    SuggestionStats,
+)
+# Profile Comparison
+from .profile_comparison import (
+    ColumnComparison,
+    ColumnComparisonSummary,
+    ColumnTrend,
+    LatestComparisonResponse,
+    ProfileComparisonRequest,
+    ProfileComparisonResponse,
+    ProfileListResponse,
+    ProfileSummary,
+    ProfileTrendPoint,
+    ProfileTrendRequest,
+    ProfileTrendResponse,
+    TrendDirection,
+)
+# Advanced Notifications
+from .notifications_advanced import (
+    AcknowledgeRequest,
+    AdvancedNotificationStats,
+    DeduplicationConfigCreate,
+    DeduplicationConfigListResponse,
+    DeduplicationConfigResponse,
+    DeduplicationConfigUpdate,
+    DeduplicationPolicy,
+    DeduplicationStats,
+    DeduplicationStrategy,
+    EscalationEventBase,
+    EscalationIncidentBase,
+    EscalationIncidentListResponse,
+    EscalationIncidentResponse,
+    EscalationLevelBase,
+    EscalationPolicyBase,
+    EscalationPolicyCreate,
+    EscalationPolicyListResponse,
+    EscalationPolicyResponse,
+    EscalationPolicyUpdate,
+    EscalationState,
+    EscalationStats,
+    EscalationTargetBase,
+    ResolveRequest,
+    RoutingRuleCreate,
+    RoutingRuleListResponse,
+    RoutingRuleResponse,
+    RoutingRuleUpdate,
+    RuleConfig,
+    RuleType,
+    RuleTypeInfo,
+    RuleTypesResponse,
+    TargetType,
+    ThrottlingConfigCreate,
+    ThrottlingConfigListResponse,
+    ThrottlingConfigResponse,
+    ThrottlingConfigUpdate,
+    ThrottlingStats,
+)
+# OpenLineage Export
+from .openlineage import (
+    OpenLineageDataset,
+    OpenLineageEmitRequest,
+    OpenLineageEmitResponse,
+    OpenLineageEvent,
+    OpenLineageExportFormat,
+    OpenLineageExportRequest,
+    OpenLineageExportResponse,
+    OpenLineageJob,
+    OpenLineageRun,
+    OpenLineageWebhookConfig,
+    RunState,
+    build_dataset_namespace,
+)
+# Cross-Alerts
+from .cross_alerts import (
+    AlertType,
+    AutoTriggerConfig,
+    AutoTriggerConfigCreate,
+    AutoTriggerConfigUpdate,
+    AutoTriggerEvent,
+    AutoTriggerEventListResponse,
+    AutoTriggerThresholds,
+    CorrelatedAlert,
+    CorrelationStrength,
+    CrossAlertCorrelation,
+    CrossAlertCorrelationListResponse,
+    CrossAlertSummary,
+    TriggerDirection,
+)
+# Triggers
+from .triggers import (
+    CompositeTriggerConfig,
+    CronTriggerConfig,
+    DataChangeTriggerConfig,
+    EventTriggerConfig,
+    IntervalTriggerConfig,
+    ManualTriggerConfig,
+    TriggerConfig,
+    TriggerConfigCreate,
+    TriggerEvaluationResult,
+    TriggerOperator,
+    TriggerResponse,
+    TriggerState,
+    TriggerType,
+    parse_trigger_config,
+)
+# Plugins
+from .plugins import (
+    CustomReporterCreate,
+    CustomReporterListResponse,
+    CustomReporterResponse,
+    CustomReporterUpdate,
+    CustomValidatorCreate,
+    CustomValidatorListResponse,
+    CustomValidatorResponse,
+    CustomValidatorUpdate,
+    MarketplaceSearchRequest,
+    MarketplaceStats,
+    PluginCreate,
+    PluginInstallRequest,
+    PluginInstallResponse,
+    PluginListResponse,
+    PluginPermission,
+    PluginResponse,
+    PluginSource,
+    PluginStatus,
+    PluginSummary,
+    PluginType,
+    PluginUninstallRequest,
+    PluginUninstallResponse,
+    PluginUpdate,
+    PluginUpdateCheckResponse,
+    ReporterGenerateRequest,
+    ReporterGenerateResponse,
+    ReporterOutputFormat,
+    SecurityLevel,
+    ValidatorTestRequest,
+    ValidatorTestResponse,
+)
 
 __all__ = [
     # Base
@@ -178,12 +387,22 @@ __all__ = [
     "MessageResponse",
     # Source
     "SourceType",
+    "SourceCategory",
+    "FieldType",
     "SourceBase",
     "SourceCreate",
     "SourceUpdate",
     "SourceResponse",
     "SourceListResponse",
     "SourceSummary",
+    # Source Type Definitions (dynamic forms)
+    "FieldOption",
+    "FieldDefinitionSchema",
+    "SourceTypeDefinitionSchema",
+    "SourceTypeCategorySchema",
+    "SourceTypesResponse",
+    "TestConnectionRequest",
+    "TestConnectionResponse",
     # Rule
     "RuleBase",
     "RuleCreate",
@@ -211,6 +430,14 @@ __all__ = [
     "ProfileRequest",
     "ColumnProfile",
     "ProfileResponse",
+    "DetectedPattern",
+    "HistogramBucket",
+    "PatternDetectionConfig",
+    "PatternType",
+    "SamplingConfig",
+    "SamplingMetadata",
+    "SamplingStrategy",
+    "SamplingStrategyType",
     # Data Masking
     "MaskingStrategy",
     "MaskingStrategyLiteral",
@@ -312,4 +539,172 @@ __all__ = [
     "ActivityResponse",
     "ActivityListResponse",
     "ActivityCreate",
+    # Reports
+    "ReportFormatType",
+    "ReportThemeType",
+    "ReportGenerateRequest",
+    "ReportMetadataResponse",
+    "ReportResponse",
+    "AvailableFormatsResponse",
+    # Maintenance
+    "RetentionPolicyConfig",
+    "RetentionPolicyResponse",
+    "CleanupResultItem",
+    "MaintenanceReportResponse",
+    "CleanupTriggerRequest",
+    "MaintenanceStatusResponse",
+    "CacheStatsResponse",
+    "CacheClearRequest",
+    # Schema Evolution
+    "SchemaChangeType",
+    "SchemaChangeSeverity",
+    "SchemaVersionResponse",
+    "SchemaVersionSummary",
+    "SchemaVersionListResponse",
+    "SchemaChangeResponse",
+    "SchemaChangeListResponse",
+    "SchemaEvolutionRequest",
+    "SchemaEvolutionResponse",
+    "SchemaEvolutionSummary",
+    # Rule Suggestion
+    "SuggestedRule",
+    "CrossColumnRuleSuggestion",
+    "CrossColumnRuleType",
+    "RuleSuggestionRequest",
+    "RuleSuggestionResponse",
+    "ApplyRulesRequest",
+    "ApplyRulesResponse",
+    "SuggestionStats",
+    "RuleCategory",
+    "RulePreset",
+    "StrictnessLevel",
+    "RuleExportFormat",
+    "ExportRulesRequest",
+    "ExportRulesResponse",
+    "PresetInfo",
+    "PresetsResponse",
+    # Profile Comparison
+    "TrendDirection",
+    "ProfileSummary",
+    "ProfileListResponse",
+    "ColumnComparison",
+    "ColumnComparisonSummary",
+    "ProfileComparisonRequest",
+    "ProfileComparisonResponse",
+    "ProfileTrendPoint",
+    "ColumnTrend",
+    "ProfileTrendRequest",
+    "ProfileTrendResponse",
+    "LatestComparisonResponse",
+    # Advanced Notifications
+    "RuleType",
+    "DeduplicationStrategy",
+    "DeduplicationPolicy",
+    "EscalationState",
+    "TargetType",
+    "RuleConfig",
+    "RoutingRuleCreate",
+    "RoutingRuleUpdate",
+    "RoutingRuleResponse",
+    "RoutingRuleListResponse",
+    "RuleTypeInfo",
+    "RuleTypesResponse",
+    "DeduplicationConfigCreate",
+    "DeduplicationConfigUpdate",
+    "DeduplicationConfigResponse",
+    "DeduplicationConfigListResponse",
+    "DeduplicationStats",
+    "ThrottlingConfigCreate",
+    "ThrottlingConfigUpdate",
+    "ThrottlingConfigResponse",
+    "ThrottlingConfigListResponse",
+    "ThrottlingStats",
+    "EscalationTargetBase",
+    "EscalationLevelBase",
+    "EscalationPolicyBase",
+    "EscalationPolicyCreate",
+    "EscalationPolicyUpdate",
+    "EscalationPolicyResponse",
+    "EscalationPolicyListResponse",
+    "EscalationEventBase",
+    "EscalationIncidentBase",
+    "EscalationIncidentResponse",
+    "EscalationIncidentListResponse",
+    "AcknowledgeRequest",
+    "ResolveRequest",
+    "EscalationStats",
+    "AdvancedNotificationStats",
+    # OpenLineage Export
+    "RunState",
+    "OpenLineageDataset",
+    "OpenLineageJob",
+    "OpenLineageRun",
+    "OpenLineageEvent",
+    "OpenLineageExportFormat",
+    "OpenLineageExportRequest",
+    "OpenLineageExportResponse",
+    "OpenLineageWebhookConfig",
+    "OpenLineageEmitRequest",
+    "OpenLineageEmitResponse",
+    "build_dataset_namespace",
+    # Cross-Alerts
+    "AlertType",
+    "CorrelationStrength",
+    "TriggerDirection",
+    "CorrelatedAlert",
+    "CrossAlertCorrelation",
+    "CrossAlertCorrelationListResponse",
+    "AutoTriggerThresholds",
+    "AutoTriggerConfig",
+    "AutoTriggerConfigCreate",
+    "AutoTriggerConfigUpdate",
+    "AutoTriggerEvent",
+    "AutoTriggerEventListResponse",
+    "CrossAlertSummary",
+    # Triggers
+    "TriggerType",
+    "TriggerOperator",
+    "CronTriggerConfig",
+    "IntervalTriggerConfig",
+    "DataChangeTriggerConfig",
+    "EventTriggerConfig",
+    "CompositeTriggerConfig",
+    "ManualTriggerConfig",
+    "TriggerConfig",
+    "TriggerState",
+    "TriggerEvaluationResult",
+    "TriggerConfigCreate",
+    "TriggerResponse",
+    "parse_trigger_config",
+    # Plugins
+    "PluginType",
+    "PluginStatus",
+    "PluginSource",
+    "SecurityLevel",
+    "PluginPermission",
+    "ReporterOutputFormat",
+    "PluginCreate",
+    "PluginUpdate",
+    "PluginResponse",
+    "PluginListResponse",
+    "PluginSummary",
+    "PluginInstallRequest",
+    "PluginInstallResponse",
+    "PluginUninstallRequest",
+    "PluginUninstallResponse",
+    "PluginUpdateCheckResponse",
+    "CustomValidatorCreate",
+    "CustomValidatorUpdate",
+    "CustomValidatorResponse",
+    "CustomValidatorListResponse",
+    "ValidatorTestRequest",
+    "ValidatorTestResponse",
+    "CustomReporterCreate",
+    "CustomReporterUpdate",
+    "CustomReporterResponse",
+    "CustomReporterListResponse",
+    "ReporterGenerateRequest",
+    "ReporterGenerateResponse",
+    "MarketplaceSearchRequest",
+    "MarketplaceStats",
 ]