mirror of
https://cdm-project.com/Download-Tools/udemy-downloader.git
synced 2025-04-30 04:14:24 +02:00
b922294135
- Removed .env support + Added TOML configuration file support + Added selenium with "undetected_chrome_driver" + Fix changing logging level not working + Updated README to reflect code changes
48 lines
1.1 KiB
Python
48 lines
1.1 KiB
Python
import base64
|
|
import codecs
|
|
import os
|
|
import random
|
|
import time
|
|
|
|
from selenium.webdriver.remote.webelement import WebElement
|
|
|
|
import mp4parse
|
|
import widevine_pssh_pb2
|
|
|
|
|
|
def extract_kid(mp4_file):
|
|
"""
|
|
Parameters
|
|
----------
|
|
mp4_file : str
|
|
MP4 file with a PSSH header
|
|
|
|
|
|
Returns
|
|
-------
|
|
String
|
|
|
|
"""
|
|
|
|
boxes = mp4parse.F4VParser.parse(filename=mp4_file)
|
|
if not os.path.exists(mp4_file):
|
|
raise Exception("File does not exist")
|
|
for box in boxes:
|
|
if box.header.box_type == 'moov':
|
|
pssh_box = next(x for x in box.pssh if x.system_id == "edef8ba979d64acea3c827dcd51d21ed")
|
|
hex = codecs.decode(pssh_box.payload, "hex")
|
|
|
|
pssh = widevine_pssh_pb2.WidevinePsshData()
|
|
pssh.ParseFromString(hex)
|
|
content_id = base64.b16encode(pssh.content_id)
|
|
return content_id.decode("utf-8")
|
|
|
|
# No Moof or PSSH header found
|
|
return None
|
|
|
|
|
|
def slow_type(element: WebElement, text: str):
|
|
for character in text:
|
|
element.send_keys(character)
|
|
delay = random.randint(1, 5) / 10
|
|
time.sleep(delay) |