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

PSSH: Set Key IDs more effectively via set_key_ids() 

This reduces reading complexity of why and when pssh.set_key_ids() was being run. Generally less code repetition effectively.
This commit is contained in:
rlaphoenix 2022-11-18 09:40:55 +00:00
parent 05b30b3a89
commit 1b40c2b369
1 changed files with 9 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
pywidevine/pssh.py
View File

@ -165,18 +165,21 @@ class PSSH:
f"Expecting init_data to be {WidevinePsshData}, hex, base64, or bytes, not {init_data!r}" f"Expecting init_data to be {WidevinePsshData}, hex, base64, or bytes, not {init_data!r}"
) )
box = Box.parse(Box.build(dict( pssh = cls(Box.parse(Box.build(dict(
type=b"pssh", type=b"pssh",
version=version, version=version,
flags=flags, flags=flags,
system_ID=PSSH.SystemId.Widevine, system_ID=PSSH.SystemId.Widevine,
key_IDs=key_ids if key_ids and version == 1 else None,
init_data=[init_data, b""][init_data is None] init_data=[init_data, b""][init_data is None]
))) # key_IDs should not be set yet
))))
pssh = cls(box) if key_ids:
# We must reinforce the version because pymp4 forces v0 if key_IDs is not set.
if key_ids and version == 0: # The set_key_ids() func will set it efficiently in both init_data and the box where needed.
# The version must be reinforced ONLY if we have key_id data or there's a possibility of making
# a v1 PSSH box, that did not have key_IDs set in the PSSH box.
pssh.version = version
pssh.set_key_ids(key_ids) pssh.set_key_ids(key_ids)
return pssh return pssh