72 lines
2.4 KiB
Python
72 lines
2.4 KiB
Python
import asyncio
|
|
import socket
|
|
import time
|
|
from typing import Any
|
|
|
|
import psutil
|
|
from fastapi import APIRouter, Request
|
|
from fastapi.responses import HTMLResponse, JSONResponse
|
|
|
|
from app.collect import net as net_col
|
|
from app.render import render, uptime_str
|
|
|
|
router = APIRouter(prefix="/api", tags=["overview"])
|
|
|
|
|
|
@router.get("/overview")
|
|
async def overview(request: Request):
|
|
store = request.app.state.store
|
|
s: dict[str, Any] = store.latest() or {}
|
|
mem_total = s.get("mem_total") or 0
|
|
vram_total = s.get("vram_total") or 0
|
|
vram_used = s.get("vram_used")
|
|
c = {
|
|
"cpu": s.get("cpu"),
|
|
"cpu_temp": s.get("cpu_temp"),
|
|
"load1": s.get("load1"),
|
|
"load5": s.get("load5"),
|
|
"load15": s.get("load15"),
|
|
"mem_used": s.get("mem_used"),
|
|
"mem_total": mem_total,
|
|
"mem_pct": s.get("mem_pct"),
|
|
"swap_used": s.get("swap_used"),
|
|
"swap_total": s.get("swap_total") or 0,
|
|
"swap_pct": s.get("swap_pct"),
|
|
"gpu": s.get("gpu"),
|
|
"gpu_name": s.get("gpu_name"),
|
|
"gpu_temp": s.get("gpu_temp"),
|
|
"vram_used": vram_used,
|
|
"vram_total": vram_total,
|
|
"vram_pct": s.get("vram_pct")
|
|
or ((vram_used / vram_total * 100) if (vram_total and vram_used is not None) else None),
|
|
"battery": s.get("battery"),
|
|
"battery_status": s.get("battery_status"),
|
|
"ac_online": s.get("ac_online"),
|
|
"uptime": uptime_str(time.time() - psutil.boot_time()),
|
|
"hostname": socket.gethostname(),
|
|
"cores": psutil.cpu_count(logical=True) or 1,
|
|
**await asyncio.to_thread(net_col.sample),
|
|
}
|
|
return HTMLResponse(render("overview.html", c=c))
|
|
|
|
|
|
@router.get("/history")
|
|
async def history(request: Request):
|
|
snap = request.app.state.store.snapshot()
|
|
ts = [round(t, 1) for t, _ in snap]
|
|
keys: set[str] = set()
|
|
for _, sample in snap:
|
|
for k, v in sample.items():
|
|
if isinstance(v, (int, float)) and not isinstance(v, bool):
|
|
keys.add(k)
|
|
series: dict[str, list[Any]] = {}
|
|
for _, sample in snap:
|
|
for k in keys:
|
|
v = sample.get(k)
|
|
if isinstance(v, (int, float)) and not isinstance(v, bool):
|
|
if isinstance(v, float):
|
|
v = round(v, 1)
|
|
else:
|
|
v = None
|
|
series.setdefault(k, []).append(v)
|
|
return JSONResponse({"ts": ts, "series": series})
|