urlretrieve_tqdm_progressbar.py

"""
Usage:

    urlretrieve(source_url, filename, reporthook=ProgressBar())
"""

class ProgressBar:
    def __init__(self):
        self.tqdm = None

    def __call__(self, block_num, block_size, total_size):
        if self.tqdm is None:
            self.tqdm = tqdm(
                total=total_size,
                unit_divisor=1024,
                unit_scale=True,
                unit="B",
                leave=False,
            )

        progress = block_num * block_size
        if progress >= total_size:
            self.tqdm.close()
            return

        self.tqdm.update(progress - self.tqdm.n)