ninjajiraiya
/
devine
forked from DRMTalks/devine
1
0
Fork 0
Code Pull Requests Activity

Was not able to use the initialization from a DASH segment_list (#47) 

* Was not able to use the initialization from a DASH segment_list

* Check if initialization in DASH has attribute range
This commit is contained in:
Hollander_1908 2023-03-26 21:01:17 +02:00 committed by GitHub
parent 33a9c307f3
commit d894e5bbe0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -410,12 +410,17 @@ class DASH:
elif segment_list is not None: elif segment_list is not None:
init_data = None init_data = None
initialization = segment_list.find("Initialization") initialization = segment_list.find("Initialization")
if initialization: if initialization is not None:
source_url = initialization.get("sourceURL") source_url = initialization.get("sourceURL")
if source_url is None: if source_url is None:
source_url = rep_base_url source_url = rep_base_url
res = session.get(source_url) if initialization.get("range"):
headers = {"Range": f"bytes={initialization.get('range')}"}
else:
headers = None
res = session.get(url=source_url, headers=headers)
res.raise_for_status() res.raise_for_status()
init_data = res.content init_data = res.content
track_kid = track.get_key_id(init_data) track_kid = track.get_key_id(init_data)
@ -468,13 +473,13 @@ class DASH:
try: try:
if segment_range: if segment_range:
# aria2(c) doesn't support byte ranges, let's use python-requests (likely slower) # aria2(c) doesn't support byte ranges, let's use python-requests (likely slower)
r = session.get( res = session.get(
url=segment_uri, url=segment_uri,
headers={ headers={
"Range": f"bytes={segment_range}" "Range": f"bytes={segment_range}"
} }
) )
r.raise_for_status() res.raise_for_status()
segment_save_path.parent.mkdir(parents=True, exist_ok=True) segment_save_path.parent.mkdir(parents=True, exist_ok=True)
segment_save_path.write_bytes(res.content) segment_save_path.write_bytes(res.content)
else: else: