wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-11 06:40:48 +08:00
Code Issues Actions 12 Packages Projects Releases Wiki Activity

Improve typing and file path handling

Browse Source
This commit is contained in:
doctorpangloss 2024-02-29 19:29:06 -08:00
parent e6623a1359
commit 44882eab0c
2 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
comfy/component_model/file_output_path.py
View File

@ -29,12 +29,15 @@ def file_output_path(filename: str, type: Literal["input", "output", "temp"] = "
output_dir = folder_paths.get_directory_by_type(type)
if output_dir is None:
raise ValueError(f"no such output directory because invalid type specified (type={type})")
if subfolder is not None:
if subfolder is not None and subfolder != "":
full_output_dir = os.path.join(output_dir, subfolder)
if os.path.commonpath((os.path.abspath(full_output_dir), output_dir)) != output_dir:
if os.path.commonpath([os.path.abspath(full_output_dir), output_dir]) != output_dir:
raise PermissionError("insecure")
output_dir = full_output_dir
filename = os.path.basename(filename)
else:
if os.path.commonpath([os.path.abspath(output_dir), os.path.join(output_dir, filename)]) != output_dir:
raise PermissionError("insecure")
filename = os.path.basename(filename)
file = os.path.join(output_dir, filename)
return file

7
comfy/nodes/package_typing.py
View File

@ -32,7 +32,7 @@ class StringSpecOptions(TypedDict, total=True):
# todo: analyze the base_nodes for these types
CommonReturnTypes = Union[
Literal["IMAGE", "STRING", "INT", "FLOAT", "CONDITIONING", "LATENT", "MASK", "MODEL", "VAE", "CLIP"], str]
Literal["IMAGE", "STRING", "INT", "BOOLEAN", "FLOAT", "CONDITIONING", "LATENT", "MASK", "MODEL", "VAE", "CLIP"], str]
IntSpec = Tuple[Literal["INT"], IntSpecOptions]
@ -96,7 +96,7 @@ class CustomNode(Protocol):
def INPUT_TYPES(cls) -> InputTypes: ...
# Optional method signature for VALIDATE_INPUTS
VALIDATE_INPUTS: ClassVar[ValidateInputsMethod] = None
VALIDATE_INPUTS: Optional[ClassVar[ValidateInputsMethod]]
RETURN_TYPES: ClassVar[Tuple[CommonReturnTypes, ...]]
RETURN_NAMES: Optional[ClassVar[Tuple[str, ...]]]
@ -106,9 +106,6 @@ class CustomNode(Protocol):
CATEGORY: ClassVar[str]
OUTPUT_NODE: Optional[ClassVar[bool]]
def __call__(self) -> T:
...
@dataclass
class ExportedNodes: