commit 0c3e287168a025086e030e05665a5514ecb8d68e Author: drago87 Date: Mon Nov 20 20:53:02 2023 +0100 Add files via upload diff --git a/__init__.py b/__init__.py new file mode 100644 index 0000000..7a0a98f --- /dev/null +++ b/__init__.py @@ -0,0 +1 @@ +from .file_padding import * \ No newline at end of file diff --git a/__pycache__/Files.cpython-310.pyc b/__pycache__/Files.cpython-310.pyc new file mode 100644 index 0000000..c04ee91 Binary files /dev/null and b/__pycache__/Files.cpython-310.pyc differ diff --git a/__pycache__/__init__.cpython-310.pyc b/__pycache__/__init__.cpython-310.pyc new file mode 100644 index 0000000..bda7735 Binary files /dev/null and b/__pycache__/__init__.cpython-310.pyc differ diff --git a/__pycache__/file_padding.cpython-310.pyc b/__pycache__/file_padding.cpython-310.pyc new file mode 100644 index 0000000..f99c837 Binary files /dev/null and b/__pycache__/file_padding.cpython-310.pyc differ diff --git a/file_padding.py b/file_padding.py new file mode 100644 index 0000000..f3881ad --- /dev/null +++ b/file_padding.py @@ -0,0 +1,55 @@ +import math +import glob + +class file_padding: + def __init__(self): + pass + + @classmethod + def INPUT_TYPES(s): + return { + "required": { + "path": ("STRING",{ + "multiline": False, + "default": "./ComfyUI/output/" + }), + "padding": ("INT", { + "default": 4, + "min": 0, + "max": 10, + "step": 1, + "display": "number" + }) + } + } + RETURN_TYPES = ("STRING",) + RETURN_NAMES = ("Number of Images (as str)",) + + FUNCTION = "run" + + CATEGORY = "DragosNodes" + + def run(self, path, padding): + #print("init "+str(path)) + if (path[-1]!="/"): + path = path+"/" + #print("after "+ str(path)) + #print(str(glob.glob(path+"*.png"))) + if padding > 0: + + lenght = str(len(glob.glob(path+"*.png"))) + return(lenght) + else: + lenght = "" + for x in padding-1: + lenght = lenght+"0" + lenght = lenght + str(len(glob.glob(path+"*.png"))) + return(lenght) + +NODE_CLASS_MAPPINGS = { + "file_padding": file_padding +} + +NODE_DISPLAY_NAME_MAPPINGS = { + "file_padding": "File Padding" +} \ No newline at end of file