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

Merge 7b7becc34c into 7bbf1e8169

Browse Source
This commit is contained in:
huyua9 2026-05-09 14:33:02 +03:00 committed by GitHub
commit df1c6c083e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
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

@ -122,6 +122,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)