fix(setup): Include ffprobe

yshalsager · Aug 19, 2022 · cb547bf · cb547bf
1 parent 53f682d
commit cb547bf
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -7,11 +7,13 @@
 __version__ = "1.0.0"
 base = None
 include_files: List[str] = []
-
+ffmpeg_files: List[str] = ["ffmpeg.exe", "ffprobe.exe"]
+
 if sys.platform == "win32":
     base = "Win32GUI"
-    if (Path(__file__).parent / "ffmpeg.exe").exists():
-        include_files.append("ffmpeg.exe")
+    for file in ffmpeg_files:
+        if (Path(__file__).parent / file).exists():
+            include_files.append(file)
 
 includes = ["tkinter"]
 excludes = ["matplotlib", "sqlite3"]