diff --git a/custom-node-list.json b/custom-node-list.json index 140ce400..7069acdf 100644 --- a/custom-node-list.json +++ b/custom-node-list.json @@ -525,6 +525,16 @@ "install_type": "git-clone", "description": "This extension provides: Auto Arrange Graph, Workflow SVG, Favicon Status, Image Feed, Latent Upscale By, Lock Nodes & Groups, Lora Subfolders, Preset Text, Show Text, Touch Support, Link Render Mode, Locking, Node Finder, Quick Nodes, Show Image On Menu, Show Text, Workflow Managements, Custom Widget Default Values" }, + { + "author": "melMass", + "title": "MTB Nodes", + "reference": "https://github.com/melMass/comfy_mtb", + "files": [ + "https://github.com/melMass/comfy_mtb" + ], + "install_type": "git-clone", + "description": "NODES: Latent Lerp, Int To Number, Bounding Box, Crop, Uncrop, ImageBlur, Denoise, ImageCompare, RGV to HSV, HSV to RGB, Color Correct, Modulo, Deglaze Image, Smart Step" + }, { "author": "xXAdonesXx", "title": "NodeGPT", diff --git a/extension-node-map.json b/extension-node-map.json index 59c79fc3..0a143f67 100644 --- a/extension-node-map.json +++ b/extension-node-map.json @@ -763,6 +763,37 @@ "RawTextEncode", "RawTextReplace" ], + "https://github.com/melMass/comfy_mtb": [ + "Bounding Box (mtb)", + "Bounding Box From Mask (mtb)", + "Color Correct (mtb)", + "Colored Image (mtb)", + "Crop (mtb)", + "Deep Bump (mtb)", + "Deglaze Image (mtb)", + "Denoise (mtb)", + "Face Swap [roop] (mtb)", + "Geometry Info (mtb)", + "HSV to RGB (mtb)", + "Image Premultiply (mtb)", + "Image Remove Background RemBG (mtb)", + "ImageBlur (mtb)", + "ImageCompare (mtb)", + "Int to Number (mtb)", + "Latent Lerp (mtb) [DEPRECATED]", + "Load Geometry (mtb)", + "Load Image Sequence (mtb)", + "MMPose Estimation (mtb)", + "Mask to Image (mtb)", + "Modulo (mtb)", + "QR Code (mtb)", + "RGB to HSV (mtb)", + "Save Image Sequence (mtb)", + "Smart Step (mtb)", + "Styles Loader (mtb)", + "Text to Image (mtb)", + "Uncrop (mtb)" + ], "https://github.com/omar92/ComfyUI-QualityOfLifeSuit_Omar92": [ "CLIPStringEncode _O", "Chat completion _O", diff --git a/scanner.py b/scanner.py index 88e4288c..7a4f9c9d 100644 --- a/scanner.py +++ b/scanner.py @@ -12,14 +12,21 @@ def scan_in_file(filename): pattern = r"NODE_CLASS_MAPPINGS\s*=\s*{([^}]*)}" regex = re.compile(pattern, re.MULTILINE | re.DOTALL) - match = regex.search(code) - if match: - dict_text = match.group(1) - else: + matches = regex.findall(code) + if not matches: return [] nodes = set() class_dict = {} + + for match in matches: + dict_text = match + + pattern2 = r'NODE_CLASS_MAPPINGS\["(.*?)"\]' + keys = re.findall(pattern2, code) + for key in keys: + nodes.add(key) + key_value_pairs = re.findall(r"\"([^\"]*)\"\s*:\s*([^,\n]*)", dict_text) for key, value in key_value_pairs: class_dict[key] = value.strip() @@ -137,7 +144,7 @@ def update_custom_nodes(): name = name[:-4] node_info[name] = url - clone_or_pull_git_repository(url) + # clone_or_pull_git_repository(url) py_urls = get_py_urls_from_json('custom-node-list.json')