unMcFucking

sub-and-audio_drop_to_file.py

@@ -0,0 +1,68 @@
+import re
+import subprocess
+import sys
+from pathlib import Path
+import os
+output_folder_name = "/out_sub_audio/"
+regex_subtitle = re.compile("Stream #0:\d+(\(\w*\))?: Subtitle:.*")
+regex_audio = re.compile("Stream #0:\d(\(\w*\))?: Audio:.*")
+nr_files = 0
+files_processed = []
+droppedFile = sys.argv[1]
+droppedName = Path(droppedFile).name
+
+
+#get path of script location and remove the tail
+head_tail = os.path.split(sys.argv[0])
+output_path = head_tail[0]
+del sys.argv[0]
+
+#check if output folder exists, otherwise create it
+isExist = os.path.exists(output_path+output_folder_name)
+if not isExist:
+    os.makedirs(output_path+output_folder_name)
+
+#droppedFile = "test.mkv"
+
+
+
+
+
+for i, file in enumerate(sys.argv):
+    arg_list = 'ffprobe.exe -i "{file}"'.format(file=file)
+
+    cmd = subprocess.Popen(arg_list, stdout=subprocess.PIPE, stderr=subprocess.PIPE, encoding='UTF-8')
+        
+    subprocess_err = cmd.communicate()[1]
+    
+    print("file", i+1,":", Path(file).name)
+
+    for r, match in enumerate(regex_audio.finditer(subprocess_err)):
+        print("Audio Track", r,": ", match.group(0))
+    for r, match in enumerate(regex_subtitle.finditer(subprocess_err)):
+        print("Subtitle Track", r,": ", match.group(0))
+    print('\n')
+
+
+x = (int(x) for x in input ("Enter audio tracks to keep: ").split())
+audio_track = "".join(" -map 0:a:{}".format(y) for y in x)
+
+
+z = (int(z) for z in input ("Enter subtitle tracks to keep: ").split())
+subtitle_track = "".join(" -map 0:s:{}".format(y) for y in z)
+
+for file in sys.argv:
+ 
+    basename = os.path.basename(file)
+
+    arg_list = 'ffmpeg.exe -i "{file}" -map 0:v {audio_track} {subtitle_track} -c copy "{output_path}{output_folder_name}{basename}"'.format(file=file, audio_track=audio_track, subtitle_track=subtitle_track, output_path=output_path, output_folder_name=output_folder_name, basename=basename)
+    print(arg_list)
+    with open ('input.txt', 'a') as of:
+        of.write(arg_list + "\n")
+    nr_files = nr_files+1
+    files_processed.append(file)
+
+
+print("Number of files processed: ", nr_files)
+print(*files_processed, sep='\n')
+input("press enter to exit")