File size: 1,352 Bytes
06357e7 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 | from __future__ import annotations
import os
import time
from pathlib import Path
from typing import Mapping
from .classifier import classify_all
from .dryrun import build_dry_run
from .models import DashboardSnapshot, ProcessSample
from .procfs import scan_processes
from .risk import assess_groups
from .tree import build_groups, build_nodes
def build_snapshot(
proc_root: str | Path = "/proc",
*,
previous: Mapping[str, ProcessSample] | None = None,
exclude_self: bool = True,
) -> DashboardSnapshot:
exclude_pids = {os.getpid()} if exclude_self else set()
processes = scan_processes(
proc_root,
previous=previous,
current_uid_only=True,
exclude_pids=exclude_pids,
)
classifications = classify_all(processes)
nodes = build_nodes(processes, classifications)
groups = build_groups(nodes)
stats, warnings = assess_groups(groups)
dry_run = build_dry_run(groups)
return DashboardSnapshot(
sample_ts=time.time(),
processes=processes,
classifications=classifications,
nodes=nodes,
groups=groups,
dry_run=dry_run,
stats=stats,
warnings=warnings,
)
def samples_by_key(snapshot: DashboardSnapshot) -> dict[str, ProcessSample]:
return {sample.process_key: sample for sample in snapshot.processes}
|