Refactor src layout and add logging lifecycle + tests

2026-04-02 12:32:02 +02:00
parent e3c2b9ddee
commit a0a66264d2
13 changed files with 172 additions and 202 deletions
--- a/src/merge_audio_video.py
+++ b/src/merge_audio_video.py
@@ -0,0 +1,90 @@
+#!/usr/bin/env python3
+"""Merge videos/output_n.mp4 with audios/output_n.mp3 into merged/merged_n.mp4."""
+
+from __future__ import annotations
+
+import argparse
+import logging
+import re
+import subprocess
+from pathlib import Path
+
+from logging_config import configure_logging, debug_log_lifecycle
+
+
+SCRIPT_DIR = Path(__file__).resolve().parent
+DEFAULT_BASE_DIR = SCRIPT_DIR.parents[1]
+DEFAULT_VIDEOS_DIR = DEFAULT_BASE_DIR / "videos"
+DEFAULT_AUDIOS_DIR = DEFAULT_BASE_DIR / "audios"
+DEFAULT_OUTPUT_DIR = DEFAULT_BASE_DIR / "merged"
+
+LOGGER = logging.getLogger(__name__)
+
+
+def shot_number(path: Path) -> int:
+    match = re.search(r"output_(\d+)\.mp4$", path.name)
+    return int(match.group(1)) if match else -1
+
+
+def parse_args() -> argparse.Namespace:
+    parser = argparse.ArgumentParser(description=__doc__)
+    parser.add_argument("--videos-dir", type=Path, default=DEFAULT_VIDEOS_DIR)
+    parser.add_argument("--audios-dir", type=Path, default=DEFAULT_AUDIOS_DIR)
+    parser.add_argument("--output-dir", type=Path, default=DEFAULT_OUTPUT_DIR)
+    parser.add_argument(
+        "--log-level",
+        default=None,
+        help="Logging level (overrides LOG_LEVEL env var)",
+    )
+    return parser.parse_args()
+
+
+@debug_log_lifecycle
+def main() -> int:
+    args = parse_args()
+    configure_logging(args.log_level)
+    args.output_dir.mkdir(parents=True, exist_ok=True)
+
+    videos = sorted(args.videos_dir.glob("output_*.mp4"), key=shot_number)
+    if not videos:
+        LOGGER.warning("No videos found in %s", args.videos_dir)
+        return 1
+
+    for video in videos:
+        num = shot_number(video)
+        audio = args.audios_dir / f"output_{num}.mp3"
+        output = args.output_dir / f"merged_{num}.mp4"
+
+        if not audio.exists():
+            LOGGER.warning("No audio found for shot %s (%s); skipped", num, audio)
+            continue
+
+        if output.exists():
+            LOGGER.info("Already exists; skipped shot %s", num)
+            continue
+
+        LOGGER.info("Merging shot %s: %s + %s -> %s", num, video, audio, output)
+        subprocess.run(
+            [
+                "ffmpeg",
+                "-i",
+                str(video),
+                "-i",
+                str(audio),
+                "-c:v",
+                "copy",
+                "-c:a",
+                "aac",
+                "-shortest",
+                "-y",
+                str(output),
+            ],
+            check=True,
+        )
+        LOGGER.info("Done: %s", output)
+
+    return 0
+
+
+if __name__ == "__main__":
+    raise SystemExit(main())