current commit
200426.5 · initial commit of ~/bin
#!/usr/bin/env python2

from __future__ import absolute_import, division, print_function
import argparse
import sys

p = argparse.ArgumentParser()
p.add_argument('size', nargs='?', default=None, type=int,
               help='If not set, size is read from stdin.')
args = p.parse_args()

size = float(args.size if args.size is not None else int(sys.stdin.read()))
units = ['B', 'KB', 'MB', 'GB', 'TB', 'PB']
unit = 0
while size > 1024:
    size /= 1024
    unit += 1
print('%0.2f %s' % (size, units[unit]))