FFMPEGWrapper: check MIME type

lens-biophotonics · Mar 3, 2021 · 863a39b · 863a39b
1 parent 52e5922
commit 863a39b
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/zetastitcher/io/ffmpeg_wrapper.py b/zetastitcher/io/ffmpeg_wrapper.py
@@ -1,4 +1,5 @@
 import json
+import mimetypes
 import subprocess as sp
 from pathlib import Path
 
@@ -32,6 +33,10 @@ def open(self, file_path=None):
         if file_path is not None:
             self.file_path = Path(file_path)
 
+        mtype = mimetypes.guess_type(self.file_path)[0]
+        if mtype is None or not mtype.startswith('video'):
+            raise ValueError(f'Invalid MIME type:{mtype}')
+
         cmd = [
             'ffprobe',
             '-v', 'quiet',