honney/movie_sync
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: refactor: extract.py to accept multiple input files and improve output directory handling

Browse Source
This commit is contained in:
Hannes committed 2026-09-13 21:23:15 +02:00
1 parent db4531dd24
commit 2bfe696f2d
1 file changed
+35 -28
Download Patch File Download Diff File Expand all files Collapse all files
extract.py
+35 -28
View File
@@ -13,30 +13,37 @@ import sys
import shutil
# ─── Hardcoded input (relative to this script's directory) ───
# INPUT_FILE = os.path.join(os.path.dirname(os.path.abspath(__file__)), "/path/to/movie_b.mp4")
INPUT_FILE = os.path.join(os.path.dirname(os.path.abspath(__file__)), "/path/to/movie_a.mkv")
inputs = [
"/path/to/movie_b.mp4",
"/path/to/movie_a.mkv"
]
INPUT_FILES = []
for input_file in inputs:
INPUT_FILES.append(os.path.join(os.path.dirname(os.path.abspath(__file__)), input_file))
DURATION = 90 # 1.5 minutes in seconds
OUT_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), "extracted")
# ─── Check ffmpeg ───
if shutil.which("ffmpeg") is None:
sys.exit("ERROR: 'ffmpeg' not found on PATH. Install it and try again.")
if shutil.which(f"ffmpeg") is None:
sys.exit(f"ERROR: 'ffmpeg' not found on PATH. Install it and try again.")
# ─── Prepare output folder ───
os.makedirs(OUT_DIR, exist_ok=True)
# ─── Derive output filenames ───
stem = os.path.splitext(os.path.basename(INPUT_FILE))[0]
video_out = os.path.join(OUT_DIR, f"{stem}_video.mkv")
audio_out = os.path.join(OUT_DIR, f"{stem}_audio.mp3")
for INPUT_FILE in INPUT_FILES:
print(f"Input : {INPUT_FILE}")
print(f"Video → {video_out}")
print(f"Audio → {audio_out}")
print("-" * 50)
# ─── Derive output filenames ───
stem = os.path.splitext(os.path.basename(INPUT_FILE))[0]
video_out = os.path.join(OUT_DIR, f"{stem}_video.mkv")
audio_out = os.path.join(OUT_DIR, f"{stem}_audio.mp3")
# ─── Extract video-only (first 90s, no audio) ───
cmd_video = [
print(f"Input : {INPUT_FILE}")
print(f"Video → {video_out}")
print(f"Audio → {audio_out}")
print(f"-" * 50)
# ─── Extract video-only (first 90s, no audio) ───
cmd_video = [
"ffmpeg",
"-y", # overwrite output
"-i", INPUT_FILE,
@@ -44,15 +51,15 @@ cmd_video = [
"-an", # no audio
"-c:v", "copy", # copy video stream (no re-encode = fast + lossless)
video_out
]
print("[video] running ffmpeg…")
r = subprocess.run(cmd_video, capture_output=True, text=True)
if r.returncode != 0:
]
print(f"[video: {stem}] running ffmpeg…")
r = subprocess.run(cmd_video, capture_output=True, text=True)
if r.returncode != 0:
sys.exit(f"ffmpeg (video) failed:\n{r.stderr}")
print("[video] done ✔")
print(f"[video: {stem}] done ✔")
# ─── Extract audio-only (first 90s, no video) ───
cmd_audio = [
# ─── Extract audio-only (first 90s, no video) ───
cmd_audio = [
"ffmpeg",
"-y",
"-i", INPUT_FILE,
@@ -61,12 +68,12 @@ cmd_audio = [
"-c:a", "libmp3lame",
"-b:a", "192k", # 192 kbps
audio_out
]
print("[audio] running ffmpeg…")
r = subprocess.run(cmd_audio, capture_output=True, text=True)
if r.returncode != 0:
]
print(f"[audio: {stem}] running ffmpeg…")
r = subprocess.run(cmd_audio, capture_output=True, text=True)
if r.returncode != 0:
sys.exit(f"ffmpeg (audio) failed:\n{r.stderr}")
print("[audio] done ✔")
print(f"[audio: {stem}] done ✔")
print("-" * 50)
print("All done. Files are in:", OUT_DIR)
print(f"-" * 50)
print(f"All done. Files are in:", OUT_DIR)