truthound-dashboard 1.0.0__py3-none-any.whl

truthound_dashboard/api/rules.py

@@ -0,0 +1,277 @@
+"""Rules API endpoints.
+
+This module provides endpoints for managing custom validation rules.
+"""
+
+from __future__ import annotations
+
+from typing import Annotated
+
+from fastapi import APIRouter, HTTPException, Path, Query
+
+from truthound_dashboard.schemas import (
+    MessageResponse,
+    RuleCreate,
+    RuleListItem,
+    RuleListResponse,
+    RuleResponse,
+    RuleUpdate,
+)
+
+from .deps import RuleServiceDep, SourceServiceDep
+
+router = APIRouter()
+
+
+@router.get(
+    "/sources/{source_id}/rules",
+    response_model=RuleListResponse,
+    summary="List source rules",
+    description="Get all validation rules for a data source",
+)
+async def list_source_rules(
+    service: RuleServiceDep,
+    source_service: SourceServiceDep,
+    source_id: Annotated[str, Path(description="Source ID")],
+    limit: Annotated[int, Query(ge=1, le=100, description="Maximum items")] = 50,
+    active_only: Annotated[bool, Query(description="Only return active rules")] = False,
+) -> RuleListResponse:
+    """List all rules for a source.
+
+    Args:
+        service: Injected rule service.
+        source_service: Injected source service.
+        source_id: Source to get rules for.
+        limit: Maximum rules to return.
+        active_only: Only return active rules.
+
+    Returns:
+        List of rules.
+
+    Raises:
+        HTTPException: 404 if source not found.
+    """
+    # Verify source exists
+    source = await source_service.get_by_id(source_id)
+    if source is None:
+        raise HTTPException(status_code=404, detail="Source not found")
+
+    rules = await service.get_rules_for_source(
+        source_id,
+        limit=limit,
+        active_only=active_only,
+    )
+
+    return RuleListResponse(
+        data=[RuleListItem.from_model(r) for r in rules],
+        total=len(rules),
+        limit=limit,
+    )
+
+
+@router.get(
+    "/sources/{source_id}/rules/active",
+    response_model=RuleResponse | None,
+    summary="Get active rule",
+    description="Get the currently active rule for a source",
+)
+async def get_active_rule(
+    service: RuleServiceDep,
+    source_service: SourceServiceDep,
+    source_id: Annotated[str, Path(description="Source ID")],
+) -> RuleResponse | None:
+    """Get the active rule for a source.
+
+    Args:
+        service: Injected rule service.
+        source_service: Injected source service.
+        source_id: Source to get active rule for.
+
+    Returns:
+        Active rule or None.
+
+    Raises:
+        HTTPException: 404 if source not found.
+    """
+    # Verify source exists
+    source = await source_service.get_by_id(source_id)
+    if source is None:
+        raise HTTPException(status_code=404, detail="Source not found")
+
+    rule = await service.get_active_rule(source_id)
+    if rule is None:
+        return None
+    return RuleResponse.from_model(rule)
+
+
+@router.post(
+    "/sources/{source_id}/rules",
+    response_model=RuleResponse,
+    status_code=201,
+    summary="Create rule",
+    description="Create a new validation rule for a source",
+)
+async def create_rule(
+    service: RuleServiceDep,
+    source_service: SourceServiceDep,
+    source_id: Annotated[str, Path(description="Source ID")],
+    rule: RuleCreate,
+    activate: Annotated[bool, Query(description="Activate this rule")] = True,
+) -> RuleResponse:
+    """Create a new rule for a source.
+
+    Args:
+        service: Injected rule service.
+        source_service: Injected source service.
+        source_id: Source to create rule for.
+        rule: Rule creation data.
+        activate: Whether to make this the active rule.
+
+    Returns:
+        Created rule.
+
+    Raises:
+        HTTPException: 404 if source not found, 400 if YAML is invalid.
+    """
+    # Verify source exists
+    source = await source_service.get_by_id(source_id)
+    if source is None:
+        raise HTTPException(status_code=404, detail="Source not found")
+
+    try:
+        created = await service.create_rule(
+            source_id,
+            rules_yaml=rule.rules_yaml,
+            name=rule.name,
+            description=rule.description,
+            activate=activate,
+        )
+        return RuleResponse.from_model(created)
+    except ValueError as e:
+        raise HTTPException(status_code=400, detail=str(e))
+
+
+@router.get(
+    "/rules/{rule_id}",
+    response_model=RuleResponse,
+    summary="Get rule",
+    description="Get a specific rule by ID",
+)
+async def get_rule(
+    service: RuleServiceDep,
+    rule_id: Annotated[str, Path(description="Rule ID")],
+) -> RuleResponse:
+    """Get a specific rule.
+
+    Args:
+        service: Injected rule service.
+        rule_id: Rule unique identifier.
+
+    Returns:
+        Rule details.
+
+    Raises:
+        HTTPException: 404 if rule not found.
+    """
+    rule = await service.get_rule(rule_id)
+    if rule is None:
+        raise HTTPException(status_code=404, detail="Rule not found")
+    return RuleResponse.from_model(rule)
+
+
+@router.put(
+    "/rules/{rule_id}",
+    response_model=RuleResponse,
+    summary="Update rule",
+    description="Update an existing rule",
+)
+async def update_rule(
+    service: RuleServiceDep,
+    rule_id: Annotated[str, Path(description="Rule ID")],
+    update: RuleUpdate,
+) -> RuleResponse:
+    """Update an existing rule.
+
+    Args:
+        service: Injected rule service.
+        rule_id: Rule unique identifier.
+        update: Update data.
+
+    Returns:
+        Updated rule.
+
+    Raises:
+        HTTPException: 404 if rule not found, 400 if YAML is invalid.
+    """
+    try:
+        updated = await service.update_rule(
+            rule_id,
+            name=update.name,
+            description=update.description,
+            rules_yaml=update.rules_yaml,
+            version=update.version,
+            is_active=update.is_active,
+        )
+        if updated is None:
+            raise HTTPException(status_code=404, detail="Rule not found")
+        return RuleResponse.from_model(updated)
+    except ValueError as e:
+        raise HTTPException(status_code=400, detail=str(e))
+
+
+@router.delete(
+    "/rules/{rule_id}",
+    response_model=MessageResponse,
+    summary="Delete rule",
+    description="Delete a rule",
+)
+async def delete_rule(
+    service: RuleServiceDep,
+    rule_id: Annotated[str, Path(description="Rule ID")],
+) -> MessageResponse:
+    """Delete a rule.
+
+    Args:
+        service: Injected rule service.
+        rule_id: Rule unique identifier.
+
+    Returns:
+        Success message.
+
+    Raises:
+        HTTPException: 404 if rule not found.
+    """
+    deleted = await service.delete_rule(rule_id)
+    if not deleted:
+        raise HTTPException(status_code=404, detail="Rule not found")
+    return MessageResponse(message="Rule deleted successfully")
+
+
+@router.post(
+    "/rules/{rule_id}/activate",
+    response_model=RuleResponse,
+    summary="Activate rule",
+    description="Activate a rule and deactivate others for the same source",
+)
+async def activate_rule(
+    service: RuleServiceDep,
+    rule_id: Annotated[str, Path(description="Rule ID")],
+) -> RuleResponse:
+    """Activate a rule.
+
+    This will deactivate any other active rules for the same source.
+
+    Args:
+        service: Injected rule service.
+        rule_id: Rule unique identifier.
+
+    Returns:
+        Activated rule.
+
+    Raises:
+        HTTPException: 404 if rule not found.
+    """
+    rule = await service.activate_rule(rule_id)
+    if rule is None:
+        raise HTTPException(status_code=404, detail="Rule not found")
+    return RuleResponse.from_model(rule)

truthound_dashboard/api/schedules.py

@@ -0,0 +1,329 @@
+"""Schedule management API endpoints.
+
+Provides CRUD endpoints for validation schedules.
+"""
+
+from __future__ import annotations
+
+from typing import Annotated
+
+from fastapi import APIRouter, Depends, HTTPException, Query
+
+from truthound_dashboard.core import ScheduleService, ValidationService
+from truthound_dashboard.schemas import (
+    ScheduleActionResponse,
+    ScheduleCreate,
+    ScheduleListResponse,
+    ScheduleUpdate,
+)
+
+from .deps import SessionDep
+
+router = APIRouter()
+
+
+async def get_schedule_service(session: SessionDep) -> ScheduleService:
+    """Get schedule service dependency."""
+    return ScheduleService(session)
+
+
+ScheduleServiceDep = Annotated[ScheduleService, Depends(get_schedule_service)]
+
+
+def _schedule_to_response(schedule) -> dict:
+    """Convert schedule model to response dict."""
+    return {
+        "id": schedule.id,
+        "name": schedule.name,
+        "source_id": schedule.source_id,
+        "cron_expression": schedule.cron_expression,
+        "is_active": schedule.is_active,
+        "notify_on_failure": schedule.notify_on_failure,
+        "last_run_at": (
+            schedule.last_run_at.isoformat() if schedule.last_run_at else None
+        ),
+        "next_run_at": (
+            schedule.next_run_at.isoformat() if schedule.next_run_at else None
+        ),
+        "config": schedule.config,
+        "created_at": schedule.created_at.isoformat() if schedule.created_at else None,
+        "updated_at": schedule.updated_at.isoformat() if schedule.updated_at else None,
+    }
+
+
+@router.get(
+    "/schedules",
+    response_model=ScheduleListResponse,
+    summary="List schedules",
+    description="List all validation schedules.",
+)
+async def list_schedules(
+    service: ScheduleServiceDep,
+    source_id: str | None = Query(None, description="Filter by source ID"),
+    active_only: bool = Query(False, description="Only return active schedules"),
+    limit: int = Query(100, ge=1, le=500, description="Maximum results"),
+) -> ScheduleListResponse:
+    """List validation schedules.
+
+    Args:
+        service: Schedule service.
+        source_id: Optional source ID filter.
+        active_only: Only return active schedules.
+        limit: Maximum results.
+
+    Returns:
+        List of schedules.
+    """
+    schedules = await service.list_schedules(
+        source_id=source_id,
+        active_only=active_only,
+        limit=limit,
+    )
+
+    return ScheduleListResponse(
+        success=True,
+        data=[_schedule_to_response(s) for s in schedules],
+        total=len(schedules),
+    )
+
+
+@router.post(
+    "/schedules",
+    response_model=dict,
+    status_code=201,
+    summary="Create schedule",
+    description="Create a new validation schedule.",
+)
+async def create_schedule(
+    request: ScheduleCreate,
+    service: ScheduleServiceDep,
+) -> dict:
+    """Create a new schedule.
+
+    Args:
+        request: Schedule creation request.
+        service: Schedule service.
+
+    Returns:
+        Created schedule.
+    """
+    try:
+        schedule = await service.create_schedule(
+            source_id=request.source_id,
+            name=request.name,
+            cron_expression=request.cron_expression,
+            notify_on_failure=request.notify_on_failure,
+            config=request.config,
+        )
+
+        return {
+            "success": True,
+            "data": _schedule_to_response(schedule),
+        }
+    except ValueError as e:
+        raise HTTPException(status_code=400, detail=str(e))
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
+
+@router.get(
+    "/schedules/{schedule_id}",
+    response_model=dict,
+    summary="Get schedule",
+    description="Get a specific schedule by ID.",
+)
+async def get_schedule(
+    schedule_id: str,
+    service: ScheduleServiceDep,
+) -> dict:
+    """Get a schedule by ID.
+
+    Args:
+        schedule_id: Schedule ID.
+        service: Schedule service.
+
+    Returns:
+        Schedule details.
+    """
+    schedule = await service.get_schedule(schedule_id)
+    if schedule is None:
+        raise HTTPException(status_code=404, detail="Schedule not found")
+
+    return {
+        "success": True,
+        "data": _schedule_to_response(schedule),
+    }
+
+
+@router.put(
+    "/schedules/{schedule_id}",
+    response_model=dict,
+    summary="Update schedule",
+    description="Update an existing schedule.",
+)
+async def update_schedule(
+    schedule_id: str,
+    request: ScheduleUpdate,
+    service: ScheduleServiceDep,
+) -> dict:
+    """Update a schedule.
+
+    Args:
+        schedule_id: Schedule ID.
+        request: Update request.
+        service: Schedule service.
+
+    Returns:
+        Updated schedule.
+    """
+    try:
+        schedule = await service.update_schedule(
+            schedule_id,
+            name=request.name,
+            cron_expression=request.cron_expression,
+            notify_on_failure=request.notify_on_failure,
+            config=request.config,
+        )
+
+        if schedule is None:
+            raise HTTPException(status_code=404, detail="Schedule not found")
+
+        return {
+            "success": True,
+            "data": _schedule_to_response(schedule),
+        }
+    except ValueError as e:
+        raise HTTPException(status_code=400, detail=str(e))
+
+
+@router.delete(
+    "/schedules/{schedule_id}",
+    response_model=dict,
+    summary="Delete schedule",
+    description="Delete a schedule.",
+)
+async def delete_schedule(
+    schedule_id: str,
+    service: ScheduleServiceDep,
+) -> dict:
+    """Delete a schedule.
+
+    Args:
+        schedule_id: Schedule ID.
+        service: Schedule service.
+
+    Returns:
+        Success message.
+    """
+    deleted = await service.delete_schedule(schedule_id)
+    if not deleted:
+        raise HTTPException(status_code=404, detail="Schedule not found")
+
+    return {"success": True, "message": "Schedule deleted"}
+
+
+@router.post(
+    "/schedules/{schedule_id}/pause",
+    response_model=ScheduleActionResponse,
+    summary="Pause schedule",
+    description="Pause a schedule.",
+)
+async def pause_schedule(
+    schedule_id: str,
+    service: ScheduleServiceDep,
+) -> ScheduleActionResponse:
+    """Pause a schedule.
+
+    Args:
+        schedule_id: Schedule ID.
+        service: Schedule service.
+
+    Returns:
+        Action result with updated schedule.
+    """
+    schedule = await service.pause_schedule(schedule_id)
+    if schedule is None:
+        raise HTTPException(status_code=404, detail="Schedule not found")
+
+    return ScheduleActionResponse(
+        success=True,
+        message="Schedule paused",
+        schedule=_schedule_to_response(schedule),
+    )
+
+
+@router.post(
+    "/schedules/{schedule_id}/resume",
+    response_model=ScheduleActionResponse,
+    summary="Resume schedule",
+    description="Resume a paused schedule.",
+)
+async def resume_schedule(
+    schedule_id: str,
+    service: ScheduleServiceDep,
+) -> ScheduleActionResponse:
+    """Resume a paused schedule.
+
+    Args:
+        schedule_id: Schedule ID.
+        service: Schedule service.
+
+    Returns:
+        Action result with updated schedule.
+    """
+    schedule = await service.resume_schedule(schedule_id)
+    if schedule is None:
+        raise HTTPException(status_code=404, detail="Schedule not found")
+
+    return ScheduleActionResponse(
+        success=True,
+        message="Schedule resumed",
+        schedule=_schedule_to_response(schedule),
+    )
+
+
+@router.post(
+    "/schedules/{schedule_id}/run",
+    response_model=dict,
+    summary="Run schedule now",
+    description="Trigger immediate execution of a scheduled validation.",
+)
+async def run_schedule_now(
+    schedule_id: str,
+    service: ScheduleServiceDep,
+    session: SessionDep,
+) -> dict:
+    """Run a scheduled validation immediately.
+
+    Args:
+        schedule_id: Schedule ID.
+        service: Schedule service.
+        session: Database session.
+
+    Returns:
+        Validation result.
+    """
+    schedule = await service.get_schedule(schedule_id)
+    if schedule is None:
+        raise HTTPException(status_code=404, detail="Schedule not found")
+
+    # Run validation using ValidationService
+    validation_service = ValidationService(session)
+    config = schedule.config or {}
+
+    try:
+        validation = await validation_service.run_validation(
+            schedule.source_id,
+            validators=config.get("validators"),
+            schema_path=config.get("schema_path"),
+            auto_schema=config.get("auto_schema", False),
+        )
+
+        return {
+            "success": True,
+            "message": "Validation triggered",
+            "validation_id": validation.id,
+            "passed": validation.passed,
+        }
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))