ninjajiraiya
/
devine
forked from DRMTalks/devine
1
0
Fork 0
Code Pull Requests 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.pssh import PSSH
from requests import Session
from tqdm import tqdm
from devine.core.constants import AnyTrack
from devine.core.downloaders import aria2c
@ -392,7 +393,7 @@ class DASH:
res.raise_for_status()
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_save_path = (save_dir / segment_filename).with_suffix(".mp4")
@ -447,7 +448,7 @@ class DASH:
res.raise_for_status()
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_save_path = (save_dir / segment_filename).with_suffix(".mp4")