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

Use Widevine.from_init_data when downloading DASH 

This is required as some DASH manifests do not explicitly list the PSSH with the Widevine ContentProtection, only listing that its a supported.
This commit is contained in:
rlaphoenix 2023-02-21 08:02:32 +00:00
parent 8c312440a3
commit 8268825ba8
1 changed files with 22 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -392,6 +392,17 @@ class DASH:
)) ))
res.raise_for_status() res.raise_for_status()
init_data = res.content init_data = res.content
if not drm:
try:
drm = Widevine.from_init_data(init_data)
except Widevine.Exceptions.PSSHNotFound:
# it might not have Widevine DRM, or might not have found the PSSH
log.warning("No Widevine PSSH was found for this track, is it DRM free?")
else:
# license and grab content keys
if not license_widevine:
raise ValueError("license_widevine func must be supplied to use Widevine DRM")
license_widevine(drm)
for i, segment_url in enumerate(tqdm(segment_urls, unit="segments")): 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))))
@ -447,6 +458,17 @@ class DASH:
res = session.get(source_url) res = session.get(source_url)
res.raise_for_status() res.raise_for_status()
init_data = res.content init_data = res.content
if not drm:
try:
drm = Widevine.from_init_data(init_data)
except Widevine.Exceptions.PSSHNotFound:
# it might not have Widevine DRM, or might not have found the PSSH
log.warning("No Widevine PSSH was found for this track, is it DRM free?")
else:
# license and grab content keys
if not license_widevine:
raise ValueError("license_widevine func must be supplied to use Widevine DRM")
license_widevine(drm)
for i, segment_url in enumerate(tqdm(segment_list.findall("SegmentURL"), unit="segments")): 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))))