Compare commits
2 Commits
5a9f765281
...
6752425682
Author | SHA1 | Date | |
---|---|---|---|
6752425682 | |||
2bb1b1d778 |
@ -4,9 +4,39 @@ import encode_single_video
|
|||||||
|
|
||||||
import os
|
import os
|
||||||
|
|
||||||
|
import argparse
|
||||||
|
|
||||||
|
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
|
# tuple of video file extensions
|
||||||
videofileextensions = (".webm", ".mp4")
|
videofileextensions = (".webm", ".mp4")
|
||||||
|
|
||||||
|
# search all video files in "source"
|
||||||
videofiles: list[str] = []
|
videofiles: list[str] = []
|
||||||
for root, dirs, files in os.walk("source"):
|
for root, dirs, files in os.walk("source"):
|
||||||
for file in files:
|
for file in files:
|
||||||
@ -14,5 +44,11 @@ for root, dirs, files in os.walk("source"):
|
|||||||
filepath = os.path.join(root, file)
|
filepath = os.path.join(root, file)
|
||||||
videofiles.append(filepath)
|
videofiles.append(filepath)
|
||||||
|
|
||||||
|
try:
|
||||||
|
# encode all video files
|
||||||
for video in videofiles:
|
for video in videofiles:
|
||||||
encode_single_video.main(inputfile=video)
|
encode_single_video.main(
|
||||||
|
inputfile=video, outputpath=outputpath, keepencodes=keepencodes
|
||||||
|
)
|
||||||
|
except KeyboardInterrupt:
|
||||||
|
encode_single_video.cleanup(keepencodes=keepencodes)
|
||||||
|
@ -417,21 +417,12 @@ if __name__ == "__main__":
|
|||||||
help="Input file to encode",
|
help="Input file to encode",
|
||||||
)
|
)
|
||||||
|
|
||||||
# in case you wanted to rerun the conversion for everything
|
|
||||||
parser.add_argument(
|
|
||||||
"-r",
|
|
||||||
"--reset",
|
|
||||||
required=False,
|
|
||||||
action="store_true",
|
|
||||||
help="Rerun conversion for all files",
|
|
||||||
)
|
|
||||||
|
|
||||||
parser.add_argument(
|
parser.add_argument(
|
||||||
"-k",
|
"-k",
|
||||||
"--keep-encodes",
|
"--keep-encodes",
|
||||||
action="store_true",
|
action="store_true",
|
||||||
required=False,
|
required=False,
|
||||||
help="Don't delete encodes after getting their scores. Default is delete (False)",
|
help="Don't delete encodes after getting their scores. Default is False (delete)",
|
||||||
)
|
)
|
||||||
|
|
||||||
args = parser.parse_args()
|
args = parser.parse_args()
|
||||||
|
Loading…
Reference in New Issue
Block a user