python-3.x - tqdm 以人类可读的单位显示下载

Question

我正在使用 tqdm 并请求在 Python 中管理文件下载。但是我不知道如何让 tqdm 以人类可读的格式（即 MB/s）显示进度条。

这是我的代码

import requests
import os
from tqdm import tqdm

def download_file(url, local_path="./"):
    local_filename = url.split('/')[-1]
    path = local_path + local_filename

    r = requests.get(url, stream=True)
    total_size = int(r.headers.get('content-length', 0))
    chunk_size = 32*1024
    with open(local_filename, 'wb') as f:
        # 1KB = 1024 bytes
        for chunk in tqdm(r.iter_content(chunk_size), total=total_size, unit_scale=True, 
                          unit_divisor=1024):
            if chunk:
                f.write(chunk)

    return path

weightLink = "https://pjreddie.com/media/files/yolov3.weights"
weigthPath = r"/PyTorch-YOLOv3/weights/"
weightLink = "https://pjreddie.com/media/files/yolov3.weights"
download_file(weightLink, weigthPath)

使用上面的代码，下载大小的进度是错误的，我看到下载速度在it/s. 我希望一切都以MB/s.

标签： python-3.xpython-requeststqdm

Answer 1

为 tqdm 传递额外的参数unit='B', unit_scale=True, unit_divisor=1024。

例子：

from tqdm import tqdm

for i in tqdm(range(9999999), unit='B', unit_scale=True, unit_divisor=1024):
    pass

输出：

100%|███████████████████████████████████████████████ ██████████████████████████████████████████████████ █████████████████| 9.54M/9.54M [00:03<00:00, 2.66MB/s]