68 lines
3.3 KiB
Python
68 lines
3.3 KiB
Python
import os
|
|
import yaml
|
|
import requests
|
|
|
|
|
|
|
|
def check_for_wvd_cdm():
|
|
with open(f'{os.getcwd()}/configs/config.yaml', 'r') as file:
|
|
config = yaml.safe_load(file)
|
|
if config['default_wv_cdm'] == '':
|
|
answer = ' '
|
|
while answer[0].upper() != 'Y' and answer[0].upper() != 'N':
|
|
answer = input('No default Widevine CDM specified, would you like to download one from The CDM Project? (Y)es/(N)o: ')
|
|
if answer[0].upper() == 'Y':
|
|
response = requests.get(url='https://cdm-project.com/CDRM-Team/CDMs/raw/branch/main/Widevine/L3/public.wvd')
|
|
if response.status_code == 200:
|
|
with open(f'{os.getcwd()}/configs/CDMs/WV/public.wvd', 'wb') as file:
|
|
file.write(response.content)
|
|
config['default_wv_cdm'] = 'public'
|
|
with open(f'{os.getcwd()}/configs/config.yaml', 'w') as file:
|
|
yaml.dump(config, file)
|
|
print("Successfully downloaded Widevine CDM")
|
|
else:
|
|
exit(f"Download failed, please try again or place a .wvd file in {os.getcwd()}/configs/CDMs/WV and specify the name in {os.getcwd()}/configs/config.yaml")
|
|
if answer[0].upper() == 'N':
|
|
exit(f"Place a .wvd file in {os.getcwd()}/configs/CDMs/WV and specify the name in {os.getcwd()}/configs/config.yaml")
|
|
else:
|
|
base_name = config["default_wv_cdm"]
|
|
if not base_name.endswith(".wvd"):
|
|
base_name += ".wvd"
|
|
if os.path.exists(f'{os.getcwd()}/configs/CDMs/WV/{base_name}'):
|
|
return
|
|
else:
|
|
exit(f"Widevine CDM {base_name} does not exist in {os.getcwd()}/configs/CDMs/WV")
|
|
|
|
def check_for_prd_cdm():
|
|
with open(f'{os.getcwd()}/configs/config.yaml', 'r') as file:
|
|
config = yaml.safe_load(file)
|
|
if config['default_pr_cdm'] == '':
|
|
answer = ' '
|
|
while answer[0].upper() != 'Y' and answer[0].upper() != 'N':
|
|
answer = input('No default PlayReady CDM specified, would you like to download one from The CDM Project? (Y)es/(N)o: ')
|
|
if answer[0].upper() == 'Y':
|
|
response = requests.get(url='https://cdm-project.com/CDRM-Team/CDMs/raw/branch/main/Playready/SL2000/public.prd')
|
|
if response.status_code == 200:
|
|
with open(f'{os.getcwd()}/configs/CDMs/PR/public.prd', 'wb') as file:
|
|
file.write(response.content)
|
|
config['default_pr_cdm'] = 'public'
|
|
with open(f'{os.getcwd()}/configs/config.yaml', 'w') as file:
|
|
yaml.dump(config, file)
|
|
print("Successfully downloaded PlayReady CDM")
|
|
else:
|
|
exit(f"Download failed, please try again or place a .prd file in {os.getcwd()}/configs/CDMs/PR and specify the name in {os.getcwd()}/configs/config.yaml")
|
|
if answer[0].upper() == 'N':
|
|
exit(f"Place a .prd file in {os.getcwd()}/configs/CDMs/PR and specify the name in {os.getcwd()}/configs/config.yaml")
|
|
else:
|
|
base_name = config["default_pr_cdm"]
|
|
if not base_name.endswith(".prd"):
|
|
base_name += ".prd"
|
|
if os.path.exists(f'{os.getcwd()}/configs/CDMs/PR/{base_name}'):
|
|
return
|
|
else:
|
|
exit(f"PlayReady CDM {base_name} does not exist in {os.getcwd()}/configs/CDMs/WV")
|
|
|
|
|
|
def check_for_cdms():
|
|
check_for_wvd_cdm()
|
|
check_for_prd_cdm() |