Add script to encode all videos in "source" subfolder

encode_all.py

@@ -4,15 +4,51 @@ import encode_single_video
 
 import os
 
-# tuple of video file extensions
-videofileextensions = (".webm", ".mp4")
+import argparse
 
-videofiles: list[str] = []
-for root, dirs, files in os.walk("source"):
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser(description="")
+
+    parser.add_argument(
+        "-o",
+        "--output-dir",
+        required=False,
+        type=str,
+        help='Output directory for encodes. Default is "encodes" in current working directory',
+        default="encodes",
+    )
+
+    parser.add_argument(
+        "-k",
+        "--keep-encodes",
+        action="store_true",
+        required=False,
+        help="Don't delete encodes after getting their scores. Default is False (delete)",
+    )
+
+    args = parser.parse_args()
+
+    # REVIEW expose this as variable/argument?
+    outputpath: str = args.output_dir
+
+    keepencodes: bool = args.keep_encodes
+
+    # tuple of video file extensions
+    videofileextensions = (".webm", ".mp4")
+
+    # search all video files in "source"
+    videofiles: list[str] = []
+    for root, dirs, files in os.walk("source"):
         for file in files:
             if file.endswith(videofileextensions):
                 filepath = os.path.join(root, file)
                 videofiles.append(filepath)
 
-for video in videofiles:
-    encode_single_video.main(inputfile=video)
+    try:
+        # encode all video files
+        for video in videofiles:
+            encode_single_video.main(
+                inputfile=video, outputpath=outputpath, keepencodes=keepencodes
+            )
+    except KeyboardInterrupt:
+        encode_single_video.cleanup(keepencodes=keepencodes)