mirror of
https://gitlab.com/kelteseth/ScreenPlay.git
synced 2024-09-01 16:19:47 +02:00
156 lines
5.4 KiB
Python
156 lines
5.4 KiB
Python
#!/usr/bin/python3
|
|
# SPDX-License-Identifier: LicenseRef-EliasSteurerTachiom OR AGPL-3.0-only
|
|
from fileinput import filename
|
|
import sys
|
|
from zipfile import ZipFile
|
|
import platform
|
|
from urllib.request import urlopen
|
|
import subprocess
|
|
import os
|
|
import defines
|
|
from shutil import move, rmtree
|
|
from util import cd_repo_root_path
|
|
from sys import stdout
|
|
from defines import FFMPEG_VERSION
|
|
|
|
stdout.reconfigure(encoding='utf-8')
|
|
|
|
def download(
|
|
download_server_base_url,
|
|
extraction_path,
|
|
name) -> str:
|
|
|
|
url = download_server_base_url + name
|
|
print(f"Downloading {name} from: {url}")
|
|
filedata = urlopen(url)
|
|
datatowrite = filedata.read()
|
|
path_and_filename = os.path.join(extraction_path, name)
|
|
|
|
print("Save tmp file: ", path_and_filename)
|
|
with open(path_and_filename, 'wb') as f:
|
|
f.write(datatowrite)
|
|
|
|
return path_and_filename
|
|
|
|
|
|
def extract_zip(extraction_path, path_and_filename):
|
|
print(f"Extract {path_and_filename} to {extraction_path}")
|
|
|
|
with ZipFile(path_and_filename, "r") as zip_ref:
|
|
zip_ref.extractall(extraction_path)
|
|
|
|
print(f"Delete tmp file: {path_and_filename}")
|
|
os.remove(path_and_filename)
|
|
|
|
|
|
def extract_zip_executables(extraction_path, path_and_filename):
|
|
files = []
|
|
listOfFileNames = []
|
|
with ZipFile(path_and_filename, 'r') as zipObj:
|
|
# Get a list of all archived file names from the zip
|
|
listOfFileNames = zipObj.namelist()
|
|
# Iterate over the file names
|
|
for fileName in listOfFileNames:
|
|
# Check filename endswith csv
|
|
if fileName.endswith('.exe'):
|
|
# Extract a single file from zip
|
|
zipObj.extract(fileName, extraction_path)
|
|
print(f"Extract {fileName} to {extraction_path}")
|
|
files.append(fileName)
|
|
|
|
for file in files:
|
|
abs_file_path = os.path.join(extraction_path, file)
|
|
move(abs_file_path, extraction_path)
|
|
print(f"Move {abs_file_path} to {extraction_path}")
|
|
|
|
print(f"Delete tmp zip file: {path_and_filename}")
|
|
os.remove(path_and_filename)
|
|
# We need the first folder name of a file:
|
|
# ffmpeg-5.0.1-essentials_build/bin/ffmpeg.exe
|
|
empty_ffmpeg_folder = os.path.join(extraction_path,os.path.dirname(os.path.dirname(files[0])))
|
|
ffplay = os.path.join(extraction_path,"ffplay.exe")
|
|
print(f"Delete not needed {ffplay}")
|
|
os.remove(ffplay)
|
|
print(f"Delete empty folder: {empty_ffmpeg_folder}")
|
|
rmtree(empty_ffmpeg_folder)
|
|
|
|
|
|
def download_prebuild_ffmpeg_mac(extraction_path: str):
|
|
|
|
print("Setup ffmpeg mac")
|
|
ffmpeg_zip_name = f'ffmpeg-{defines.FFMPEG_VERSION}.zip'
|
|
ffprobe_zip_name = f'ffprobe-{defines.FFMPEG_VERSION}.zip'
|
|
download_server_base_url = 'https://evermeet.cx/ffmpeg/'
|
|
|
|
ffmpeg_path_and_filename = download(
|
|
download_server_base_url, extraction_path, ffmpeg_zip_name)
|
|
extract_zip(extraction_path, ffmpeg_path_and_filename)
|
|
|
|
ffprobe_path_and_filename = download(
|
|
download_server_base_url, extraction_path, ffprobe_zip_name)
|
|
extract_zip(extraction_path, ffprobe_path_and_filename)
|
|
|
|
|
|
def download_prebuild_ffmpeg_windows(extraction_path: str):
|
|
|
|
print("Setup ffmpeg Windows")
|
|
ffmpeg_shared_7zip_name = "ffmpeg-release-essentials.zip"
|
|
download_server_base_url = 'https://www.gyan.dev/ffmpeg/builds/'
|
|
|
|
ffmpeg_path_and_filename = download(
|
|
download_server_base_url, extraction_path, ffmpeg_shared_7zip_name)
|
|
|
|
extract_zip_executables(extraction_path, ffmpeg_path_and_filename)
|
|
|
|
def execute() ->bool:
|
|
# Make sure the script is always started from the same folder
|
|
root_path = cd_repo_root_path()
|
|
extraction_path = os.path.join(root_path, "ThirdParty/ffmpeg")
|
|
ffmpeg_binary_path = os.path.join(extraction_path, "ffmpeg") # Adjust this if FFmpeg binary is inside another subdirectory
|
|
|
|
if sys.platform == "win32":
|
|
ffmpeg_binary_path += ".exe"
|
|
|
|
# Check if ffmpeg is already installed and matches the required version
|
|
if os.path.isfile(ffmpeg_binary_path):
|
|
result = subprocess.run([ffmpeg_binary_path, "-version"], capture_output=True, text=True)
|
|
output = result.stdout
|
|
version_line = next((line for line in output.split('\n') if 'ffmpeg version' in line), None)
|
|
if version_line:
|
|
installed_version = version_line.split(' ')[2].split('-')[0]
|
|
if installed_version == FFMPEG_VERSION:
|
|
print(f"FFmpeg version {installed_version} is already installed.")
|
|
return True
|
|
else:
|
|
print(f"FFmpeg version {installed_version} found, but version {FFMPEG_VERSION} is required.")
|
|
|
|
try:
|
|
if os.path.exists(extraction_path):
|
|
rmtree(extraction_path)
|
|
print("Directory removed successfully.")
|
|
except Exception as e:
|
|
print(f"An error occurred while trying to remove the directory: {str(e)}")
|
|
return False
|
|
|
|
try:
|
|
if os.path.exists(extraction_path):
|
|
rmtree(extraction_path)
|
|
print("Directory removed successfully.")
|
|
except Exception as e:
|
|
print(f"An error occurred while trying to remove the directory: {str(e)}")
|
|
return False
|
|
|
|
|
|
os.makedirs(extraction_path)
|
|
|
|
if platform.system() == "Windows":
|
|
download_prebuild_ffmpeg_windows(extraction_path)
|
|
elif platform.system() == "Darwin":
|
|
download_prebuild_ffmpeg_mac(extraction_path)
|
|
|
|
return True
|
|
|
|
|
|
if __name__ == "__main__":
|
|
execute()
|