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