fix(core): temp file written to cwd

Now uses temp file module to prevent race conditions.
This commit is contained in:
2026-04-27 23:13:23 +01:00
parent e32f214a03
commit d97f15233b

View File

@@ -2,6 +2,7 @@
import os import os
import datetime import datetime
import subprocess import subprocess
import tempfile
from rewind.state import load_state, add_marker_to_state, remove_marker_from_state from rewind.state import load_state, add_marker_to_state, remove_marker_from_state
from rewind.config import Config from rewind.config import Config
@@ -120,61 +121,62 @@ def _get_ts_files(start_timestamp: float, end_timestamp: float) -> tuple[list[st
return selected_files, start_offset, end_offset return selected_files, start_offset, end_offset
def _concat_ts_files(file_list: list[str], start_offset: float, end_offset: float, length: float, output_file: str) -> None: def _concat_ts_files(file_list: list[str], start_offset: float, end_offset: float, length: float, output_file: str) -> None:
with open("file_list.txt", "w") as f: tmp_file = tempfile.NamedTemporaryFile(mode="w", suffix=".txt", delete=False)
try:
for file_path in file_list: for file_path in file_list:
f.write(f"file '{file_path}'\n") tmp_file.write(f"file '{file_path}'\n")
tmp_file.close()
cmd = ["ffmpeg", "-y", "-hide_banner", "-loglevel", "error", "-nostats", "-progress", "pipe:1"] cmd = ["ffmpeg", "-y", "-hide_banner", "-loglevel", "error", "-nostats", "-progress", "pipe:1"]
if start_offset > 0: if start_offset > 0:
cmd += ["-ss", str(start_offset)] cmd += ["-ss", str(start_offset)]
if end_offset > 0: if end_offset > 0:
cmd += ["-t", str(length)] cmd += ["-t", str(length)]
cmd += ["-f", "concat", "-safe", "0", "-i", "file_list.txt", "-c", "copy"] cmd += ["-f", "concat", "-safe", "0", "-i", tmp_file.name, "-c", "copy"]
cmd.append(output_file) cmd.append(output_file)
process = subprocess.Popen( process = subprocess.Popen(
cmd, cmd,
stdout=subprocess.PIPE, stdout=subprocess.PIPE,
stderr=subprocess.PIPE, stderr=subprocess.PIPE,
text=True, text=True,
bufsize=1, bufsize=1,
) )
init_ms_val = 0 init_ms_val = 0
init_ms_val_set = False init_ms_val_set = False
with tqdm( with tqdm(
total=length, total=length,
unit="s", unit="s",
unit_scale=True, unit_scale=True,
unit_divisor=60, unit_divisor=60,
desc="Processing", desc="Processing",
leave=True, leave=True,
) as pbar: ) as pbar:
for line in process.stdout: for line in process.stdout: # type: ignore[union-attr]
line = line.strip() line = line.strip()
if line.startswith("out_time_ms="): if line.startswith("out_time_ms="):
out_time_ms = int(line.split("=")[1]) out_time_ms = int(line.split("=")[1])
if not init_ms_val_set: if not init_ms_val_set:
init_ms_val = out_time_ms init_ms_val = out_time_ms
init_ms_val_set = True init_ms_val_set = True
out_time_ms -= init_ms_val out_time_ms -= init_ms_val
seconds = abs(out_time_ms / 1_000_000) seconds = abs(out_time_ms / 1_000_000)
pbar.n = min(seconds, length) pbar.n = min(seconds, length)
pbar.refresh() pbar.refresh()
elif line == "progress=end": elif line == "progress=end":
break break
ret = process.wait() ret = process.wait()
os.remove("file_list.txt") if ret != 0:
raise RuntimeError("ffmpeg failed")
if ret != 0: finally:
raise RuntimeError("ffmpeg failed") os.unlink(tmp_file.name)
def get_duration(file_path: str) -> float: def get_duration(file_path: str) -> float:
result = subprocess.run( result = subprocess.run(