V 1.1 - tqdm progress

main.py

@@ -33,8 +33,34 @@ def mux_audio_into_output(Quality_Video, audio_file, output_dir):
     return final_output
 
 def process_one(German_Video: Path, Quality_Video: Path, output_dir: Path, offset: Optional[int] = None) -> bool:
+    steps = [
+        "Probe video",
+        "Compute offset",
+        "Extract pre beginning",
+        "Extract german audio",
+        "Extract post end",
+        "Combine audio",
+        "Mux",
+        "Cleanup",
+        "Final check",
+    ]
 
-    print("probe_video")
+    weights = [
+        1,
+        3,
+        1,
+        2,
+        1,
+        2,
+        2,
+        1,
+        2,
+    ]
+
+    with tqdm(total=len(steps), desc=Quality_Video.name, leave=False) as pbar:
+        for step, w in zip(steps, weights):
+            pbar.set_description(step)
+            if step == "Probe video":
                 # --- Probe video ---
                 fps, duration, _, vstart = probe_video(German_Video)
                 fps2, duration2, _, vstart2 = probe_video(Quality_Video)
@@ -44,7 +70,7 @@ def process_one(German_Video: Path, Quality_Video: Path, output_dir: Path, offse
                 if duration2 <= 0:
                     raise ValueError("Duration of Quality_Video must be greater than 0")
 
-    print("Compute offset")
+            if step == "Compute offset":
                 # --- Offset ---
                 if offset == None:
                     offset0 = corelate_videos(Quality_Video, German_Video, fps2)
@@ -52,9 +78,9 @@ def process_one(German_Video: Path, Quality_Video: Path, output_dir: Path, offse
                     offset0 = offset
                 if offset0 > duration or offset0 > duration2:
                     raise ValueError("Offset needs to be smaller than the video durations")
-    print(f"Offset: {offset0}s")
+                pbar.set_postfix_str(f"Offset: {offset0}s")
 
-    print("Extract pre beginning")
+            if step == "Extract pre beginning":
                 german_audio_file = output_dir / f"{Path(Quality_Video).stem}_german_audio.mka"
                 quality_audio_file1 = output_dir / f"{Path(Quality_Video).stem}_original_audio1.mka"
                 quality_audio_file2 = output_dir / f"{Path(Quality_Video).stem}_original_audio2.mka"
@@ -80,7 +106,7 @@ def process_one(German_Video: Path, Quality_Video: Path, output_dir: Path, offse
                             f"while the audio has a length of {get_audio_duration(Quality_Video)}"
                         )
 
-    print("Extract german audio")
+            if step == "Extract german audio":
 
                 ### Get the German Audio
                 if offset0 < 0:
@@ -99,7 +125,7 @@ def process_one(German_Video: Path, Quality_Video: Path, output_dir: Path, offse
                         f"while the audio has a length of {get_audio_duration(German_Video)}"
                     )
 
-    print("Extract post end")
+            if step == "Extract post end":
 
                 ### Get the end from the first original audio if it is longer than the german part
                 start2 = vstart2 + end
@@ -117,7 +143,7 @@ def process_one(German_Video: Path, Quality_Video: Path, output_dir: Path, offse
                             f"while the audio has a length of {get_audio_duration(Quality_Video)}"
                         )
 
-    print("Combine audio")
+            if step == "Combine audio":
 
                 # Combine the audio files
                 if os.path.isfile(quality_audio_file1):
@@ -125,19 +151,19 @@ def process_one(German_Video: Path, Quality_Video: Path, output_dir: Path, offse
                 if os.path.isfile(quality_audio_file2):
                     stick_together_audio(german_audio_file, quality_audio_file2, german_audio_file)
 
-    print("Mux")
+            if step == "Mux":
 
                 # --- Mux trimmed audio ---
                 final_file = mux_audio_into_output(Quality_Video, german_audio_file, output_dir)
 
-    print("Cleanup")
+            if step == "Cleanup":
 
                 if os.path.isfile(quality_audio_file1):
                     os.remove(quality_audio_file1)
                 if os.path.isfile(quality_audio_file2):
                     os.remove(quality_audio_file2)
 
-    print("Final check")
+            if step == "Final check":
                 try:
                     files_acceptable, op_ed_offset, stream_num = check_intro_outro_sync(final_file)
                 except:
@@ -148,20 +174,22 @@ def process_one(German_Video: Path, Quality_Video: Path, output_dir: Path, offse
                     return False
 
                 if files_acceptable and isinstance(op_ed_offset, float) and isinstance(stream_num, int):
-        print(f"✔ {final_file.name} | offset={op_ed_offset:.4f}")
+                    tqdm.write(f"✔ {final_file.name} | offset={op_ed_offset:.4f}")
                     if np.abs(op_ed_offset) > (1/fps2) and os.path.isfile(german_audio_file):
                         replace_with_shifted_mka(final_file, german_audio_file, stream_num, op_ed_offset)
                     if os.path.isfile(german_audio_file):
                         os.remove(path=german_audio_file)
                     return True
                 else:
-        print(f"✖ FAILED: {final_file.name}")
+                    tqdm.write(f"✖ FAILED: {final_file.name}")
                     if os.path.isfile(german_audio_file):
                         os.remove(path=german_audio_file)
                     new_name = final_file.with_name(f"[FAILED]_{final_file.name}")
                     final_file.rename(new_name)
                     return False
 
+            pbar.update(w)
+
 
 def main():
     output_dir = "./output_dir"
@@ -171,7 +199,7 @@ def main():
 
     episodes = range(start_num, episode_num + 1)
 
-    for i in episodes:
+    for i in tqdm(episodes, desc="Episodes"):
         german_pattern = f"Folder0/S01E{i:02}*.mkv"
         if os.path.isfile(german_pattern):
             German_Video = [german_pattern]
@@ -192,9 +220,9 @@ def main():
         if not process_one(Path(German_Video[0]), Path(Quality_Video[0]), Path(output_dir)):
             fault_catch.append([Path(German_Video[0]), Path(Quality_Video[0])])
     if fault_catch:
-        print("\nFailed Videos:")
+        tqdm.write("\nFailed Videos:")
         for ger, orig in fault_catch:
-            print(f"{ger} + {orig}")
+            tqdm.write(f"{ger} + {orig}")
 
 if __name__ == '__main__':
     if len(sys.argv) == 3: