用Python写一个带进度条的简单下载器

一个带进度条的下载器（单进程单线程）

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