prefer video stream duration, fallback to container duration if unava…

…ilable (#64)
amietn · Jun 14, 2019 · 7ab5119 · 7ab5119
1 parent 1c5ae90
commit 7ab5119
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/vcsi/vcsi.py b/vcsi/vcsi.py
@@ -215,7 +215,13 @@ def compute_format(self):
         """
         format_dict = self.ffprobe_dict["format"]
 
-        self.duration_seconds = float(format_dict["duration"])
+        try:
+            # try getting video stream duration first
+            self.duration_seconds = float(self.video_stream["duration"])
+        except (KeyError, AttributeError):
+            # otherwise fallback to format duration
+            self.duration_seconds = float(format_dict["duration"])
+
         self.duration = MediaInfo.pretty_duration(self.duration_seconds)
 
         self.filename = os.path.basename(format_dict["filename"])