getSize.py

# coding: utf-8
import os
import sys


def formatSize(size):
    s = 'BKMGTP'
    ct = 0
    while size >= (1 << ct):
        ct += 10
    if ct >= 10: ct -= 10
    return '{sz:.2f}{a}'.format(sz=size / (1 << ct), a=s[ct // 10])


def getSize(path='.'):
    if os.path.isdir(path):
        gen = os.walk(path)
        li = []
        for root, dirs, files in gen:
            for f in files:
                sz = os.path.getsize(os.path.join(root, f))
                li.append(sz)
        #li.insert(('.',sum(i[1] for i in li)),0)
        #size  = [f'{i[0]}: {formatSize(i[1])}' for i in li]
        return formatSize(sum(li))
    else:
        return formatSize(os.path.getsize(path))


if __name__ == "__main__":
    items = sys.argv[1:]
    for i in items:
        print('{i}: {sz}'.format(i=i, sz=getSize(i)))