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

Prepare DRM on URL tracks if they already have DRM 

Previously it would only prepare the DRM, if it had to find DRM from the init data. Now it prepares DRM if already pre-provided with DRM objects.
This commit is contained in:
rlaphoenix 2023-03-08 21:34:08 +00:00
parent d73256f1b3
commit 8337162991
1 changed files with 11 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
devine/commands/dl.py
View File

@ -790,13 +790,20 @@ class dl:
# the service might not have explicitly defined the `drm` property
# try find widevine DRM information from the init data of URL
try:
drm = Widevine.from_track(track, service.session)
track.drm = [Widevine.from_track(track, service.session)]
except Widevine.Exceptions.PSSHNotFound:
# it might not have Widevine DRM, or might not have found the PSSH
self.log.warning("No Widevine PSSH was found for this track, is it DRM free?")
else:
track.drm = [drm]
if track.drm:
drm = track.drm[0] # just use the first supported DRM system for now
if isinstance(drm, Widevine):
# license and grab content keys
if not prepare_drm:
raise ValueError("prepare_drm func must be supplied to use Widevine DRM")
prepare_drm(drm)
else:
drm = None
asyncio.run(aria2c(
uri=track.url,
@ -808,8 +815,7 @@ class dl:
track.path = save_path
if track.drm:
drm = track.drm[0] # just use the first supported DRM system for now
if drm:
drm.decrypt(save_path)
track.drm = None
if callable(track.OnDecrypted):