????
Current Path : /lib64/python2.7/Tools/scripts/ |
Current File : //lib64/python2.7/Tools/scripts/dutree.py |
#! /usr/bin/python2.7 # 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, e: if e.errno != errno.EPIPE: raise def store(size, comps, total, d): if comps == []: return size, d if not d.has_key(comps[0]): 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 d.has_key(key): show(tsub, d[key][1], psub) if __name__ == '__main__': main()