Added tests and more

chapter_OP_ED_detection.py

@@ -0,0 +1,62 @@
+from pathlib import Path
+from typing import Tuple, List, Optional
+import subprocess
+import json
+
+def run(cmd):
+    """
+    Run a subprocess command.
+
+    Args:
+        cmd: Command and arguments as a list.
+
+    Raises:
+        subprocess.CalledProcessError: If the command exits with a non-zero status.
+    """
+    try:
+        result = subprocess.run(
+            cmd,
+            check=True,
+            stdout=subprocess.PIPE,
+            stderr=subprocess.STDOUT,
+            text=True
+        )
+        return result
+    except subprocess.CalledProcessError as e:
+        print(e.stdout)
+        raise
+
+
+def _get_chapters(file: Path) -> List[Tuple[float, float]]:
+    """Return list of (start_time, duration) for each chapter."""
+    cmd = [
+        "ffprobe",
+        "-i", str(file),
+        "-print_format", "json",
+        "-show_chapters",
+        "-loglevel", "error"
+    ]
+
+    result = run(cmd)
+    data = json.loads(result.stdout)
+
+    chapters = []
+    for ch in data.get("chapters", []):
+        start = float(ch["start_time"])
+        end = float(ch["end_time"])
+        duration = end - start
+        chapters.append((start, duration))
+
+    return chapters
+
+if __name__ == '__main__':
+    file = "/home/honney/projects/mkv/output_dir/08.mkv"
+    chapters = _get_chapters(Path(file))
+    time = 90
+    min_len = time / 2
+    max_len = time * 2
+    filtered = [
+        (start, dur) for start, dur in chapters
+        if min_len <= dur <= max_len
+    ]
+    print(filtered)