一个带进度条的下载器(单进程单线程)
import requests
from tqdm import tqdm
def download(url, file_name):
headers = {'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/98.0.4758.102 Safari/537.36 Edg/98.0.1108.62'}
# 设置请求头
head = requests.head(url, headers=headers)
# 请求获取【响应头信息】
file_size = head.headers.get('Content-Length')
# 从响应头信息中提取【文件大小】(以B为单位)
if file_size is not None:
file_size = int(file_size)
# 判断提取到的文件大小是否为空
response = requests.get(url, headers=headers, stream=True)
# 请求获取下载内容(加入stream参数,暂不执行)
chunk_size = 1024
# 定义进度块大小
bar = tqdm(total=file_size, desc=f'下载文件 {file_name}')
# 创建tqdm对象(进度条),加入文件大小和文件名信息
with open(file_name, mode='wb') as f:
# 创建文件载体
for chunk in response.iter_content(chunk_size=chunk_size):
# 按照进度块大小提取字节流数据
f.write(chunk)
# 写入字节流数据
bar.update(chunk_size)
# 按照进度快大小更新进度条
bar.close()
# 关闭进度条
print(f'{file_name} 下载成功!')
if __name__ == '__main__':
URL = 'https://issuepcdn.baidupcs.com/issue/netdisk/MACguanjia/BaiduNetdisk_mac_4.6.0.dmg'
F