cluster-capacity.py

## Generate table to parse:
#
# kubectl get nodes -o custom-columns=NAME:.metadata.name,TYPE:.metadata.labels.nodetype,CPU:.status.capacity.cpu,MEMORY:.status.capacity.memory,DISK:.status.capacity.ephemeral-storage --sort-by .metadata.labels.nodetype >> output.txt

f = open('output.txt');

f.readline(); # skip header row

aggregated = {
    "computer": { "cpu": 0, "memory": 0, "disk": 0, "nodes": 0 }, # Memory and Disk in Ki
    "edge": { "cpu": 0, "memory": 0, "disk": 0, "nodes": 0 },
    "server": { "cpu": 0, "memory": 0, "disk": 0, "nodes": 0 },
}

for line in f:
    columns = line.split()

    nodetype = columns[1]
    cpu = int(columns[2])
    memory = int(columns[3].split('Ki')[0])
    disk = int(columns[4].split('Ki')[0])

    aggregated[nodetype]["cpu"] = aggregated[nodetype]["cpu"] + cpu 
    aggregated[nodetype]["memory"] = aggregated[nodetype]["memory"] + memory 
    aggregated[nodetype]["disk"] = aggregated[nodetype]["disk"] + disk 
    aggregated[nodetype]["nodes"] += 1    

cpu = 0
memory = 0
disk = 0
nodes = 0

for nodetype in aggregated:
    print(nodetype, aggregated[nodetype])
    cpu += aggregated[nodetype]["cpu"]
    memory += aggregated[nodetype]["memory"]
    disk += aggregated[nodetype]["disk"]
    nodes += aggregated[nodetype]["nodes"]

def sizeof_fmt(num, suffix="B"):
    for unit in ["", "Ki", "Mi", "Gi", "Ti", "Pi", "Ei", "Zi"]:
        if abs(num) < 1024.0:
            return f"{num:3.1f}{unit}{suffix}"
        num /= 1024.0
    return f"{num:.1f}Yi{suffix}"

print('')
print('Total CPUs:', cpu)
for nodetype in aggregated:
    print('\t'+nodetype+":", aggregated[nodetype]["cpu"])

print('Total Memory:', sizeof_fmt(memory*1024))
for nodetype in aggregated:
    print('\t'+nodetype+":", sizeof_fmt(aggregated[nodetype]["memory"]*1024))

print('Total Disk:', sizeof_fmt(disk*1024))
for nodetype in aggregated:
    print('\t'+nodetype+":", sizeof_fmt(aggregated[nodetype]["disk"]*1024))

print('Total Nodes:', nodes)
for nodetype in aggregated:
    print('\t'+nodetype+":", aggregated[nodetype]["nodes"])