使用requests和tqdm實現簡單的下載視頻並顯示進度條

使用requests下載視頻的同時顯示進度條,可以使用tqdm庫,關於tqdm庫的教程搜索就有很多。
這個簡單的demo直接上代碼

import os
import requests
from tqdm import tqdm

url = 'https://vd4.bdstatic.com/mda-jhdv4x3jakf2n8d3/sc/mda-jhdv4x3jakf2n8d3.mp4?auth_key=1565846233-0-0-d059103292b1a6ab006a79da979a32e7&bcevod_channel=searchbox_feed&pd=unknown&abtest=all'
dts = 'test.mp4'
headers = {
    "User-Agent": "Mozilla/5.0 (Windows NT 6.2; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Maxthon/4.3.2.1000 Chrome/30.0.1599.101 Safari/537.36"}
# 設置stream=True,避免文件過大,內存不夠,當前不會立即下載,在調用時下載
res = requests.get(url, stream=True, headers=headers)
# 獲取視頻總長度
length = float(res.headers['content-length'])
f = open(dts, 'wb')
'''
使用手動設置更新
total 設置總大小
initial 當前操作文件的大小
desc 進度條前的描述
ncols=120設置進度條顯示長度  
nit_scale 如果設置,迭代的次數會自動按照十、百、千來添加前綴,默認爲false
'''
pbar = tqdm(total=length, initial=os.path.getsize(dts), unit_scale=True, desc=dts, ncols=120)
for chuck in res.iter_content(chunk_size=512):
    f.write(chuck)
    # 手動更新的大小
    pbar.update(512)
f.close()

備註:我沒有找到使用tqdm對生成器操作的example,因此手動的設置tqdm的使用。之後對tqdm瞭解更多了再來更新。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章