#! /usr/bin/env python3
# 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 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
## 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 key in d:
 show(tsub, d[key][1], psub)

if __name__ == '__main__':
 main()