checkDirSizes.py

#!/usr/bin/env python
from __future__ import print_function

import sys
from pickle import Pickler

from _py2with3compatibility import run_cmd


def doDu(what):
    error, out = run_cmd("du -k -s %s" % what)
    if error:
        print("Error while getting directory size.")
        sys.exit(1)
    results = [l.split() for l in out.split("\n")]
    return dict(
        [(pkg.strip().replace("src/", ""), int(sz.strip() * 1024)) for (sz, pkg) in results]
    )


if __name__ == "__main__":
    try:
        f = open("dirSizeInfo.pkl", "wb")
        pklr = Pickler(f, protocol=2)
        pklr.dump(doDu("src lib bin"))
        pklr.dump(doDu("src/*/*"))
        f.close()
    except Exception as e:
        print("ERROR during pickling results for dir size:", str(e))
        sys.exit(1)
    print("Successfully pickled results for dir size !")