我想创建一个.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!")
找到了怎么做
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")