Claude
3adebb731f
Old project/workflow names and browser settings migrate on load; /funpack redirects to /fun. Co-Authored-By: Claude Sonnet 5.5 <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_016acRss69eH1235eRdFTZHY
49 lines
2.0 KiB
Python
49 lines
2.0 KiB
Python
"""Real ffmpeg, a fake ComfyUI folder pair, and clips made on the spot."""
|
|
|
|
import shutil
|
|
import subprocess
|
|
|
|
import pytest
|
|
|
|
from core import config
|
|
|
|
needs_ffmpeg = pytest.mark.skipif(shutil.which("ffmpeg") is None, reason="ffmpeg is not installed")
|
|
|
|
|
|
@pytest.fixture
|
|
def comfy(tmp_path, monkeypatch):
|
|
"""ComfyUI's output and temp folders, plus FunUI's own stores, all under tmp_path."""
|
|
from modules.output.render import files
|
|
out, temp = tmp_path / "output", tmp_path / "temp"
|
|
out.mkdir(), temp.mkdir()
|
|
monkeypatch.setattr(files, "comfy_dir", lambda kind: str(out if kind == "output" else temp))
|
|
monkeypatch.setattr(config, "PROJECTS_DIR", tmp_path / "projects")
|
|
monkeypatch.setattr(config, "MEDIA_DIR", tmp_path / "media")
|
|
return type("Comfy", (), {"output": out, "temp": temp, "root": tmp_path})
|
|
|
|
|
|
def make_clip(path, seconds=2.0, size="160x120", rate=25, audio=True, faststart=True):
|
|
"""A real mp4: a test pattern, with a sine tone unless `audio` is False."""
|
|
cmd = ["ffmpeg", "-y", "-f", "lavfi", "-i", f"testsrc=duration={seconds}:size={size}:rate={rate}"]
|
|
if audio:
|
|
cmd += ["-f", "lavfi", "-i", f"sine=frequency=440:duration={seconds}"]
|
|
cmd += ["-c:v", "libx264", "-pix_fmt", "yuv420p"]
|
|
if audio:
|
|
cmd += ["-c:a", "aac"]
|
|
if faststart:
|
|
cmd += ["-movflags", "+faststart"]
|
|
cmd += [str(path)]
|
|
subprocess.run(cmd, check=True, capture_output=True)
|
|
return path
|
|
|
|
|
|
def probe(path):
|
|
"""(duration seconds, width, height, has audio) of a real file."""
|
|
import json
|
|
out = subprocess.run(["ffprobe", "-v", "error", "-show_entries", "stream=codec_type,width,height:format=duration",
|
|
"-of", "json", str(path)], check=True, capture_output=True, text=True).stdout
|
|
data = json.loads(out)
|
|
video = next(s for s in data["streams"] if s["codec_type"] == "video")
|
|
return (float(data["format"]["duration"]), video["width"], video["height"],
|
|
any(s["codec_type"] == "audio" for s in data["streams"]))
|