1 #! /usr/bin/env python 2 # Format du output in a tree shape 3 4 import os, sys, errno 5 6 def main(): 7 p = os.popen('du ' + ' '.join(sys.argv[1:]), 'r') 8 total, d = None, {} 9 for line in p.readlines(): 10 i = 0 11 while line[i] in '0123456789': i = i+1 12 size = eval(line[:i]) 13 while line[i] in ' \t': i = i+1 14 filename = line[i:-1] 15 comps = filename.split('/') 16 if comps[0] == '': comps[0] = '/' 17 if comps[len(comps)-1] == '': del comps[len(comps)-1] 18 total, d = store(size, comps, total, d) 19 try: 20 display(total, d) 21 except IOError, e: 22 if e.errno != errno.EPIPE: 23 raise 24 25 def store(size, comps, total, d): 26 if comps == []: 27 return size, d 28 if not d.has_key(comps[0]): 29 d[comps[0]] = None, {} 30 t1, d1 = d[comps[0]] 31 d[comps[0]] = store(size, comps[1:], t1, d1) 32 return total, d 33 34 def display(total, d): 35 show(total, d, '') 36 37 def show(total, d, prefix): 38 if not d: return 39 list = [] 40 sum = 0 41 for key in d.keys(): 42 tsub, dsub = d[key] 43 list.append((tsub, key)) 44 if tsub is not None: sum = sum + tsub 45 ## if sum < total: 46 ## list.append((total - sum, os.curdir)) 47 list.sort() 48 list.reverse() 49 width = len(repr(list[0][0])) 50 for tsub, key in list: 51 if tsub is None: 52 psub = prefix 53 else: 54 print prefix + repr(tsub).rjust(width) + ' ' + key 55 psub = prefix + ' '*(width-1) + '|' + ' '*(len(key)+1) 56 if d.has_key(key): 57 show(tsub, d[key][1], psub) 58 59 if __name__ == '__main__': 60 main() 61
