aegis-stack 0.2.0rc2__py3-none-any.whl

aegis/templates/cookiecutter-aegis-project/{{cookiecutter.project_slug}}/app/cli/tasks.py.j2

@@ -0,0 +1,328 @@
+{%- if cookiecutter.scheduler_backend != "memory" -%}
+"""
+Scheduled task CLI commands.
+
+Provides command-line interface for managing scheduled jobs,
+viewing job statistics, and manually triggering tasks.
+"""
+
+import asyncio
+import json
+from datetime import datetime
+from typing import Any
+
+from rich import print as rprint
+from rich.console import Console
+from rich.panel import Panel
+from rich.table import Table
+import typer
+
+from app.core.log import logger
+from app.services.scheduler import ScheduledTaskManager
+from app.services.scheduler.models import ScheduledTask
+
+app = typer.Typer(
+    name="tasks",
+    help="Scheduled task management commands",
+    no_args_is_help=True,
+)
+
+console = Console()
+
+
+@app.command("list")
+def list_jobs() -> None:
+    """
+    List all scheduled jobs with their current status.
+
+    Shows job details including next run time, status, and configuration.
+    """
+
+    rprint("📋 [bold blue]Listing Scheduled Jobs[/bold blue]")
+
+    try:
+        tasks = asyncio.run(_get_scheduled_tasks())
+
+        if not tasks:
+            rprint("ℹ️  [yellow]No scheduled jobs found[/yellow]")
+            return
+
+        table = Table(
+            title="🕒 Scheduled Jobs",
+            show_header=True,
+            header_style="bold magenta",
+        )
+        table.add_column("Job ID", style="cyan", no_wrap=True)
+        table.add_column("Name", style="green")
+        table.add_column("Status", style="bold")
+        table.add_column("Next Run", style="yellow")
+        table.add_column("Trigger", style="dim")
+
+        for task in tasks:
+            status_color = "green" if task.is_active else "red"
+            next_run = (
+                task.next_run_time.strftime("%Y-%m-%d %H:%M:%S")
+                if task.next_run_time
+                else "Not scheduled"
+            )
+
+            table.add_row(
+                task.job_id,
+                task.name or task.job_id,
+                (
+                    f"[{status_color}]{'Active' if task.is_active else 'Paused'}"
+                    f"[/{status_color}]"
+                ),
+                next_run,
+                task.trigger_type or "Unknown"
+            )
+
+        console.print(table)
+        rprint(f"\n📊 [cyan]Total jobs:[/cyan] {len(tasks)}")
+
+    except Exception as e:
+        rprint(f"❌ [red]Failed to list jobs:[/red] {e}")
+        raise typer.Exit(1)
+
+
+@app.command("stats")
+def job_statistics(
+    job_id: str | None = typer.Argument(
+        None, help="Specific job ID to show stats for"
+    ),
+    json_output: bool = typer.Option(
+        False, "--json", help="Output stats in JSON format"
+    ),
+) -> None:
+    """
+    Show job execution statistics and performance metrics.
+
+    Without job_id: Shows overall scheduler statistics
+    With job_id: Shows detailed statistics for specific job
+    """
+
+    try:
+        if job_id:
+            rprint(f"📊 [bold blue]Job Statistics for: {job_id}[/bold blue]")
+            stats = asyncio.run(_get_job_statistics(job_id))
+        else:
+            rprint("📊 [bold blue]Overall Scheduler Statistics[/bold blue]")
+            stats = asyncio.run(_get_overall_statistics())
+
+        if json_output:
+            print(json.dumps(stats, indent=2, default=str))
+            return
+
+        _display_statistics(stats, job_id)
+
+    except Exception as e:
+        rprint(f"❌ [red]Failed to get statistics:[/red] {e}")
+        raise typer.Exit(1)
+
+
+@app.command("history")
+def job_history(
+    limit: int = typer.Option(
+        20, "--limit", "-l", help="Number of recent executions to show", min=1, max=100
+    ),
+    job_id: str | None = typer.Option(
+        None, "--job-id", "-j", help="Filter by specific job ID"
+    ),
+    json_output: bool = typer.Option(
+        False, "--json", help="Output history in JSON format"
+    ),
+) -> None:
+    """
+    Show recent job execution history.
+
+    Displays execution times, durations, and results for recent job runs.
+    """
+
+    try:
+        if job_id:
+            rprint(
+                f"📚 [bold blue]Execution History for: {job_id}[/bold blue] "
+                f"(last {limit})"
+            )
+        else:
+            rprint(
+                f"📚 [bold blue]Recent Job Execution History[/bold blue] "
+                f"(last {limit})"
+            )
+
+        history = asyncio.run(_get_job_history(limit, job_id))
+
+        if json_output:
+            print(json.dumps(history, indent=2, default=str))
+            return
+
+        if not history:
+            rprint("ℹ️  [yellow]No execution history found[/yellow]")
+            return
+
+        table = Table(
+            title="🕰️  Job Execution History",
+            show_header=True,
+            header_style="bold magenta",
+        )
+        table.add_column("Timestamp", style="cyan")
+        table.add_column("Job ID", style="green")
+        table.add_column("Duration", style="yellow")
+        table.add_column("Status", style="bold")
+        table.add_column("Details", style="dim")
+
+        for execution in history:
+            status_color = "green" if execution.get("status") == "success" else "red"
+            duration = f"{execution.get('duration_ms', 0):.1f}ms"
+
+            status_value = execution.get('status', 'Unknown')
+            status_display = f"[{status_color}]{status_value}[/{status_color}]"
+            details = execution.get("details", "")[:50]
+            if len(execution.get("details", "")) > 50:
+                details += "..."
+
+            table.add_row(
+                execution.get("timestamp", "Unknown"),
+                execution.get("job_id", "Unknown"),
+                duration,
+                status_display,
+                details,
+            )
+
+        console.print(table)
+
+    except Exception as e:
+        rprint(f"❌ [red]Failed to get job history:[/red] {e}")
+        raise typer.Exit(1)
+
+
+@app.command("trigger")
+def trigger_job(
+    job_id: str = typer.Argument(..., help="Job ID to trigger manually"),
+    wait: bool = typer.Option(True, "--wait/--no-wait", help="Wait for job completion"),
+    timeout: int = typer.Option(
+        30, "--timeout", help="Timeout for waiting (seconds)", min=1, max=300
+    ),
+) -> None:
+    """
+    Manually trigger a scheduled job.
+
+    Executes the specified job immediately, bypassing the normal schedule.
+    Useful for testing job functionality or running jobs on-demand.
+    """
+
+    rprint(f"🚀 [bold blue]Triggering job:[/bold blue] {job_id}")
+
+    try:
+        # First verify the job exists
+        tasks = asyncio.run(_get_scheduled_tasks())
+        job_exists = any(task.job_id == job_id for task in tasks)
+
+        if not job_exists:
+            rprint(f"❌ [red]Job not found:[/red] {job_id}")
+            rprint("💡 [yellow]Use 'tasks list' to see available jobs[/yellow]")
+            raise typer.Exit(1)
+
+        # Trigger the job
+        result = asyncio.run(_trigger_job_execution(job_id, wait, timeout))
+
+        if result.get("status") == "success":
+            rprint(f"✅ [green]Job triggered successfully![/green]")
+            if wait and result.get("execution_details"):
+                details = result["execution_details"]
+                duration_ms = details.get('duration_ms', 0)
+                rprint(f"⏱️  [cyan]Duration:[/cyan] {duration_ms:.1f}ms")
+                rprint(f"📋 [cyan]Result:[/cyan] {details.get('result', 'No result')}")
+        else:
+            error_msg = result.get('error', 'Unknown error')
+            rprint(f"❌ [red]Job execution failed:[/red] {error_msg}")
+
+    except Exception as e:
+        rprint(f"❌ [red]Failed to trigger job:[/red] {e}")
+        raise typer.Exit(1)
+
+
+async def _get_scheduled_tasks() -> list[ScheduledTask]:
+    """Get list of scheduled tasks from the scheduler."""
+    manager = ScheduledTaskManager()
+    return await manager.list_tasks()
+
+
+async def _get_job_statistics(job_id: str) -> dict[str, Any]:
+    """Get statistics for a specific job."""
+    manager = ScheduledTaskManager()
+    return await manager.get_job_statistics(job_id)
+
+
+async def _get_overall_statistics() -> dict[str, Any]:
+    """Get overall scheduler statistics."""
+    manager = ScheduledTaskManager()
+    return await manager.get_overall_statistics()
+
+
+async def _get_job_history(limit: int, job_id: str | None) -> list[dict[str, Any]]:
+    """Get recent job execution history."""
+    manager = ScheduledTaskManager()
+    return await manager.get_job_history(limit, job_id)
+
+
+async def _trigger_job_execution(
+    job_id: str, wait: bool, timeout: int
+) -> dict[str, Any]:
+    """Trigger manual job execution."""
+    manager = ScheduledTaskManager()
+    return await manager.trigger_job(job_id, wait, timeout)
+
+
+def _display_statistics(stats: dict[str, Any], job_id: str | None) -> None:
+    """Display formatted statistics."""
+
+    if job_id:
+        # Job-specific statistics
+        info_text = f"""
+[bold cyan]Job Performance Metrics[/bold cyan]
+
+🏃 [yellow]Executions:[/yellow] {stats.get('total_executions', 0)}
+✅ [yellow]Successful:[/yellow] {stats.get('successful_executions', 0)}
+❌ [yellow]Failed:[/yellow] {stats.get('failed_executions', 0)}
+📊 [yellow]Success Rate:[/yellow] {stats.get('success_rate_percent', 0):.1f}%
+
+⏱️  [yellow]Avg Duration:[/yellow] {stats.get('avg_duration_ms', 0):.1f}ms
+🚀 [yellow]Min Duration:[/yellow] {stats.get('min_duration_ms', 0):.1f}ms
+🐌 [yellow]Max Duration:[/yellow] {stats.get('max_duration_ms', 0):.1f}ms
+
+🕐 [yellow]Last Execution:[/yellow] {stats.get('last_execution', 'Never')}
+🕑 [yellow]Next Scheduled:[/yellow] {stats.get('next_run', 'Not scheduled')}
+        """.strip()
+
+        title = f"📊 Statistics for {job_id}"
+        console.print(Panel(info_text, title=title, border_style="blue"))
+
+    else:
+        # Overall scheduler statistics
+        info_text = f"""
+[bold cyan]Scheduler Overview[/bold cyan]
+
+📋 [yellow]Total Jobs:[/yellow] {stats.get('total_jobs', 0)}
+🟢 [yellow]Active Jobs:[/yellow] {stats.get('active_jobs', 0)}
+🔴 [yellow]Paused Jobs:[/yellow] {stats.get('paused_jobs', 0)}
+
+🏃 [yellow]Total Executions:[/yellow] {stats.get('total_executions', 0)}
+✅ [yellow]Successful:[/yellow] {stats.get('successful_executions', 0)}
+❌ [yellow]Failed:[/yellow] {stats.get('failed_executions', 0)}
+
+📊 [yellow]Overall Success Rate:[/yellow] {stats.get('success_rate_percent', 0):.1f}%
+⏱️  [yellow]Avg Execution Time:[/yellow] {stats.get('avg_duration_ms', 0):.1f}ms
+
+🕐 [yellow]Scheduler Uptime:[/yellow] {stats.get('uptime', 'Unknown')}
+🔄 [yellow]Last Activity:[/yellow] {stats.get('last_activity', 'No recent activity')}
+        """.strip()
+
+        console.print(
+            Panel(info_text, title="📊 Scheduler Statistics", border_style="green")
+        )
+
+
+if __name__ == "__main__":
+    app()
+{%- endif -%}

aegis/templates/cookiecutter-aegis-project/{{cookiecutter.project_slug}}/app/cli/{% if cookiecutter.include_scheduler == /"yes/" %}tasks.py{% endif %}"

@@ -0,0 +1,340 @@
+"""
+Scheduled task CLI commands.
+
+Provides command-line interface for managing scheduled jobs,
+viewing job statistics, and manually triggering tasks.
+"""
+
+import asyncio
+import json
+from typing import Any
+
+from rich import print as rprint
+from rich.console import Console
+from rich.panel import Panel
+from rich.table import Table
+import typer
+
+from app.services.scheduler import ScheduledTaskManager
+
+app = typer.Typer(
+    name="tasks",
+    help="Scheduled task management commands",
+    no_args_is_help=True,
+)
+
+console = Console()
+
+
+@app.command("list")
+def list_jobs() -> None:
+    """
+    List all scheduled jobs with their current status.
+
+    Shows job details including next run time, status, and configuration.
+    """
+
+    rprint("📋 [bold blue]Listing Scheduled Jobs[/bold blue]")
+
+    try:
+        tasks = asyncio.run(_get_scheduled_tasks())
+
+        if not tasks:
+            rprint("ℹ️  [yellow]No scheduled jobs found[/yellow]")
+            return
+
+        table = Table(
+            title="🕒 Scheduled Jobs",
+            show_header=True,
+            header_style="bold magenta",
+        )
+        table.add_column("Job ID", style="cyan", no_wrap=True)
+        table.add_column("Name", style="green")
+        table.add_column("Status", style="bold")
+        table.add_column("Next Run", style="yellow")
+        table.add_column("Trigger", style="dim")
+
+        for task in tasks:
+            status_color = "green" if task.is_active else "red"
+            next_run = (
+                task.next_run_time.strftime("%Y-%m-%d %H:%M:%S")
+                if task.next_run_time
+                else "Not scheduled"
+            )
+
+            table.add_row(
+                task.job_id,
+                task.name or task.job_id,
+                (
+                    f"[{status_color}]{'Active' if task.is_active else 'Paused'}"
+                    f"[/{status_color}]"
+                ),
+                next_run,
+                task.trigger_type or "Unknown"
+            )
+
+        console.print(table)
+        rprint(f"\n📊 [cyan]Total jobs:[/cyan] {len(tasks)}")
+
+    except Exception as e:
+        rprint(f"❌ [red]Failed to list jobs:[/red] {e}")
+        raise typer.Exit(1)
+
+
+@app.command("stats")
+def job_statistics(
+    job_id: str | None = typer.Argument(None, help="Specific job ID to show stats for"),
+    json_output: bool = typer.Option(
+        False, "--json", help="Output stats in JSON format"
+    ),
+) -> None:
+    """
+    Show job execution statistics and performance metrics.
+
+    Without job_id: Shows overall scheduler statistics
+    With job_id: Shows detailed statistics for specific job
+    """
+
+    try:
+        if job_id:
+            rprint(f"📊 [bold blue]Job Statistics for: {job_id}[/bold blue]")
+            stats = asyncio.run(_get_job_statistics(job_id))
+        else:
+            rprint("📊 [bold blue]Overall Scheduler Statistics[/bold blue]")
+            stats = asyncio.run(_get_overall_statistics())
+
+        if json_output:
+            print(json.dumps(stats, indent=2, default=str))
+            return
+
+        _display_statistics(stats, job_id)
+
+    except Exception as e:
+        rprint(f"❌ [red]Failed to get statistics:[/red] {e}")
+        raise typer.Exit(1)
+
+
+@app.command("history")
+def job_history(
+    limit: int = typer.Option(
+        20,
+        "--limit",
+        "-l",
+        help="Number of recent executions to show",
+        min=1,
+        max=100,
+    ),
+    job_id: str | None = typer.Option(
+        None, "--job-id", "-j", help="Filter by specific job ID"
+    ),
+    json_output: bool = typer.Option(
+        False, "--json", help="Output history in JSON format"
+    ),
+) -> None:
+    """
+    Show recent job execution history.
+
+    Displays execution times, durations, and results for recent job runs.
+    """
+
+    try:
+        if job_id:
+            rprint(
+                f"📚 [bold blue]Execution History for: {job_id}[/bold blue] "
+                f"(last {limit})"
+            )
+        else:
+            rprint(
+                f"📚 [bold blue]Recent Job Execution History[/bold blue] "
+                f"(last {limit})"
+            )
+
+        history = asyncio.run(_get_job_history(limit, job_id))
+
+        if json_output:
+            print(json.dumps(history, indent=2, default=str))
+            return
+
+        if not history:
+            rprint("ℹ️  [yellow]No execution history found[/yellow]")
+            return
+
+        table = Table(
+            title="🕰️  Job Execution History",
+            show_header=True,
+            header_style="bold magenta",
+        )
+        table.add_column("Timestamp", style="cyan")
+        table.add_column("Job ID", style="green")
+        table.add_column("Duration", style="yellow")
+        table.add_column("Status", style="bold")
+        table.add_column("Details", style="dim")
+
+        for execution in history:
+            status_color = "green" if execution.get("status") == "success" else "red"
+            duration = f"{execution.get('duration_ms', 0):.1f}ms"
+
+            status_value = execution.get('status', 'Unknown')
+            status_display = f"[{status_color}]{status_value}[/{status_color}]"
+            details = execution.get("details", "")[:50]
+            if len(execution.get("details", "")) > 50:
+                details += "..."
+
+            table.add_row(
+                execution.get("timestamp", "Unknown"),
+                execution.get("job_id", "Unknown"),
+                duration,
+                status_display,
+                details,
+            )
+
+        console.print(table)
+
+    except Exception as e:
+        rprint(f"❌ [red]Failed to get job history:[/red] {e}")
+        raise typer.Exit(1)
+
+
+@app.command("trigger")
+def trigger_job(
+    job_id: str = typer.Argument(..., help="Job ID to trigger manually"),
+    wait: bool = typer.Option(True, "--wait/--no-wait", help="Wait for job completion"),
+    timeout: int = typer.Option(
+        30, "--timeout", help="Timeout for waiting (seconds)", min=1, max=300
+    ),
+) -> None:
+    """
+    Manually trigger a scheduled job.
+
+    Executes the specified job immediately, bypassing the normal schedule.
+    Useful for testing job functionality or running jobs on-demand.
+    """
+
+    rprint(f"🚀 [bold blue]Triggering job:[/bold blue] {job_id}")
+
+    try:
+        # First verify the job exists
+        tasks = asyncio.run(_get_scheduled_tasks())
+        job_exists = any(task.job_id == job_id for task in tasks)
+
+        if not job_exists:
+            rprint(f"❌ [red]Job not found:[/red] {job_id}")
+            rprint("💡 [yellow]Use 'tasks list' to see available jobs[/yellow]")
+            raise typer.Exit(1)
+
+        # Trigger the job
+        result = asyncio.run(_trigger_job_execution(job_id, wait, timeout))
+
+        if result.get("status") == "success":
+            rprint("✅ [green]Job triggered successfully![/green]")
+            if wait and result.get("execution_details"):
+                details = result["execution_details"]
+                rprint(
+                    f"⏱️  [cyan]Duration:[/cyan] {details.get('duration_ms', 0):.1f}ms"
+                )
+                rprint(f"📋 [cyan]Result:[/cyan] {details.get('result', 'No result')}")
+        else:
+            rprint(
+                f"❌ [red]Job execution failed:[/red] "
+                f"{result.get('error', 'Unknown error')}"
+            )
+
+    except Exception as e:
+        rprint(f"❌ [red]Failed to trigger job:[/red] {e}")
+        raise typer.Exit(1)
+
+
+async def _get_scheduled_tasks() -> list[Any]:
+    """Get list of scheduled tasks from the scheduler."""
+    manager = ScheduledTaskManager()
+    return await manager.list_tasks()
+
+
+async def _get_job_statistics(job_id: str) -> dict[str, Any]:
+    """Get statistics for a specific job."""
+    manager = ScheduledTaskManager()
+    return await manager.get_job_statistics(job_id)
+
+
+async def _get_overall_statistics() -> dict[str, Any]:
+    """Get overall scheduler statistics."""
+    manager = ScheduledTaskManager()
+    return await manager.get_overall_statistics()
+
+
+async def _get_job_history(
+    limit: int, job_id: str | None
+) -> list[dict[str, Any]]:
+    """Get recent job execution history."""
+    manager = ScheduledTaskManager()
+    return await manager.get_job_history(limit, job_id)
+
+
+async def _trigger_job_execution(
+    job_id: str, wait: bool, timeout: int
+) -> dict[str, Any]:
+    """Trigger manual job execution."""
+    manager = ScheduledTaskManager()
+    return await manager.trigger_job(job_id, wait, timeout)
+
+
+def _display_statistics(stats: dict[str, Any], job_id: str | None) -> None:
+    """Display formatted statistics."""
+
+    if job_id:
+        # Job-specific statistics
+        info_text = f"""
+[bold cyan]Job Performance Metrics[/bold cyan]
+
+🏃 [yellow]Executions:[/yellow] {stats.get('total_executions', 0)}
+✅ [yellow]Successful:[/yellow] {stats.get('successful_executions', 0)}
+❌ [yellow]Failed:[/yellow] {stats.get('failed_executions', 0)}
+📊 [yellow]Success Rate:[/yellow] {stats.get('success_rate_percent', 0):.1f}%
+
+⏱️  [yellow]Avg Duration:[/yellow] {stats.get('avg_duration_ms', 0):.1f}ms
+🚀 [yellow]Min Duration:[/yellow] {stats.get('min_duration_ms', 0):.1f}ms
+🐌 [yellow]Max Duration:[/yellow] {stats.get('max_duration_ms', 0):.1f}ms
+
+🕐 [yellow]Last Execution:[/yellow] {stats.get('last_execution', 'Never')}
+🕑 [yellow]Next Scheduled:[/yellow] {stats.get('next_run', 'Not scheduled')}
+        """.strip()
+
+        console.print(
+            Panel(
+                info_text,
+                title=f"📊 Statistics for {job_id}",
+                border_style="blue",
+            )
+        )
+
+    else:
+        # Overall scheduler statistics
+        info_text = f"""
+[bold cyan]Scheduler Overview[/bold cyan]
+
+📋 [yellow]Total Jobs:[/yellow] {stats.get('total_jobs', 0)}
+🟢 [yellow]Active Jobs:[/yellow] {stats.get('active_jobs', 0)}
+🔴 [yellow]Paused Jobs:[/yellow] {stats.get('paused_jobs', 0)}
+
+🏃 [yellow]Total Executions:[/yellow] {stats.get('total_executions', 0)}
+✅ [yellow]Successful:[/yellow] {stats.get('successful_executions', 0)}
+❌ [yellow]Failed:[/yellow] {stats.get('failed_executions', 0)}
+
+📊 [yellow]Overall Success Rate:[/yellow] {stats.get('success_rate_percent', 0):.1f}%
+⏱️  [yellow]Avg Execution Time:[/yellow] {stats.get('avg_duration_ms', 0):.1f}ms
+
+🕐 [yellow]Scheduler Uptime:[/yellow] {stats.get('uptime', 'Unknown')}
+🔄 [yellow]Last Activity:[/yellow] {stats.get('last_activity', 'No recent activity')}
+        """.strip()
+
+        console.print(
+            Panel(
+                info_text,
+                title="📊 Scheduler Statistics",
+                border_style="green",
+            )
+        )
+
+
+if __name__ == "__main__":
+    app()

aegis/templates/cookiecutter-aegis-project/{{cookiecutter.project_slug}}/app/components/backend/api/ai/__init__.py

@@ -0,0 +1,8 @@
+"""
+AI service API routes.
+
+REST API endpoints for AI chat functionality.
+"""
+
+# API routes will be exported here in ticket #159
+__all__ = []