62 lines
1.8 KiB
Python
62 lines
1.8 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""Minimal supervisor for scripted tasks and long-running utilities."""
|
|
|
|
from datetime import datetime
|
|
from typing import Any, Dict, Iterable, Optional
|
|
|
|
from .models import ProcessRun, ProcessRunState
|
|
from .registry import RunRegistry
|
|
|
|
|
|
class ProcessSupervisor:
|
|
"""Tracks supervised runs without executing real processes yet."""
|
|
|
|
def __init__(self, registry: Optional[RunRegistry] = None) -> None:
|
|
self.registry = registry or RunRegistry()
|
|
|
|
def spawn(
|
|
self,
|
|
run_id: str,
|
|
command: str,
|
|
scope_key: str,
|
|
metadata: Optional[Dict[str, Any]] = None,
|
|
) -> ProcessRun:
|
|
run = ProcessRun(
|
|
run_id=run_id,
|
|
command=command,
|
|
scope_key=scope_key,
|
|
metadata=metadata or {},
|
|
)
|
|
run.state = ProcessRunState.RUNNING
|
|
run.updated_at = datetime.utcnow()
|
|
self.registry.add(run)
|
|
return run
|
|
|
|
def update_state(
|
|
self,
|
|
run_id: str,
|
|
state: ProcessRunState,
|
|
metadata: Optional[Dict[str, Any]] = None,
|
|
) -> Optional[ProcessRun]:
|
|
run = self.registry.get(run_id)
|
|
if not run:
|
|
return None
|
|
run.state = state
|
|
run.metadata.update(metadata or {})
|
|
run.updated_at = datetime.utcnow()
|
|
self.registry.update(run)
|
|
return run
|
|
|
|
def cancel(self, run_id: str, reason: Optional[str] = None) -> Optional[ProcessRun]:
|
|
run = self.registry.get(run_id)
|
|
if not run:
|
|
return None
|
|
run.state = ProcessRunState.CANCELLED
|
|
run.metadata.setdefault("cancel_reason", reason or "manual")
|
|
run.updated_at = datetime.utcnow()
|
|
self.registry.update(run)
|
|
return run
|
|
|
|
def list_runs(self) -> Iterable[ProcessRun]:
|
|
return self.registry.list()
|