快捷導(dǎo)航

利用Python展示文件下載進(jìn)度條

更新時間：2021年11月05日 15:40:24 作者：米洛丶

這篇文章主要介紹了Python展示文件下載進(jìn)度條，requests庫相信大家都用過，做接口測試少不了它。其實我們平時下載文件，也可以用requests做到的，通過一些地址，下面我們就來看看具體是什么樣的地址及詳細(xì)內(nèi)容,需要的朋友可以參考一下

1、前言

大家在用Python寫一些小程序的時候，經(jīng)常都會用到文件下載，對于一些較小的文件，大家可能不太在乎文件的下載進(jìn)度，因為一會就下載完畢了。

但是當(dāng)文件較大，比如下載chromedriver的時候，我們?nèi)绻軌蚩吹较螺d的進(jìn)度條，那該多么友好。畢竟在npm，pip安裝包的時候都有類似的進(jìn)度條。

那筆者今天就給大家分享一個展示文件下載進(jìn)度條的方法。

2、requests

requests庫相信大家都用過，做接口測試少不了它。其實我們平時下載文件，也可以用requests做到的，比如有這樣一個地址:https://autoupdate.termius.com/windows/Termius.exe

我們要下載它，應(yīng)該怎么做呢？

這里來寫一下偽代碼，非常好懂。

# 第一步: 訪問這個鏈接
import requests

r = requests.get("https://autoupdate.termius.com/windows/Termius.exe")

# 第二步: 獲取返回的文件內(nèi)容，并寫到本地

with open(r"./termius.exe", "wb") as f:
    f.write(r.content)

由于我這個文件很大，所以過程很漫長，我這邊就不展示具體細(xì)節(jié)了。大家可以找個幾M的小文件試試。放上最終結(jié)果：

可以看到，執(zhí)行完畢，文件也獲取到了

3、思考

有沒有覺得，這樣下載文件很單調(diào)，我也不知道進(jìn)度，比如下載了多少了，特別是針對大一點(diǎn)的文件，一直這樣等著我以為他斷開連接了呢。

如果咱們能在下載文件的時候，顯示出進(jìn)度條該多好，比如迅雷/百度網(wǎng)盤那樣子的。

只需要安裝一個庫即可
要做到這些，我們只需要安裝tqdm庫就行了。

pip install tqdm

下面是帶注釋的示例代碼:

import requests
from tqdm import tqdm

def download(url: str, fname: str):
    # 用流stream的方式獲取url的數(shù)據(jù)
    resp = requests.get(url, stream=True)
    # 拿到文件的長度，并把total初始化為0
    total = int(resp.headers.get('content-length', 0))
    # 打開當(dāng)前目錄的fname文件(名字你來傳入)
    # 初始化tqdm，傳入總數(shù)，文件名等數(shù)據(jù)，接著就是寫入，更新等操作了
    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的內(nèi)容基本都是固定寫法，大家不用疑惑，注意好requests等自己熟悉的部分就好。

我們來看看gif效果圖，是不是狂炫酷霸吊炸天（別好奇我的cmd為何這么帥氣，我是不會告訴你滴，除非你點(diǎn)贊）: