tqdm_file_progress_bar

tqdm_file_progress_bar.py

import io

from tqdm import tqdm


def get_file_end_position(fileobj):
    current_pos = fileobj.tell()
    end = fileobj.seek(0, io.SEEK_END)
    fileobj.seek(current_pos)
    return end


def _tqdm_file_progress_bar(iterable, fileobj, *args, **kwargs):
    total = get_file_end_position(fileobj)
    with tqdm(total=total, *args, **kwargs) as progress:
        for x in iterable:
            progress.n = fileobj.tell()
            progress.update(0)
            yield x


def tqdm_file_progress_bar(iterable, fileobj, *args, **kwargs):
    if fileobj.seekable():
        return _tqdm_file_progress_bar(iterable, fileobj, *args, **kwargs)
    else:
        return tqdm(iterable, *args, **kwargs)