initial commit

ffmpeg_resolution.py

@@ -0,0 +1,44 @@
+#!/usr/bin/env python3.9
+
+from pathlib import Path
+import subprocess
+import sys
+from ffmpeg import ffprobe
+import logging
+
+
+def get_resolution(video_path: Path) -> str:
+    try:
+        info = ffprobe(video_path)
+    except subprocess.CalledProcessError:
+        return None
+
+    if 700 <= info.height <= 800:
+        return "720p"
+    elif 1000 <= info.height <= 1200:
+        return "1080p"
+    elif info.height > 2000:
+        return "4K"
+    return None
+
+
+def main():
+    for it in sys.argv[1:]:
+        video_path = Path(it)
+        if not video_path.is_file():
+            continue
+        resolution = get_resolution(video_path)
+        if resolution is None:
+            logging.info("unknown resolution: %s", video_path)
+            continue
+        if resolution in video_path.stem:
+            continue
+
+        logging.info("%s: %s", video_path.stem, resolution)
+        new_path = video_path.with_stem(f"{video_path.stem} [{resolution}]")
+        video_path.rename(new_path)
+
+
+if __name__ == "__main__":
+    logging.basicConfig(level=logging.INFO)
+    main()