Refactor & Rewrite

utils/mediainfo.py

@@ -1,30 +1,23 @@
 from pymediainfo import MediaInfo
 
-def get_aspect_ratio(inputFile):
-    media_info = MediaInfo.parse(inputFile)
+def get_aspect_ratio(video_input: str) -> str:
+    media_info = MediaInfo.parse(video_input)
+    video_track = media_info.video_tracks[0]
+    aspect_ratio = video_track.other_display_aspect_ratio
+    print('Detected Aspect Ratio : ' + aspect_ratio[0])
+    return aspect_ratio[0]
 
-    for track in media_info.tracks:
-        if track.track_type == "Video":
-            aspect_ratio = track.other_display_aspect_ratio
-            print('Detected Aspect Ratio : ' + aspect_ratio[0])
-            return aspect_ratio[0]
+def get_framerate(video_input: str) -> str:
+    media_info = MediaInfo.parse(video_input)
+    video_track = media_info.video_tracks[0]
+    frame_rate = video_track.frame_rate
 
-    # Fallback value
-    print('Falling Back To Aspect Ratio: 16:9')
-    return '16:9'
-
-def get_framerate(inputFile):
-    media_info = MediaInfo.parse(inputFile)
-
-    for track in media_info.tracks:
-        if track.track_type == "Video":
-            frame_rate = track.frame_rate
-            if frame_rate == '29.970':
-                print(f"Detected Framerate    : 30000/1001")
-                return '30000/1001'
-            if frame_rate == '24.000':
-                print(f"Detected Framerate    : 24000/1000")
-                return '24000/1000'
+    if frame_rate == '29.970':
+        print(f"Detected Framerate    : 30000/1001")
+        return '30000/1001'
+    if frame_rate == '24.000':
+        print(f"Detected Framerate    : 24")
+        return '24'
 
     print(f"Detected Framerate    : 24000/1001")
     return '24000/1001'