wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-05-10 01:02:56 +08:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity

Create custom input directory when configured

Browse Source
This commit is contained in:
huyua9 2026-05-04 18:26:04 +08:00
parent f3ea976cba
commit 7b7becc34c
2 changed files with 18 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
folder_paths.py
View File

@ -118,6 +118,11 @@ def set_temp_directory(temp_dir: str) -> None:
def set_input_directory(input_dir: str) -> None:
global input_directory
input_directory = input_dir
if not os.path.exists(input_directory):
try:
os.makedirs(input_directory)
except:
logging.error("Failed to create input directory")
def get_output_directory() -> str:
global output_directory

13
tests/test_folder_paths.py Normal file
View File

@ -0,0 +1,13 @@
import folder_paths
def test_set_input_directory_creates_missing_directory(tmp_path):
original_input_directory = folder_paths.get_input_directory()
custom_input_directory = tmp_path / "custom-input"
try:
folder_paths.set_input_directory(str(custom_input_directory))
assert custom_input_directory.is_dir()
finally:
folder_paths.set_input_directory(original_input_directory)