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