import os, sys ################################################################################ def main(): try: tree = SizeTree(os.path.abspath(sys.argv[1])) except IndexError: print('Usage: {} '.format(os.path.basename(sys.argv[0]))) else: show(tree) def show(tree): print('{} [{}]'.format(tree.path, convert(tree.total_size))) walk(tree, '') if not tree.children: print('No subfolders exist') def walk(tree, prefix): dir_prefix, walk_prefix = prefix + '+---', prefix + '| ' for pos, neg, child in enumerate2(tree.children): if neg == -1: dir_prefix, walk_prefix = prefix + '\\---', prefix + ' ' print('{}{} [{}]'.format(dir_prefix, child.name, convert(child.total_size))) walk(child, walk_prefix) def enumerate2(sequence): length = len(sequence) for count, value in enumerate(sequence): yield count, count - length, value ################################################################################ class SizeTree: def __init__(self, path, name=None): self.path = path self.name = os.path.basename(path) if name is None else name self.children = [] self.file_size = 0 self.total_size = 0 try: dir_list = os.listdir(path) except OSError: pass else: for name in dir_list: path_name = os.path.join(path, name) if os.path.isdir(path_name): size_tree = SizeTree(path_name, name) self.children.append(size_tree) self.total_size += size_tree.total_size elif os.path.isfile(path_name): try: self.file_size += os.path.getsize(path_name) except OSError: pass self.total_size += self.file_size ################################################################################ def convert(number): "Convert bytes into human-readable representation." if not number: return '0 Bytes' assert 0 < number < 1 << 110, 'number out of range' ordered = reversed(tuple(format_bytes(partition_number(number, 1 << 10)))) cleaned = ', '.join(item for item in ordered if item[0] != '0') return cleaned def partition_number(number, base): "Continually divide number by base until zero." div, mod = divmod(number, base) yield mod while div: div, mod = divmod(div, base) yield mod def format_bytes(parts): "Format partitioned bytes into human-readable strings." for power, number in enumerate(parts): yield '{} {}'.format(number, format_suffix(power, number)) def format_suffix(power, number): "Compute the suffix for a certain power of bytes." return (PREFIX[power] + 'byte').capitalize() + ('s' if number != 1 else '') PREFIX = ' kilo mega giga tera peta exa zetta yotta bronto geop'.split(' ') ################################################################################ if __name__ == '__main__': main()