DRMTalks
/
devine
mirror of https://github.com/devine-dl/devine.git
1
0
Fork 1
Code Issues Packages Projects Releases Wiki Activity

Add missing tqdm progress bar to DASH downloads

This commit is contained in:
rlaphoenix 2023-02-21 06:35:40 +00:00
parent 61270d3af4
commit 50193856c2
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
devine/core/manifests/dash.py
View File

@ -18,6 +18,7 @@ from langcodes import Language, tag_is_valid
from pywidevine.cdm import Cdm as WidevineCdm from pywidevine.cdm import Cdm as WidevineCdm
from pywidevine.pssh import PSSH from pywidevine.pssh import PSSH
from requests import Session from requests import Session
from tqdm import tqdm
from devine.core.constants import AnyTrack from devine.core.constants import AnyTrack
from devine.core.downloaders import aria2c from devine.core.downloaders import aria2c
@ -392,7 +393,7 @@ class DASH:
res.raise_for_status() res.raise_for_status()
init_data = res.content init_data = res.content
for i, segment_url in enumerate(segment_urls): for i, segment_url in enumerate(tqdm(segment_urls, unit="segments")):
segment_filename = str(i).zfill(len(str(len(segment_urls)))) segment_filename = str(i).zfill(len(str(len(segment_urls))))
segment_save_path = (save_dir / segment_filename).with_suffix(".mp4") segment_save_path = (save_dir / segment_filename).with_suffix(".mp4")
@ -447,7 +448,7 @@ class DASH:
res.raise_for_status() res.raise_for_status()
init_data = res.content init_data = res.content
for i, segment_url in enumerate(segment_list.findall("SegmentURL")): for i, segment_url in enumerate(tqdm(segment_list.findall("SegmentURL"), unit="segments")):
segment_filename = str(i).zfill(len(str(len(segment_urls)))) segment_filename = str(i).zfill(len(str(len(segment_urls))))
segment_save_path = (save_dir / segment_filename).with_suffix(".mp4") segment_save_path = (save_dir / segment_filename).with_suffix(".mp4")