Mirrors/TPD-Keys
1
0
Fork 0
mirror of https://cdm-project.com/Decryption-Tools/TPD-Keys.git synced 2025-04-30 00:14:26 +02:00
Code Issues Packages Projects Releases Wiki Activity

Update binary_check.py

Browse Source 
fixed yt-dlp for Linux
This commit is contained in:
TPD94 2023-12-25 04:05:20 -05:00
parent 1ecbd46f0a
commit 48b2eae31d
1 changed files with 17 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
Helpers/binary_check.py
View File

@ -29,7 +29,7 @@ def create_binaries():
# Set binary dictionaries for Windows / Linux # Set binary dictionaries for Windows / Linux
windows_binaries = ["n_m3u8dl-re.exe", "mp4decrypt.exe", "ffmpeg.exe", "yt-dlp.exe"] windows_binaries = ["n_m3u8dl-re.exe", "mp4decrypt.exe", "ffmpeg.exe", "yt-dlp.exe"]
linux_binaries = ["n_m3u8dl-re", "mp4decrypt", "ffmpeg"] linux_binaries = ["N_m3u8DL-RE", "mp4decrypt", "ffmpeg", "yt-dlp"]
if operating_system == "Windows": if operating_system == "Windows":
binary_list = windows_binaries binary_list = windows_binaries
if operating_system == "Linux": if operating_system == "Linux":
@ -174,7 +174,7 @@ def create_binaries():
print() print()
# n_m3u8dl-re # n_m3u8dl-re
elif binary == "n_m3u8dl-re.exe" or binary == "n_m3u8dl-re": elif binary == "n_m3u8dl-re.exe" or binary == "N_m3u8DL-RE":
# Download n_m3u8dl-re zip file # Download n_m3u8dl-re zip file
if operating_system == "Windows": if operating_system == "Windows":
@ -259,7 +259,7 @@ def create_binaries():
stream=True) stream=True)
if operating_system == "Linux": if operating_system == "Linux":
yt_dlp_download = requests.get( yt_dlp_download = requests.get(
"https://github.com/yt-dlp/yt-dlp/releases/download/2023.11.16/yt-dlp_linux", "https://github.com/yt-dlp/yt-dlp/releases/download/2023.11.16/yt-dlp.tar.gz",
stream=True) stream=True)
total_size = int(yt_dlp_download.headers.get('content-length', 0)) total_size = int(yt_dlp_download.headers.get('content-length', 0))
if operating_system == "Windows": if operating_system == "Windows":
@ -270,13 +270,23 @@ def create_binaries():
download.write(data) download.write(data)
progress_bar.update(len(data)) progress_bar.update(len(data))
if operating_system == "Linux": if operating_system == "Linux":
with open(f"{os.getcwd()}/download/yt-dlp", 'wb') as download: with open(f"{os.getcwd()}/download/temp/yt-dlp.tar.gz", 'wb') as download:
with tqdm(total=total_size, unit='B', unit_scale=True, with tqdm(total=total_size, unit='B', unit_scale=True,
desc="Downloading yt-dlp") as progress_bar: desc="Downloading yt-dlp") as progress_bar:
for data in yt_dlp_download.iter_content(chunk_size=1024): for data in yt_dlp_download.iter_content(chunk_size=1024):
download.write(data) download.write(data)
progress_bar.update(len(data)) progress_bar.update(len(data))
# Untar yt-dlp if Linux
if operating_system == "Linux":
with tarfile.open(f"{os.getcwd()}/download/temp/yt-dlp.tar.gz", 'r:gz') as yt_dlp_tar_gz:
file_count = len(yt_dlp_tar_gz.getmembers())
with tqdm(total=file_count, unit='file',
desc=f"Extracting yt-dlp.tar.gz") as untar_gz_progress_bar:
for file in yt_dlp_tar_gz:
yt_dlp_tar_gz.extract(file, path=f"{os.getcwd()}/download/temp")
untar_gz_progress_bar.update(1)
# Copy yt-dlp binary to binaries if Windows # Copy yt-dlp binary to binaries if Windows
if operating_system == "Windows": if operating_system == "Windows":
shutil.copy2(f"{os.getcwd()}/download/yt-dlp.exe", shutil.copy2(f"{os.getcwd()}/download/yt-dlp.exe",
@ -284,9 +294,9 @@ def create_binaries():
# Copy yt-dlp binary to binaries if Linux # Copy yt-dlp binary to binaries if Linux
if operating_system == "Linux": if operating_system == "Linux":
shutil.copy2(f"{os.getcwd()}/download/yt-dlp", shutil.copy2(f"{os.getcwd()}/download/temp/yt-dlp/yt-dlp",
f"{os.getcwd()}/binaries") f"{os.getcwd()}/binaries")
os.chmod(f"{os.getcwd()}/binaries/yt-dlp", stat.S_IXUSR | stat.S_IXGRP | stat.S_IXOTH) subprocess.run(['chmod', '+x', f"{os.getcwd()}/binaries/yt-dlp"])
# Remove binary from download folder if Windows # Remove binary from download folder if Windows
if operating_system == "Windows": if operating_system == "Windows":
@ -294,7 +304,7 @@ def create_binaries():
# Remove binary from download folder if Linux # Remove binary from download folder if Linux
if operating_system == "Linux": if operating_system == "Linux":
os.remove(f"{os.getcwd()}/download/yt-dlp") shutil.rmtree(f"{os.getcwd()}/download/temp/yt-dlp")
# Print a new line # Print a new line
print() print()