mirror of
https://github.com/Zulko/moviepy.git
synced 2021-07-27 01:17:47 +03:00
0751d072d9
* Add module docstrings except in FXs * Update flake8 config * Remove 'find_latest_imagemagick_version.py' * Add tests modules docstrings * Don't execute doctests for tests/ modules * Prevent test error in Windows
123 lines
3.9 KiB
Python
123 lines
3.9 KiB
Python
"""Third party programs configuration for MoviePy."""
|
|
|
|
import os
|
|
import subprocess as sp
|
|
from pathlib import Path
|
|
|
|
from moviepy.tools import cross_platform_popen_params
|
|
|
|
|
|
if os.name == "nt":
|
|
import winreg as wr
|
|
|
|
try:
|
|
from dotenv import find_dotenv, load_dotenv
|
|
|
|
DOTENV = find_dotenv()
|
|
load_dotenv(DOTENV)
|
|
except ImportError:
|
|
DOTENV = None
|
|
|
|
FFMPEG_BINARY = os.getenv("FFMPEG_BINARY", "ffmpeg-imageio")
|
|
IMAGEMAGICK_BINARY = os.getenv("IMAGEMAGICK_BINARY", "auto-detect")
|
|
|
|
IS_POSIX_OS = os.name == "posix"
|
|
|
|
|
|
def try_cmd(cmd):
|
|
"""TODO: add documentation"""
|
|
try:
|
|
popen_params = cross_platform_popen_params(
|
|
{"stdout": sp.PIPE, "stderr": sp.PIPE, "stdin": sp.DEVNULL}
|
|
)
|
|
proc = sp.Popen(cmd, **popen_params)
|
|
proc.communicate()
|
|
except Exception as err:
|
|
return False, err
|
|
else:
|
|
return True, None
|
|
|
|
|
|
if FFMPEG_BINARY == "ffmpeg-imageio":
|
|
from imageio.plugins.ffmpeg import get_exe
|
|
|
|
FFMPEG_BINARY = get_exe()
|
|
|
|
elif FFMPEG_BINARY == "auto-detect":
|
|
|
|
if try_cmd(["ffmpeg"])[0]:
|
|
FFMPEG_BINARY = "ffmpeg"
|
|
elif not IS_POSIX_OS and try_cmd(["ffmpeg.exe"])[0]:
|
|
FFMPEG_BINARY = "ffmpeg.exe"
|
|
else: # pragma: no cover
|
|
FFMPEG_BINARY = "unset"
|
|
else:
|
|
success, err = try_cmd([FFMPEG_BINARY])
|
|
if not success:
|
|
raise IOError(
|
|
f"{err} - The path specified for the ffmpeg binary might be wrong"
|
|
)
|
|
|
|
if IMAGEMAGICK_BINARY == "auto-detect":
|
|
if os.name == "nt":
|
|
# Try a few different ways of finding the ImageMagick binary on Windows
|
|
try:
|
|
key = wr.OpenKey(wr.HKEY_LOCAL_MACHINE, "SOFTWARE\\ImageMagick\\Current")
|
|
IMAGEMAGICK_BINARY = wr.QueryValueEx(key, "BinPath")[0] + r"\magick.exe"
|
|
key.Close()
|
|
except Exception:
|
|
try:
|
|
imagemagick_path = sp.check_output(
|
|
r'dir /B /O-N "C:\Program Files\ImageMagick-*"',
|
|
shell=True,
|
|
encoding="utf-8",
|
|
).split("\n")[0]
|
|
IMAGEMAGICK_BINARY = sp.check_output( # pragma: no cover
|
|
rf'dir /B /S "C:\Program Files\{imagemagick_path}\*convert.exe"',
|
|
shell=True,
|
|
encoding="utf-8",
|
|
).split("\n")[0]
|
|
except Exception:
|
|
IMAGEMAGICK_BINARY = "unset"
|
|
|
|
if IMAGEMAGICK_BINARY in ["unset", "auto-detect"]:
|
|
if try_cmd(["convert"])[0]:
|
|
IMAGEMAGICK_BINARY = "convert"
|
|
elif not IS_POSIX_OS and try_cmd(["convert.exe"])[0]: # pragma: no cover
|
|
IMAGEMAGICK_BINARY = "convert.exe"
|
|
else: # pragma: no cover
|
|
IMAGEMAGICK_BINARY = "unset"
|
|
else:
|
|
if not os.path.exists(IMAGEMAGICK_BINARY):
|
|
raise IOError(f"ImageMagick binary cannot be found at {IMAGEMAGICK_BINARY}")
|
|
|
|
if not os.path.isfile(IMAGEMAGICK_BINARY):
|
|
raise IOError(f"ImageMagick binary found at {IMAGEMAGICK_BINARY} is not a file")
|
|
success, err = try_cmd([IMAGEMAGICK_BINARY])
|
|
if not success:
|
|
raise IOError(
|
|
f"{err} - The path specified for the ImageMagick binary might "
|
|
f"be wrong: {IMAGEMAGICK_BINARY}"
|
|
)
|
|
|
|
|
|
def check():
|
|
"""Check if moviepy has found the binaries of FFmpeg and ImageMagick."""
|
|
if try_cmd([FFMPEG_BINARY])[0]:
|
|
print(f"MoviePy: ffmpeg successfully found in '{FFMPEG_BINARY}'.")
|
|
else: # pragma: no cover
|
|
print(f"MoviePy: can't find or access ffmpeg in '{FFMPEG_BINARY}'.")
|
|
|
|
if try_cmd([IMAGEMAGICK_BINARY])[0]:
|
|
print(f"MoviePy: ImageMagick successfully found in '{IMAGEMAGICK_BINARY}'.")
|
|
else: # pragma: no cover
|
|
print(f"MoviePy: can't find or access ImageMagick in '{IMAGEMAGICK_BINARY}'.")
|
|
|
|
if DOTENV:
|
|
print(f"\n.env file content at {DOTENV}:\n")
|
|
print(Path(DOTENV).read_text())
|
|
|
|
|
|
if __name__ == "__main__": # pragma: no cover
|
|
check()
|