下载文件进度提醒

发布时间 2023-05-28 14:33:50作者: 一枚码农

需要使用第三发包 tqdm

安装
pip install tqdm

效果展示
image

简单文件下载代码

import requests
from tqdm import tqdm


def download(link:str):
    """下载文件"""
    response = requests.get(link, stream=True)
    response.raise_for_status()

    file = link.split("/")[-1]
    total = int(response.headers.get('content-length', 0))
    with open(file, 'wb') as f, tqdm(
            desc=file,
            total=total,
            unit='iB',
            unit_scale=True,
            unit_divisor=1024,
    ) as bar:
        for chunk in response.iter_content(chunk_size=1024):
            if chunk:
                size = f.write(chunk)
                bar.update(size)

    return file


if __name__ == '__main__':
    link = ''
    download(link)