cheap_system_monitor/generate_graphs.py

#!/usr/bin/env python3
"""Generate system-monitor PNG graphs from its RRD database."""

from __future__ import annotations

import argparse
import configparser
import html
from pathlib import Path
import subprocess


PERIODS = {"3hours": "3h", "1day": "1d", "2weeks": "2w", "1year": "1y"}
COLORS = ["#2563EB", "#DC2626", "#16A34A", "#9333EA"]
LOAD_COLORS = ["#DC2626", "#F59E0B", "#FACC15"]  # red, amber, yellow


def graph(output: Path, rrd: Path, start: str, title: str, unit: str,
          series: list[tuple[str, str, str]], logarithmic: bool = False,
          stacked: bool = False, disk_memory: bool = False,
          network_mixed: bool = False) -> None:
    args = ["rrdtool", "graph", str(output), "--start", f"end-{start}",
            "--end", "now", "--width", "900", "--height", "260",
            "--title", title, "--vertical-label", unit, "--slope-mode",
            "--watermark", "cheap-system-monitor"]
    if logarithmic:
        args += ["--logarithmic", "--lower-limit", "1"]
    value_suffix = "%%" if unit == "%" else ""
    for index, (ds, label, cf) in enumerate(series):
        var = f"v{index}"
        if network_mixed and ds == "net_out":
            drawing = f"AREA:{var}#86EFAC:{label}"
        elif network_mixed and ds == "net_in":
            drawing = f"LINE3:{var}#2563EB:{label}"
        elif disk_memory and ds == "memory":
            drawing = f"LINE3:{var}#DC2626:{label}"
        elif disk_memory:
            # Transparency keeps multiple mountpoints visible when their
            # percentage areas overlap.
            drawing = f"AREA:{var}{COLORS[index % len(COLORS)]}80:{label}"
        elif stacked:
            stack = ":STACK" if index else ""
            stacked_colors = LOAD_COLORS if title.startswith("Processor load") else COLORS
            drawing = f"AREA:{var}{stacked_colors[index]}:{label}{stack}"
        else:
            drawing = f"LINE{2 if index < 3 else 1}:{var}{COLORS[index % len(COLORS)]}:{label}"
        args += [f"DEF:{var}={rrd}:{ds}:{cf}",
                 drawing,
                 f"GPRINT:{var}:LAST:Current\\:%8.2lf%s{value_suffix}",
                 f"GPRINT:{var}:AVERAGE:Average\\:%8.2lf%s{value_suffix}",
                 f"GPRINT:{var}:MAX:Maximum\\:%8.2lf%s{value_suffix}\\n"]
    subprocess.run(args, check=True)


def main() -> None:
    parser = argparse.ArgumentParser(description=__doc__)
    parser.add_argument("--config", default="/opt/cheap_system_monitor/monitor.conf")
    args = parser.parse_args()
    config_parser = configparser.ConfigParser()
    if not config_parser.read(args.config):
        raise SystemExit(f"Configuration file not found: {args.config}")
    config = config_parser["monitor"]
    rrd = Path(config["rrd_file"])
    if not rrd.exists():
        raise SystemExit(f"RRD file does not exist: {rrd}")
    output_dir = Path(config.get("graph_dir", "/var/www/html/system-monitor"))
    output_dir.mkdir(parents=True, exist_ok=True)
    mounts = [item.strip() for item in config["mountpoints"].split(",") if item.strip()]
    if not mounts or mounts[0] != "/":
        mounts.insert(0, "/")
    definitions = [
        ("load", "Processor load", "load", [("load1", "1 minute", "AVERAGE"), ("load5", "5 minutes", "AVERAGE"), ("load15", "15 minutes", "AVERAGE")], False, True, False, False),
        ("cpu", "CPU usage", "%", [("cpu_user", "user", "AVERAGE"), ("cpu_system", "system", "AVERAGE"), ("cpu_nice", "nice", "AVERAGE")], False, True, False, False),
        ("network", "Network throughput", "bytes/s", [("net_out", "sent", "AVERAGE"), ("net_in", "received", "AVERAGE")], False, False, False, True),
        ("disk", "Disk and memory usage", "%", [(f"disk_{'root' if i == 0 else i}", mount, "AVERAGE") for i, mount in enumerate(mounts)] + [("memory", "memory", "AVERAGE")], False, False, True, False),
        ("uptime", "System uptime", "days", [("uptime", "uptime", "AVERAGE")], False, False, False, False),
    ]
    for period_name, start in PERIODS.items():
        for filename, title, unit, series, logarithmic, stacked, disk_memory, network_mixed in definitions:
            graph(output_dir / f"{filename}-{period_name}.png", rrd, start,
                  f"{title} — {period_name}", unit, series, logarithmic,
                  stacked, disk_memory, network_mixed)
    cards = "\n".join(
        f'<section><h2>{html.escape(title)}</h2>' + "".join(
            f'<figure><figcaption>{period}</figcaption>'
            f'<img src="{filename}-{period}.png" alt="{html.escape(title)}, {period}"></figure>'
            for period in PERIODS
        ) + "</section>"
        for filename, title, _unit, _series, _logarithmic, _stacked, _disk_memory, _network_mixed in definitions
    )
    page = f"""<!doctype html>
<html lang="en"><head><meta charset="utf-8"><meta name="viewport" content="width=device-width">
<meta http-equiv="refresh" content="300"><title>System monitor</title>
<style>body{{font:16px system-ui;margin:auto;max-width:1100px;padding:1rem;background:#f8fafc;color:#172033}}
section{{background:white;padding:1rem;margin:1rem 0;border-radius:.5rem;box-shadow:0 1px 4px #0002}}
figure{{margin:1rem 0}} img{{width:100%;height:auto}} figcaption{{font-weight:600}}</style>
</head><body><h1>System monitor</h1>{cards}</body></html>"""
    (output_dir / "index.html").write_text(page)


if __name__ == "__main__":
    main()