python_utils

humanize.py

def humanize(bytes):
    bytes = int(bytes)
    if bytes >= 1024 * 1024 * 1024:
        return '{:.2f} GiB'.format(bytes / 1024 / 1024 / 1024)
    elif bytes >= 1024 * 1024:
        return '{:.2f} MiB'.format(bytes / 1024 / 1024)
    elif bytes >= 1024:
        return '{:.2f} KiB'.format(bytes / 1024)
    else:
        return '{:.2f} B'.format(bytes)

progress.py

def progress(message, value, max, bytes=False):
    print('\r', end='')
    print(' ' * (shutil.get_terminal_size((80, 20)).columns - 1), end='')
    if bytes:
        bytes = '{} of {}'.format(humanize(value), humanize(max))
    else:
        bytes = ""
    print('\r{}    {:.2f}%    {}'.format(message, (value / max) * 100, bytes),
          end='')
    if value >= max:
        print()