Files
Voice-Cloning-App/application/check_ffmpeg.py

53 lines
1.5 KiB
Python
Raw Normal View History

2021-03-18 16:53:04 +00:00
import os
from sys import platform
from subprocess import check_output
import requests
from zipfile import ZipFile
FFMPEG_COMMAND = "ffmpeg -version"
FFMPEG_WINDOWS_INSTALL_PATH = os.path.abspath("ffmpeg\\bin")
FFMPEG_WINDOWS_URL = "https://www.gyan.dev/ffmpeg/builds/ffmpeg-release-essentials.zip"
FFMPEG_LINUX_INSTALL = "sudo apt-get install -y ffmpeg"
def is_ffmpeg_installed():
try:
check_output(FFMPEG_COMMAND.split(" "))
return True
except:
return False
def install_ffmpeg_windows():
r = requests.get(FFMPEG_WINDOWS_URL)
with open("ffmpeg.zip", "wb") as f:
f.write(r.content)
with ZipFile("ffmpeg.zip", "r") as zipf:
zipf.extractall()
ffmpeg_folder = [f for f in os.listdir() if f.startswith("ffmpeg-")][0]
os.rename(ffmpeg_folder, "ffmpeg")
os.remove("ffmpeg.zip")
def install_ffmpeg_linux():
check_output(FFMPEG_LINUX_INSTALL.split(" "))
def check_ffmpeg():
if not is_ffmpeg_installed():
if platform == "win32":
os.environ["PATH"] += os.pathsep + FFMPEG_WINDOWS_INSTALL_PATH
if not is_ffmpeg_installed():
print("INSTALLING FFMPEG")
install_ffmpeg_windows()
print("VERIFYING FFMPEG INSTALL")
assert is_ffmpeg_installed(), "FFMPEG did not install correctly"
else:
print("INSTALLING FFMPEG")
install_ffmpeg_linux()
print("VERIFYING FFMPEG INSTALL")
assert is_ffmpeg_installed(), "FFMPEG did not install correctly"