directory_treemap.py

# This script reads a GIT directory and prints out the contents
# appropriately to be incorporated into a Google Visualization Treemap
# https://code.google.com/apis/ajax/playground/?type=visualization#tree_map
import os

rootdir = "E:\\Git"

result = {}
for path, dirs, files in os.walk(rootdir):
    key = os.path.basename(path)
    if ".git" in path:
        continue
    parent = os.path.basename(os.path.dirname(path))
    count = len(files)
    sm = sum(os.path.getsize(os.path.join(path, f)) for f in files)
    if key not in result:
        result[key] = (parent, count, sm)

print "['Location', 'Parent', 'File Count', 'File Size'],"

for k,v in result.iteritems():
    print "['%s', '%s', %s, %s]," % (k, v[0], v[1], v[2])