From d29bf1ab9ee83377d32087c9839ad661dff6c537 Mon Sep 17 00:00:00 2001 From: Fabrice Quenneville Date: Sun, 1 Nov 2020 04:11:03 -0500 Subject: [PATCH] Added more filetypes and codec filter for mpeg4 --- converter.py | 21 +++++++++++++++++++-- 1 file changed, 19 insertions(+), 2 deletions(-) diff --git a/converter.py b/converter.py index 7e0c3d0..9f31bdb 100755 --- a/converter.py +++ b/converter.py @@ -98,6 +98,12 @@ def get_mkv_list(path): def get_mp4_list(path): return list(path.rglob("*.[mM][pP]4")) +def get_m4v_list(path): + return list(path.rglob("*.[mM]4[vV]")) + +def get_flv_list(path): + return list(path.rglob("*.[fF][lL][vV]")) + def get_videolist(parentdir): videolist = [] path = Path(parentdir) @@ -109,8 +115,10 @@ def get_videolist(parentdir): videolist += get_mkv_list(path) if "-all_mp4" in sys.argv or "-any" in sys.argv: videolist += get_mp4_list(path) - if "-all_mp4" in sys.argv or "-any" in sys.argv: - videolist += get_mp4_list(path) + if "-all_m4v" in sys.argv or "-any" in sys.argv: + videolist += get_m4v_list(path) + if "-all_flv" in sys.argv or "-any" in sys.argv: + videolist += get_flv_list(path) videolist_tmp = videolist @@ -118,12 +126,21 @@ def get_videolist(parentdir): if "-old" in sys.argv: videolist = get_old(videolist) + + if "-mpeg4" in sys.argv: + videolist = get_mpeg4(videolist) return videolist def get_old(videolist_temp): return [video for video in videolist_temp if get_codec(video) not in ["hevc", "av1"]] +def get_mpeg4(videolist_temp): + return [video for video in videolist_temp if get_codec(video) in ["mpeg4", "msmpeg4v3"]] + +def get_mpeg4(videolist_temp): + return [video for video in videolist_temp if get_codec(video) in ["mpeg4"]] + def get_codec(filename): try: args = ["/usr/bin/ffprobe", "-v", "error", "-select_streams", "v:0", "-show_entries", "stream=codec_name", "-of", "default=noprint_wrappers=1:nokey=1", str(filename)]