From c0976f4dc21d4eab24fdaa4319f35be7f3e32bf3 Mon Sep 17 00:00:00 2001 From: Level Pixel Dev Date: Wed, 30 Jul 2025 05:44:44 +0600 Subject: [PATCH] Added new IO nodes --- nodes/io/folder_workers_LP.py | 115 ++++++++++++++++++++++++++++++++-- pyproject.toml | 2 +- 2 files changed, 112 insertions(+), 5 deletions(-) diff --git a/nodes/io/folder_workers_LP.py b/nodes/io/folder_workers_LP.py index e94e9cb..cc36e24 100644 --- a/nodes/io/folder_workers_LP.py +++ b/nodes/io/folder_workers_LP.py @@ -1,11 +1,8 @@ +import folder_paths import glob import os class FileCounter: - FUNCTION = "file_counter" - OUTPUT_NODE = True - CATEGORY = "LevelPixel/IO" - @classmethod def INPUT_TYPES(cls): return { @@ -19,6 +16,9 @@ class FileCounter: RETURN_NAMES = ("Total INT","Total STRING") FUNCTION = "file_counter" + OUTPUT_NODE = True + CATEGORY = "LevelPixel/IO" + @classmethod def IS_CHANGED(cls, *v): return ALWAYS_CHANGED_FLAG @@ -34,11 +34,118 @@ class FileCounter: print("total " + str(total_int)) return (total_int, total_string) +class GetComfyUIFolderPath: + @classmethod + def INPUT_TYPES(cls): + try: + folder_names = list(folder_paths.folder_names_and_paths.keys()) + except Exception: + folder_names = [] + standard_folders = ["base", "model", "output", "temp", "input"] + for name in standard_folders: + if name not in folder_names: + folder_names.append(name) + return { + "required": { + "folder_name": (folder_names, {"default": folder_names[0] if folder_names else "input"}), + }, + } + + RETURN_TYPES = ("STRING",) + RETURN_NAMES = ("folder_path",) + + CATEGORY = "LevelPixel/IO" + FUNCTION = "get_comfyui_folder_path" + + def get_comfyui_folder_path(self, folder_name): + try: + if folder_name == "base": + return (str(folder_paths.base_path),) + elif folder_name == "model": + return (str(folder_paths.models_dir),) + elif folder_name == "output": + return (str(folder_paths.get_output_directory()),) + elif folder_name == "temp": + return (str(folder_paths.get_temp_directory()),) + elif folder_name == "input": + return (str(folder_paths.get_input_directory()),) + path = folder_paths.get_folder_paths(folder_name) + if isinstance(path, list): + return (path[0],) + return (str(path),) + except Exception as e: + return (f"Error: {e}",) + +class GetComfyUIHttpFolderPath: + @classmethod + def INPUT_TYPES(cls): + folder_names = ["output", "temp", "input"] + return { + "required": { + "folder_name": (folder_names, {"default": folder_names[0]}), + }, + } + + RETURN_TYPES = ("STRING",) + RETURN_NAMES = ("folder_path",) + + CATEGORY = "LevelPixel/IO" + FUNCTION = "get_comfyui_http_folder_path" + + def get_comfyui_http_folder_path(self, folder_name): + try: + path = folder_paths.get_directory_by_type(folder_name) + if path is not None: + return (str(path),) + return (f"Not allowed or unknown folder: {folder_name}",) + except Exception as e: + return (f"Error: {e}",) + + +class GetFilenameByIndexInFolder: + @classmethod + def INPUT_TYPES(cls): + return { + "required": { + "folder_path": ("STRING", {"default": "", "multiline": False}), + "file_index": ("INT", {"default": 0, "min": 0, "step": 1}), + "patterns": ("STRING", {"default": "*", "multiline": False}), + }, + } + + RETURN_TYPES = ("STRING", "STRING", "STRING") + RETURN_NAMES = ("full_filename", "filename", "extension") + + CATEGORY = "LevelPixel/IO" + FUNCTION = "get_filename_by_index_in_folder" + + def get_filename_by_index_in_folder(self, folder_path, file_index, patterns): + if not os.path.isdir(folder_path): + return ("Error: not a directory", "", "") + files = [] + for pattern in patterns.split("|"): + files.extend(glob.glob(os.path.join(folder_path, pattern))) + if not files: + return ("Error: no files found", "", "") + files.sort(key=lambda x: os.path.getctime(x)) + if file_index < 0 or file_index >= len(files): + return (f"Error: index {file_index} out of range (found {len(files)} files)", "", "") + full_filename = os.path.basename(files[file_index]) + filename, extension = os.path.splitext(full_filename) + extension = extension[1:] if extension.startswith(".") else extension + return (full_filename, filename, extension) + NODE_CLASS_MAPPINGS = { "FileCounter|LP": FileCounter, + "GetComfyUIFolderPath|LP": GetComfyUIFolderPath, + "GetComfyUIHttpFolderPath|LP": GetComfyUIHttpFolderPath, + "GetFilenameByIndexInFolder|LP": GetFilenameByIndexInFolder, } NODE_DISPLAY_NAME_MAPPINGS = { "FileCounter|LP": "File Counter [LP]", + "GetComfyUIFolderPath|LP": "Get ComfyUI Folder Path [LP]", + "GetComfyUIHttpFolderPath|LP": "Get ComfyUI HTTP Folder Path [LP]", + "GetFilenameByIndexInFolder|LP": "Get Filename By Index In Folder [LP]", } diff --git a/pyproject.toml b/pyproject.toml index b439dc9..5296885 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,7 +1,7 @@ [project] name = "comfyui-levelpixel" description = "Main nodes of the Level Pixel company (levelpixel, LP). In this package you will find easy-to-use atomic nodes that can help automate various processes - memory cleaning automation (RAM free, VRAM free), Lora loading like in Automatic1111, powerful inpaint crop and stitch, pipeline, declaring standard variables, converting variables, google translate, filtering tags by categories, working with file system, counting files in file system, counting objects, counting images. The names of the key nodes in this package are: Inpaint-CropAndStitch, Inpaint Crop, Inpaint Stitch, Resize Image, Resize Mask, Model Unloader, Free memory, Tag Filters, Tag Category Filters, Tag Choice Parser, File counter, Image Loader From Path (with counters), Google Translator, Language Translator, Override Device, Delay, Lora Tag Loader, Pipeline, Resize Masks, Resize Image. Licensed under GNU GPLv3. Repository: https://github.com/LevelPixel/ComfyUI-LevelPixel" -version = "1.2.8" +version = "1.2.9" license = { file = "LICENSE.txt" } dependencies = [ "torch>=2.0.1",