EmbeddedPkg: Extend NvVarStoreFormattedLib LIBRARY_CLASS

[mirror_edk2.git] / AppPkg / Applications / Python / Python-2.7.2 / Tools / scripts / dutree.py

#! /usr/bin/env python\r
# Format du output in a tree shape\r
\r
import os, sys, errno\r
\r
def main():\r
    p = os.popen('du ' + ' '.join(sys.argv[1:]), 'r')\r
    total, d = None, {}\r
    for line in p.readlines():\r
        i = 0\r
        while line[i] in '0123456789': i = i+1\r
        size = eval(line[:i])\r
        while line[i] in ' \t': i = i+1\r
        filename = line[i:-1]\r
        comps = filename.split('/')\r
        if comps[0] == '': comps[0] = '/'\r
        if comps[len(comps)-1] == '': del comps[len(comps)-1]\r
        total, d = store(size, comps, total, d)\r
    try:\r
        display(total, d)\r
    except IOError, e:\r
        if e.errno != errno.EPIPE:\r
            raise\r
\r
def store(size, comps, total, d):\r
    if comps == []:\r
        return size, d\r
    if not d.has_key(comps[0]):\r
        d[comps[0]] = None, {}\r
    t1, d1 = d[comps[0]]\r
    d[comps[0]] = store(size, comps[1:], t1, d1)\r
    return total, d\r
\r
def display(total, d):\r
    show(total, d, '')\r
\r
def show(total, d, prefix):\r
    if not d: return\r
    list = []\r
    sum = 0\r
    for key in d.keys():\r
        tsub, dsub = d[key]\r
        list.append((tsub, key))\r
        if tsub is not None: sum = sum + tsub\r
##  if sum < total:\r
##      list.append((total - sum, os.curdir))\r
    list.sort()\r
    list.reverse()\r
    width = len(repr(list[0][0]))\r
    for tsub, key in list:\r
        if tsub is None:\r
            psub = prefix\r
        else:\r
            print prefix + repr(tsub).rjust(width) + ' ' + key\r
            psub = prefix + ' '*(width-1) + '|' + ' '*(len(key)+1)\r
        if d.has_key(key):\r
            show(tsub, d[key][1], psub)\r
\r
if __name__ == '__main__':\r
    main()\r