Python3使用requests模块实现显示下载进度的方法详解

在Python中，requests是一个常用的HTTP客户端库，可以用于发送HTTP请求和处理HTTP响应。在下载大文件时，可以使用requests库实现显示下载进度的功能。以下是详细讲解Python3使用requests模块实现显示下载进度的方法的攻略，包含两个例。

使用tqdm库实现显示下载进度

tqdm是一个Python进度条库，可以用于显示进度条和估计剩余时间。可以使用tqdm库结合requests库实现显示下载进度。以下是一个示例：

import requests
from tqdm import tqdm

url = 'https://download.pytorch.org/whl/cu111/torch_stable.html'
response = requests.get(url, stream=True)

total_size = int(response.headers.get('content-length', 0))
block_size = 1024
progress_bar = tqdm(total=total_size, unit='iB', unit_scale=True)

with open('torch_stable.html', 'wb') as file:
    for data in response.iter_content(block_size):
        progress_bar.update(len(data))
        file.write(data)

progress_bar.close()

在上面的示例中，我们使用requests库的get()函数下载一个文件。我们将stream参数设置为True，以便在下载时获取响应的流。然后，我们使用tqdm库创建一个进度条，并将总大小设置为响应头中的content-length。我们使用iter_content()函数迭代响应流，并在每次迭代时更新进度条。最后，我们使用close()函数关闭进度条。

使用urllib库实现显示下载进度

除了tqdm库，还可以使用urllib库实现显示下载进度。以下是一个示例：

import requests
import urllib.request

url = 'https://download.pytorch.org/whl/cu111/torch_stable.html'
response = requests.get(url, stream=True)

total_size = int(response.headers.get('content-length', 0))
block_size = 1024
progress = 0

with open('torch_stable.html', 'wb') as file:
    for data in response.iter_content(block_size):
        progress += len(data)
        file.write(data)
        done = int(50 * progress / total_size)
        print('\r[{}{}] {:.2f}%'.format('=' * done, ' ' * (50 - done), 100 * progress / total_size), end='')

print('\nDownload finished!')

在上面的示例中，我们使用requests库的get()函数下载一个文件。我们将stream参数设置为True，以便在下载时获取响应的流。然后，我们使用urllib库的request.urlretrieve()函数下载文件，并在每次迭代时更新进度条。最后，我们使用print()函数输出下载完成的消息。

总结

在Python中，可以使用tqdm库或urllib库结合requests库实现显示下载进度的功能。tqdm库可以创建一个进度条，并在每次迭代时更新进度条。urllib库的request.urlretrieve()函数可以下载文件，并在每次迭代时更新进度条。在使用这些库和函数时，需要注意安全性和性能问题，以确保代码的可靠性和效率。