mirror of
https://github.com/Arrowar/StreamingCommunity.git
synced 2025-06-07 20:15:24 +00:00
66 lines
1.9 KiB
Python
66 lines
1.9 KiB
Python
# 24.01.2023
|
|
|
|
# Class
|
|
from Src.Util.console import console
|
|
|
|
# Import
|
|
import subprocess, os, requests, zipfile, sys, ctypes, os, sys
|
|
|
|
# Variable
|
|
|
|
|
|
# [ func ]
|
|
def isAdmin():
|
|
try:
|
|
is_admin = (os.getuid() == 0)
|
|
except AttributeError:
|
|
is_admin = ctypes.windll.shell32.IsUserAnAdmin() != 0
|
|
return is_admin
|
|
|
|
def download_ffmpeg():
|
|
|
|
# Specify the URL for the FFmpeg binary zip file for Windows
|
|
ffmpeg_url = "https://www.gyan.dev/ffmpeg/builds/ffmpeg-git-full.7z"
|
|
|
|
# Name of the directory where FFmpeg will be extracted
|
|
ffmpeg_dir = "ffmpeg"
|
|
console.print("[yellow]Downloading FFmpeg...[/yellow]")
|
|
|
|
# Download the FFmpeg zip file
|
|
response = requests.get(ffmpeg_url)
|
|
os.makedirs(ffmpeg_dir, exist_ok=True)
|
|
|
|
# Save the zip file
|
|
zip_file_path = os.path.join(ffmpeg_dir, "ffmpeg.zip")
|
|
with open(zip_file_path, "wb") as zip_file:
|
|
zip_file.write(response.content)
|
|
|
|
with zipfile.ZipFile(zip_file_path, "r") as zip_ref:
|
|
zip_ref.extractall(ffmpeg_dir)
|
|
|
|
# Add the FFmpeg directory to the system PATH
|
|
ffmpeg_bin_dir = os.path.join(os.getcwd(), ffmpeg_dir, "bin")
|
|
os.environ["PATH"] += os.pathsep + ffmpeg_bin_dir
|
|
os.remove(zip_file_path)
|
|
|
|
def check_ffmpeg():
|
|
|
|
console.print("[green]Checking ffmpeg ...")
|
|
|
|
try:
|
|
subprocess.run(["ffmpeg", "-version"], stdout=subprocess.PIPE, stderr=subprocess.PIPE, check=True)
|
|
console.print("[blue]FFmpeg [white]=> [red]Find")
|
|
except:
|
|
try:
|
|
console.print("[cyan]FFmpeg is not in the PATH. Downloading and adding to the PATH...[/cyan]")
|
|
|
|
if not isAdmin():
|
|
console.log("[red]You need to be admin to proceed!")
|
|
sys.exit(0)
|
|
|
|
download_ffmpeg()
|
|
sys.exit(0)
|
|
except:
|
|
console.print("[red]Unable to download or add FFmpeg to the PATH.[/red]")
|
|
sys.exit(0)
|