FfmpegMediaTypeError (#37)

Author: ianisnotlan

src/ffmpeg_media_type/exceptions.py

@@ -0,0 +1,2 @@
+class FfmpegMediaTypeError(Exception):
+    pass

src/ffmpeg_media_type/utils/shell.py

@@ -2,14 +2,17 @@
 import shlex
 import tempfile
 
+from ..exceptions import FfmpegMediaTypeError
+
 
 def call(cmds: list[str]) -> str:
     # call command and return stdout
 
     command = " ".join(shlex.quote(part) for part in cmds)
     with tempfile.TemporaryDirectory() as tmpdir:
         output_path = f"{tmpdir}/output.txt"
-        assert os.system(f"{command} > {output_path}") == 0, f"command failed: {command}"
+        if os.system(f"{command} > {output_path}") != 0:
+            raise FfmpegMediaTypeError(f"command failed: {command}")
 
         with open(output_path) as ifile:
             return ifile.read()