dutree.py

#! /usr/bin/env python
# Format du output in a tree shape

importos, sys, errno

defmain():
p=os.popen('du '+' '.join(sys.argv[1:]), 'r')
total, d=None, {}
forlineinp.readlines():
i=0
whileline[i] in'0123456789': i=i+1
size=eval(line[:i])
whileline[i] in' \t': i=i+1
filename=line[i:-1]
comps=filename.split('/')
ifcomps[0] =='': comps[0] ='/'
ifcomps[len(comps)-1] =='': delcomps[len(comps)-1]
total, d=store(size, comps, total, d)
try:
display(total, d)
exceptIOError, e:
ife.errno!=errno.EPIPE:
raise

defstore(size, comps, total, d):
ifcomps== []:
returnsize, d
ifnotd.has_key(comps[0]):
d[comps[0]] =None, {}
t1, d1=d[comps[0]]
d[comps[0]] =store(size, comps[1:], t1, d1)
returntotal, d

defdisplay(total, d):
show(total, d, '')

defshow(total, d, prefix):
ifnotd: return
list= []
sum=0
forkeyind.keys():
tsub, dsub=d[key]
list.append((tsub, key))
iftsubisnotNone: sum=sum+tsub
## if sum < total:
## list.append((total - sum, os.curdir))
list.sort()
list.reverse()
width=len(repr(list[0][0]))
fortsub, keyinlist:
iftsubisNone:
psub=prefix
else:
printprefix+repr(tsub).rjust(width) +' '+key
psub=prefix+' '*(width-1) +'|'+' '*(len(key)+1)
ifd.has_key(key):
show(tsub, d[key][1], psub)

if__name__=='__main__':
main()