b73b5251ae
Full-stack burn-in orchestration dashboard (Stages 1–6d complete): FastAPI backend, SQLite/WAL, SSE live dashboard, mock TrueNAS server, SMTP/webhook notifications, batch burn-in, settings UI, audit log, stats page, cancel SMART/burn-in, drag-to-reorder stages. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
136 lines
3.9 KiB
Python
136 lines
3.9 KiB
Python
"""
|
|
Jinja2 template engine + filter/helper registration.
|
|
Import `templates` from here — do not create additional Jinja2 instances.
|
|
"""
|
|
|
|
from datetime import datetime, timezone
|
|
|
|
from fastapi.templating import Jinja2Templates
|
|
|
|
templates = Jinja2Templates(directory="app/templates")
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# Template filters
|
|
# ---------------------------------------------------------------------------
|
|
|
|
def _format_bytes(value: int | None) -> str:
|
|
if value is None:
|
|
return "—"
|
|
tb = value / 1_000_000_000_000
|
|
if tb >= 1:
|
|
return f"{tb:.0f} TB"
|
|
gb = value / 1_000_000_000
|
|
return f"{gb:.0f} GB"
|
|
|
|
|
|
def _format_eta(seconds: int | None) -> str:
|
|
if not seconds or seconds <= 0:
|
|
return ""
|
|
h = seconds // 3600
|
|
m = (seconds % 3600) // 60
|
|
if h > 0:
|
|
return f"~{h}h {m}m" if m else f"~{h}h"
|
|
return f"~{m}m" if m else "<1m"
|
|
|
|
|
|
def _temp_class(celsius: int | None) -> str:
|
|
if celsius is None:
|
|
return ""
|
|
if celsius < 40:
|
|
return "temp-cool"
|
|
if celsius < 50:
|
|
return "temp-warm"
|
|
return "temp-hot"
|
|
|
|
|
|
def _format_dt(iso: str | None) -> str:
|
|
if not iso:
|
|
return "—"
|
|
try:
|
|
dt = datetime.fromisoformat(iso)
|
|
if dt.tzinfo is None:
|
|
dt = dt.replace(tzinfo=timezone.utc)
|
|
local = dt.astimezone()
|
|
return local.strftime("%H:%M:%S")
|
|
except Exception:
|
|
return iso
|
|
|
|
|
|
def _format_dt_full(iso: str | None) -> str:
|
|
"""Date + time for history tables."""
|
|
if not iso:
|
|
return "—"
|
|
try:
|
|
dt = datetime.fromisoformat(iso)
|
|
if dt.tzinfo is None:
|
|
dt = dt.replace(tzinfo=timezone.utc)
|
|
local = dt.astimezone()
|
|
return local.strftime("%Y-%m-%d %H:%M:%S")
|
|
except Exception:
|
|
return iso
|
|
|
|
|
|
def _format_duration(seconds: float | int | None) -> str:
|
|
if seconds is None or seconds < 0:
|
|
return "—"
|
|
seconds = int(seconds)
|
|
h = seconds // 3600
|
|
m = (seconds % 3600) // 60
|
|
s = seconds % 60
|
|
if h > 0:
|
|
return f"{h}h {m}m {s}s"
|
|
if m > 0:
|
|
return f"{m}m {s}s"
|
|
return f"{s}s"
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# Template globals
|
|
# ---------------------------------------------------------------------------
|
|
|
|
def _drive_status(drive: dict) -> str:
|
|
short = (drive.get("smart_short") or {}).get("state", "idle")
|
|
long_ = (drive.get("smart_long") or {}).get("state", "idle")
|
|
health = drive.get("smart_health", "UNKNOWN")
|
|
if "running" in (short, long_):
|
|
return "running"
|
|
if short == "failed" or long_ == "failed" or health == "FAILED":
|
|
return "failed"
|
|
if "passed" in (short, long_):
|
|
return "passed"
|
|
return "idle"
|
|
|
|
|
|
def _format_elapsed(iso: str | None) -> str:
|
|
"""Human-readable elapsed time since an ISO timestamp (e.g. '2h 34m')."""
|
|
if not iso:
|
|
return ""
|
|
try:
|
|
start = datetime.fromisoformat(iso)
|
|
if start.tzinfo is None:
|
|
start = start.replace(tzinfo=timezone.utc)
|
|
elapsed = int((datetime.now(timezone.utc) - start).total_seconds())
|
|
if elapsed < 0:
|
|
return ""
|
|
h = elapsed // 3600
|
|
m = (elapsed % 3600) // 60
|
|
s = elapsed % 60
|
|
if h > 0:
|
|
return f"{h}h {m}m"
|
|
if m > 0:
|
|
return f"{m}m {s}s"
|
|
return f"{s}s"
|
|
except Exception:
|
|
return ""
|
|
|
|
|
|
# Register
|
|
templates.env.filters["format_bytes"] = _format_bytes
|
|
templates.env.filters["format_eta"] = _format_eta
|
|
templates.env.filters["temp_class"] = _temp_class
|
|
templates.env.filters["format_dt"] = _format_dt
|
|
templates.env.filters["format_dt_full"] = _format_dt_full
|
|
templates.env.filters["format_duration"] = _format_duration
|
|
templates.env.filters["format_elapsed"] = _format_elapsed
|
|
templates.env.globals["drive_status"] = _drive_status
|