怎么用Python展示文件下载进度条

这篇文章主要介绍“怎么用Python展示文件下载进度条”,在日常操作中,相信很多人在怎么用Python展示文件下载进度条问题上存在疑惑,小编查阅了各式资料,整理出简单好用的操作方法,希望对大家解答”怎么用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的小文件试试。放上最终结果:

怎么用Python展示文件下载进度条

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

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为何这么帅气,我是不会告诉你滴,除非你点赞):

怎么用Python展示文件下载进度条

到此,关于“怎么用Python展示文件下载进度条”的学习就结束了,希望能够解决大家的疑惑。理论与实践的搭配能更好的帮助大家学习,快去试试吧!若想继续学习更多相关知识,请继续关注亿速云网站,小编会继续努力为大家带来更多实用的文章!

原创文章,作者:ItWorker,如若转载,请注明出处:https://blog.ytso.com/tech/dev/209079.html

(0)
上一篇 2021年12月7日 17:31
下一篇 2021年12月7日 17:31

相关推荐

发表回复

登录后才能评论