Claude
2921ff517b
Settings > Saved & Temp Files now has two tabs. Saved lists what ComfyUI wrote to its output folder; Temp is the old list. Click a file to look at it, open it in a tab, or delete it after a question. Delete only removes media files inside the chosen folder. Co-Authored-By: Claude Sonnet 5.5 <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_015xHQk4XeXzJpBR3HSwnoBZ
28 lines
1.5 KiB
Python
28 lines
1.5 KiB
Python
"""The gallery lists what ComfyUI saved, and deletes only media inside that folder."""
|
|
|
|
from core import temp_files
|
|
|
|
|
|
def test_saved_files_are_listed_from_output_not_temp(monkeypatch, tmp_path):
|
|
(tmp_path / "FunPack_00001_.mp4").write_bytes(b"x")
|
|
monkeypatch.setattr(temp_files, "output_dir", lambda: str(tmp_path))
|
|
monkeypatch.setattr(temp_files, "temp_dir", lambda: None)
|
|
assert [f["filename"] for f in temp_files.listing(where="output")["files"]] == ["FunPack_00001_.mp4"]
|
|
assert temp_files.listing()["files"] == []
|
|
|
|
|
|
def test_delete_removes_a_media_file_and_nothing_else(monkeypatch, tmp_path):
|
|
out = tmp_path / "out"
|
|
(out / "sub").mkdir(parents=True)
|
|
pic, weights, outside = out / "sub" / "a.png", out / "model.safetensors", tmp_path / "secret.png"
|
|
for f in (pic, weights, outside):
|
|
f.write_bytes(b"x")
|
|
monkeypatch.setattr(temp_files, "output_dir", lambda: str(out))
|
|
|
|
assert temp_files.delete("a.png", "sub", "output") == {"deleted": True} and not pic.exists()
|
|
assert not temp_files.delete("model.safetensors", "", "output")["deleted"] and weights.exists()
|
|
assert not temp_files.delete("secret.png", "..", "output")["deleted"] and outside.exists(), "left the folder"
|
|
assert not temp_files.delete("../secret.png", "", "output")["deleted"] and outside.exists()
|
|
assert not temp_files.delete("a.png", "sub", "output")["deleted"], "a second delete should say it is gone"
|
|
assert not temp_files.delete("a.png", "", "input")["deleted"]
|