如何将进度栏模块与urlRetrieve一起使用

2022-03-03 00:00:00 python python-3.x progress-bar

问题描述

我的pyhton3脚本使用urlRetrieve通过Internet下载多个图像,我想为每个下载添加一个进度条,其中包含完成百分比和下载速度。

progressbar模块似乎是一个很好的解决方案,但是尽管我已经浏览了their examples,并且example4似乎是正确的,但我仍然无法理解如何将其包装在urlRetrieve中。

我想我应该添加第三个参数:

urllib.request.urlretrieve('img_url', 'img_filename', some_progressbar_based_reporthook)

但我如何正确定义它?


解决方案

另一个答案中的建议对我来说没有超过1%。以下是适用于我在Python3上的完整实现:

import progressbar
import urllib.request


pbar = None


def show_progress(block_num, block_size, total_size):
    global pbar
    if pbar is None:
        pbar = progressbar.ProgressBar(maxval=total_size)
        pbar.start()

    downloaded = block_num * block_size
    if downloaded < total_size:
        pbar.update(downloaded)
    else:
        pbar.finish()
        pbar = None


urllib.request.urlretrieve(model_url, model_file, show_progress)

相关文章