vnstat2html.py

#!/usr/bin/env python3

import sys
import os
import json
import jinja2

def prettysize(num):
    for unit in ["B", "KiB", "MiB", "GiB", "TiB", "PiB", "EiB", "ZiB"]:
        if abs(num) < 1024.0:
            return f"{num:3.1f} {unit}"
        num /= 1024.0
    return f"{num:.1f} YiB"

report = json.load(sys.stdin)
try:
    rootdir = sys.argv[1]
except:
    print("Usage : {} path/to/directory".format(sys.argv[0]), file=sys.stderr)
    sys.exit(1)

os.makedirs(rootdir, exist_ok=True)

templateLoader = jinja2.FileSystemLoader(searchpath="./")
templateEnv = jinja2.Environment(loader=templateLoader)
templateEnv.filters["prettysize"] = prettysize

index_tpl = templateEnv.get_template("templates/index.html")

with open(rootdir+"/index.html", "w") as html_file:
    html_file.write(index_tpl.render(report=report))

for interface in report["interfaces"]:
    if_dir = rootdir+"/"+interface["name"]
    os.makedirs(if_dir, exist_ok=True)
    for report_type in ["5min", "hour", "day", "month", "year"]:
        with open(if_dir+"/{}.html".format(report_type), "w") as html_file:
            if_tpl = templateEnv.get_template("templates/{}.html".format(report_type))
            html_file.write(if_tpl.render(interface=interface))