From ad12d64f1e349cc9692483d357c0ab72f9c0d4bd Mon Sep 17 00:00:00 2001
From: Travis Abendshien <lvnvtravis@gmail.com>
Date: Thu, 25 Jul 2024 11:54:44 -0700
Subject: [PATCH] (fix): catch ffmpeg errors in file tester

---
 tagstudio/src/qt/helpers/file_tester.py | 20 ++++++++++++--------
 1 file changed, 12 insertions(+), 8 deletions(-)

diff --git a/tagstudio/src/qt/helpers/file_tester.py b/tagstudio/src/qt/helpers/file_tester.py
index dd115f85..3fbea090 100644
--- a/tagstudio/src/qt/helpers/file_tester.py
+++ b/tagstudio/src/qt/helpers/file_tester.py
@@ -14,12 +14,16 @@ def is_readable_video(filepath: Path | str):
     Args:
         filepath (Path | str):
     """
-    probe = ffmpeg.probe(Path(filepath))
-    for stream in probe["streams"]:
-        if stream.get("codec_tag_string") in [
-            "drma",
-            "drms",
-            "drmi",
-        ]:
-            return False
+    try:
+        probe = ffmpeg.probe(Path(filepath))
+        for stream in probe["streams"]:
+            # DRM check
+            if stream.get("codec_tag_string") in [
+                "drma",
+                "drms",
+                "drmi",
+            ]:
+                return False
+    except ffmpeg.Error:
+        return False
     return True