Initial commit

securecheck/models.py

@@ -0,0 +1,46 @@
+from __future__ import annotations
+
+from dataclasses import dataclass, field
+from datetime import datetime
+from typing import Callable, TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from .executor import ExecutionContext
+
+
+TaskHandler = Callable[["ExecutionContext"], "TaskResult"]
+
+
+@dataclass(frozen=True)
+class TaskDefinition:
+    key: str
+    label: str
+    description: str
+    category: str
+    handler: TaskHandler
+    requires_root: bool = True
+    default_selected: bool = False
+
+
+@dataclass
+class TaskResult:
+    key: str
+    label: str
+    success: bool
+    changed: bool
+    started_at: datetime
+    finished_at: datetime
+    details: list[str] = field(default_factory=list)
+    error: str | None = None
+
+    @property
+    def duration_seconds(self) -> float:
+        return (self.finished_at - self.started_at).total_seconds()
+
+
+@dataclass
+class Scenario:
+    name: str
+    task_keys: list[str]
+    description: str = ""
+    builtin: bool = False