sys/src/cmd/python/Tools/scripts/dutree.py

   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()