如何使用 tqdm 使用 Python 下载文件

Question

我想制作一个从包含进度条的链接下载图像的 .py 文件，我可以用 tdqm 来做吗？

这是我到目前为止

from tqdm import tqdm
import requests

chunk_size = 1024

url = "example.com"

r = requests.get(url, stream = True)

total_size = int(r.headers['content-length'])
filename = url.split('/')[-1]

with open(filename, 'wb') as f:
    for data in tqdm(iterable = r.iter_content(chunk_size = chunk_size)):         
        total = total_size/chunk_size, unit = 'KB')
        f.write(data)
print("Download complete!")

Answer 1

了解如何去做

from tqdm import tqdm
import requests
import math
url = "http://ipv4.download.thinkbroadband.com/5MB"
r = requests.get(url, stream=True)

total_size = int(r.headers.get('content-length', 0))
block_size = 1024
wrote = 0 
with open('output.bin', 'wb') as f:
    for data in tqdm(r.iter_content(block_size):
        total=math.ceil(total_size//block_size) , unit='KB', unit_scale=True)
        wrote = wrote  + len(data)
        f.write(data)
if total_size != 0 and wrote != total_size:
    print("ERROR, something went wrong")