Add python Windows ffmpeg downloads

Tools/download_ffmpeg.py

@@ -1,43 +1,119 @@
-import zipfile
+from fileinput import filename
+import pathlib
+from zipfile import ZipFile
+from pathlib import Path
+import py7zr
+import platform
 from urllib.request import urlopen
 import os
+from shutil import move, rmtree
+from util import cd_repo_root_path
 
-FFMPEG_VERSION ="5.0.1"
+FFMPEG_VERSION = "5.0.1"
 
-def download_and_extract(file_base_path, name):
-    print("Download: ", name)
 
-    download_server_base_url = 'https://evermeet.cx/ffmpeg/'
-    filedata = urlopen(download_server_base_url + name)
+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(file_base_path, name)
+    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)
 
-    extraction_path = os.path.join(file_base_path, "../Common/ffmpeg")
+    return path_and_filename
 
-    print("Extract to:", extraction_path)
-    with zipfile.ZipFile(path_and_filename,"r") as zip_ref:
+
+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("Delete tmp file: ", path_and_filename)
-    os.remove(path_and_filename) 
+    print(f"Delete tmp file: {path_and_filename}")
+    os.remove(path_and_filename)
 
-def download_prebuild_ffmpeg():
-    ffmpeg_7zip_name  = f'ffmpeg-{FFMPEG_VERSION}.zip'
-    ffprobe_7zip_name = f'ffprobe-{FFMPEG_VERSION}.zip'
-    current_path = os.path.join(os.path.abspath(os.getcwd()),"")
-    extraction_path = os.path.abspath(os.path.join(current_path, "../Common/ffmpeg"))
 
-    if not os.path.exists(extraction_path):
-        os.makedirs(extraction_path)
+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)
 
-    download_and_extract(current_path, ffmpeg_7zip_name)
-    download_and_extract(current_path, ffprobe_7zip_name)
+    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])))
+    ffprobe = os.path.join(extraction_path,"ffprobe.exe")
+    print(f"Delete not needed {ffprobe}")
+    os.remove(ffprobe)
+    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-{FFMPEG_VERSION}.zip'
+    ffprobe_zip_name = f'ffprobe-{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)
 
 
 if __name__ == "__main__":
-    download_prebuild_ffmpeg()
 
+    # Make sure the script is always started from the same folder
+    root_path = cd_repo_root_path()
+    tools_path = os.path.join(root_path, "Tools")
+    extraction_path = os.path.join(root_path, "Common/ffmpeg")
+
+    if os.path.exists(extraction_path):
+        rmtree(extraction_path)
+        
+    os.makedirs(extraction_path)
+    
+
+    if platform.system() == "Windows":
+        download_prebuild_ffmpeg_windows(extraction_path)
+    elif platform.os.system() == "Darwin":
+        download_prebuild_ffmpeg_mac(extraction_path)