Mirrors/udemy-downloader
1
0
Fork 0
mirror of https://cdm-project.com/Download-Tools/udemy-downloader.git synced 2025-05-04 15:44:26 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix: KID extraction error handling

Browse Source 
+ extract_kid method will raise an error if the file does not exist
+ If KID extraction fails, the lecture will be skipped instead of causing the program to exit
This commit is contained in:
Puyodead1 2021-11-22 11:36:05 -05:00
parent 59538b24ce
commit 0719800145
2 changed files with 15 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
main.py
View File

@ -918,11 +918,19 @@ def handle_segments(url, format_id, video_title,
print("Return code from the downloader was non-0 (error), skipping!") print("Return code from the downloader was non-0 (error), skipping!")
return return
try:
video_kid = extract_kid(video_filepath_enc) video_kid = extract_kid(video_filepath_enc)
print("KID for video file is: " + video_kid) print("KID for video file is: " + video_kid)
except Exception as e:
print(f"Error extracting video kid: {e}")
return
try:
audio_kid = extract_kid(audio_filepath_enc) audio_kid = extract_kid(audio_filepath_enc)
print("KID for audio file is: " + audio_kid) print("KID for audio file is: " + audio_kid)
except Exception as e:
print(f"Error extracting audio kid: {e}")
return
try: try:
decrypt(video_kid, video_filepath_enc, video_filepath_dec) decrypt(video_kid, video_filepath_enc, video_filepath_dec)

3
utils.py
View File

@ -2,6 +2,7 @@ import mp4parse
import codecs import codecs
import widevine_pssh_pb2 import widevine_pssh_pb2
import base64 import base64
import os
def extract_kid(mp4_file): def extract_kid(mp4_file):
""" """
@ -18,6 +19,8 @@ def extract_kid(mp4_file):
""" """
boxes = mp4parse.F4VParser.parse(filename=mp4_file) boxes = mp4parse.F4VParser.parse(filename=mp4_file)
if not os.path.exists(mp4_file):
raise Exception("File does not exist")
for box in boxes: for box in boxes:
if box.header.box_type == 'moov': if box.header.box_type == 'moov':
pssh_box = next(x for x in box.pssh if x.system_id == "edef8ba979d64acea3c827dcd51d21ed") pssh_box = next(x for x in box.pssh if x.system_id == "edef8ba979d64acea3c827dcd51d21ed")