利用Python展示文件下载进度条(文件下载)

1、

大家在用Python写一些小程序的时候，经常都会用到文件下载，对于一些较小的文件，大家可能不太在乎文件的下载进度，因为一会就下载完毕了。

但是当文件较大，比如下载chromedriver的时候，我们如果能够看到下载的进度条，那该多么友好。毕竟在npm，pip安装包的时候都有类似的进度条。

那笔者今天就给大家分享一个展示文件下载进度条的方法。

2、requests

requests库相信大家都用过，做接口测试少不了它。其实我们平时下载文件，也可以用requests做到的，比如有这样一个地址:https://autoupdate.termius.com/windows/Termius.exe

我们要下载它，应该怎么做呢？

这里来写一下伪代码，非常好懂。

# 第一步: 访问这个链接
import requests
r = requests.get("https://autoupdate.termius.com/windows/Termius.exe")
# 第二步: 获取返回的文件内容，并写到本地
with open(r"./termius.exe", "wb") as f:
 f.write(r.content)

由于我这个文件很大，所以过程很漫长，我这边就不展示具体细节了。大家可以找个几M的小文件试试。放上最终结果：

可以看到，执行完毕，文件也获取到了

3、思考

有没有觉得，这样下载文件很单调，我也不知道进度，比如下载了多少了，特别是针对大一点的文件，一直这样等着我以为他断开连接了呢。

如果咱们能在下载文件的时候，显示出进度条该多好，比如迅雷/百度网盘那样子的。

只需要安装一个库即可
要做到这些，我们只需要安装tqdm库就行了。

pip install tqdm

下面是带注释的示例代码:

import requests
from tqdm import tqdm
def download(url: str, fname: str):
 # 用流stream的方式获取url的数据
 resp = requests.get(url, stream=True)
 # 拿到文件的长度，并把total初始化为0
 total = int(resp.headers.get('content-length', 0))
 # 打开当前目录的fname文件(名字你来传入)
 # 初始化tqdm，传入总数，文件名等数据，接着就是写入，更新等操作了
 with open(fname, 'wb') as file, tqdm(
  desc=fname,
  total=total,
  unit='iB',
  unit_scale=True,
  unit_divisor=1024,
 ) as bar:
  for data in resp.iter_content(chunk_size=1024):
size = file.write(data)
bar.update(size)

if __name__ == "__main__":
 # 下载文件，并传入文件名
 url = "https://autoupdate.termius.com/windows/Termius.exe"
 download("https://autoupdate.termius.com/windows/Termius.exe", "haha.exe")

代码比较简单，tqdm的内容基本都是固定写法，大家不用疑惑，注意好requests等自己熟悉的部分就好。

我们来看看gif效果图，是不是狂炫酷霸吊炸天（别好奇我的cmd为何这么帅气，我是不会告诉你滴，除非你点赞）: