FfmpegMediaTypeError (#37)

livingbio · Oct 17, 2023 · 6fef444 · 6fef444
1 parent 64ea8c1
commit 6fef444
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/src/ffmpeg_media_type/exceptions.py b/src/ffmpeg_media_type/exceptions.py
@@ -0,0 +1,2 @@
+class FfmpegMediaTypeError(Exception):
+    pass
diff --git a/src/ffmpeg_media_type/utils/shell.py b/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()