size_format.py

def _size_format(bytes, precision=2):
    """Returns a humanized string for a given amount of bytes"""
    bytes = int(bytes)
    if bytes < 1000:
        return "{} bytes".format(bytes)
    log = math.floor(math.log(bytes, 1000))
    return "{:.{}f} {}".format(
        bytes / math.pow(1000, log),
        precision,
        ['bytes', 'KB', 'MB', 'GB', 'TB', 'PB', 'EB', 'ZB', 'YB'][int(log)]
    )