#!/usr/bin/env python3

# Requires root permissions
import os
import glob
from pprint import pprint

stats = {}
os.chdir('/sys/kernel/debug/zswap')
for filename in  os.listdir():
    with open(filename) as f:
        stats[filename] = int(f.read())

max_text_lenght = max( { len(x) for x in stats.keys() } )
template = "{0:" + str(max_text_lenght) + "}:{1:<20g}"

stats['pool_total_size (MB)'] = stats['pool_total_size'] / (1024*1024)
stats['stored size (MB)'] = stats['stored_pages'] * 4096 / (1024*1024)
stats['compression ratio'] = stats['stored_pages'] * 4096 / stats['pool_total_size']

for k, v in stats.items():
    print(template.format(k,v))