| |
| |
|
|
| import os, sys, errno |
|
|
| def main(): |
| total, d = None, {} |
| with os.popen('du ' + ' '.join(sys.argv[1:])) as p: |
| for line in p: |
| 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 as e: |
| if e.errno != errno.EPIPE: |
| raise |
|
|
| def store(size, comps, total, d): |
| if comps == []: |
| return size, d |
| if comps[0] not in d: |
| 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 |
| |
| |
| 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 key in d: |
| show(tsub, d[key][1], psub) |
|
|
| if __name__ == '__main__': |
| main() |
|
|
