From 68ab303cd840bd98cfd7697035f4504b137fe0fa Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?F=C3=A9lix=20Voituret?= <fvoituret@deezer.com>
Date: Wed, 10 Jun 2020 16:40:55 +0200
Subject: [PATCH] fix: add ffmpeg binaries check

---
 spleeter/audio/ffmpeg.py | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/spleeter/audio/ffmpeg.py b/spleeter/audio/ffmpeg.py
index 08218e9..2277d87 100644
--- a/spleeter/audio/ffmpeg.py
+++ b/spleeter/audio/ffmpeg.py
@@ -9,6 +9,7 @@
 """
 
 import os
+import shutil
 
 # pylint: disable=import-error
 import ffmpeg
@@ -24,6 +25,16 @@ __author__ = 'Deezer Research'
 __license__ = 'MIT License'
 
 
+def _check_ffmpeg_install():
+    """ Ensure FFMPEG binaries are available.
+
+    :raise SpleeterError: If ffmpeg or ffprobe is not found.
+    """
+    for binary in ('ffmpeg', 'ffprobe'):
+        if shutil.which(binary) is None:
+            raise SpleeterError('{} binary not found'.format(binary))
+
+
 def _to_ffmpeg_time(n):
     """ Format number of seconds to time expected by FFMPEG.
     :param n: Time in seconds to format.
@@ -66,6 +77,7 @@ class FFMPEGProcessAudioAdapter(AudioAdapter):
         :returns: Loaded data a (waveform, sample_rate) tuple.
         :raise SpleeterError: If any error occurs while loading audio.
         """
+        _check_ffmpeg_install()
         if not isinstance(path, str):
             path = path.decode()
         try:
@@ -112,6 +124,7 @@ class FFMPEGProcessAudioAdapter(AudioAdapter):
         :param bitrate: (Optional) Bitrate of the written audio file.
         :raise IOError: If any error occurs while using FFMPEG to write data.
         """
+        _check_ffmpeg_install()
         directory = os.path.dirname(path)
         if not os.path.exists(directory):
             raise SpleeterError(f'output directory does not exists: {directory}')