python - 使用 Tqdm 在下载文件时添加进度条

Question

我一直在尝试使用 python 3.6 中的 Tqdm 模块设置进度条，但似乎我已经完成了一半。

我的代码如下:

from tqdm import tqdm
import requests
import time

url = 'http://alpha.chem.umb.edu/chemistry/ch115/Mridula/CHEM%20116/documents/chapter_20au.pdf'

# Streaming, so we can iterate over the response.
r = requests.get(url, stream=True)
#Using The Url as a filename
local_filename = url.split('/')[-1]
# Total size in bytes.
total_size = int(r.headers.get('content-length', 0))
#Printing Total size in Bytes
print(total_size)

#TQDM
with open(local_filename, 'wb') as f:
    for data in tqdm(r.iter_content(chunk_size = 512), total=total_size, unit='B', unit_scale=True):
        f.write(data)

问题是，当我在 r.iter_content 中插入 chunk_size = 512 时，进度条在显示下载数据时根本不加载，但是当我完全删除 chunk_size = 512 并将括号留空，条形图加载完全正常，但下载速度非常糟糕。

我做错了什么？

Answer 1

您离目标不远，只是缺少了几乎所有使进度条相应工作的代码。假设您已经创建了界面，这是我用于进度条的方法。它下载文件并将其保存在桌面上(但您可以指定要保存的位置)。它只需要下载多少文件并将其除以总文件大小，然后使用该值来更新进度条。让我知道这段代码是否有帮助:

url = 'http://alpha.chem.umb.edu/chemistry/ch115/Mridula/CHEM%20116/documents/chapter_20au.pdf'
save = 'C:/Users/' + username + "/Desktop/"    
r = requests.get(url, stream=True)
total_size = int(r.headers["Content-Length"])
downloaded = 0  # keep track of size downloaded so far
chunkSize = 1024
bars = int(total_size / chunkSize)
print(dict(num_bars=bars))
with open(filename, "wb") as f:
    for chunk in tqdm(r.iter_content(chunk_size=chunkSize), total=bars, unit="KB",
                                  desc=filename, leave=True):
        f.write(chunk)
        downloaded += chunkSize  # increment the downloaded
        prog = ((downloaded * 100 / total_size))
        progress["value"] = (prog)  # *100 #Default max value of tkinter progress is 100
            
return

progress = 进度条