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

HLS: Ensure all segments to decrypt in range exist

This commit is contained in:
rlaphoenix 2024-02-16 16:49:13 +00:00
parent 7587243aa2
commit 2388c85894
1 changed files with 12 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
devine/core/manifests/hls.py
View File

@ -333,18 +333,25 @@ class HLS:
drm = encryption_data[2] drm = encryption_data[2]
first_segment_i = encryption_data[0] first_segment_i = encryption_data[0]
last_segment_i = max(0, i - int(not include_this_segment)) last_segment_i = max(0, i - int(not include_this_segment))
range_len = (last_segment_i - first_segment_i) + 1
segment_range = f"{str(first_segment_i).zfill(name_len)}-{str(last_segment_i).zfill(name_len)}" segment_range = f"{str(first_segment_i).zfill(name_len)}-{str(last_segment_i).zfill(name_len)}"
merged_path = segment_save_dir / f"{segment_range}{Path(segments[last_segment_i].uri).suffix}" merged_path = segment_save_dir / f"{segment_range}{Path(segments[last_segment_i].uri).suffix}"
decrypted_path = segment_save_dir / f"{merged_path.stem}_decrypted{merged_path.suffix}" decrypted_path = segment_save_dir / f"{merged_path.stem}_decrypted{merged_path.suffix}"
merge( files = [
to=merged_path,
via=[
file file
for file in sorted(segment_save_dir.iterdir()) for file in sorted(segment_save_dir.iterdir())
if file.stem.isdigit() and first_segment_i <= int(file.stem) <= last_segment_i if file.stem.isdigit() and first_segment_i <= int(file.stem) <= last_segment_i
], ]
if not files:
raise ValueError(f"None of the segment files for {segment_range} exist...")
elif len(files) != range_len:
raise ValueError(f"Missing {range_len - len(files)} segment files for {segment_range}...")
merge(
to=merged_path,
via=files,
delete=True, delete=True,
include_map_data=True include_map_data=True
) )