From 8ff0f1dac32a752dddc39dfd06971947d87a177d Mon Sep 17 00:00:00 2001 From: "Dr.Lt.Data" Date: Thu, 25 Apr 2024 22:45:23 +0900 Subject: [PATCH] update DB enhance scanner.py --- extension-node-map.json | 105 +++++++++++++++++++++++++++- node_db/new/extension-node-map.json | 105 +++++++++++++++++++++++++++- scanner.py | 5 ++ 3 files changed, 213 insertions(+), 2 deletions(-) diff --git a/extension-node-map.json b/extension-node-map.json index b034bb88..0c351419 100644 --- a/extension-node-map.json +++ b/extension-node-map.json @@ -4961,6 +4961,18 @@ "title_aux": "StableCascadeResizer" } ], + "https://github.com/ansonkao/comfyui-geometry": [ + [ + "TransformTemplateOntoFaceMask" + ], + { + "author": "Anson Kao", + "description": "A small collection of custom nodes for use with ComfyUI, by Discopixel", + "nickname": "ComfyUI Discopixel", + "title": "ComfyUI Discopixel", + "title_aux": "comfyui-geometry" + } + ], "https://github.com/antrobot1234/antrobots-comfyUI-nodepack": [ [ "composite", @@ -6954,6 +6966,18 @@ "title_aux": "ComfyUI-Vextra-Nodes" } ], + "https://github.com/discopixel-studio/comfyui-discopixel": [ + [ + "TransformTemplateOntoFaceMask" + ], + { + "author": "Anson Kao", + "description": "A small collection of custom nodes for use with ComfyUI, by Discopixel", + "nickname": "ComfyUI Discopixel", + "title": "ComfyUI Discopixel", + "title_aux": "ComfyUI Discopixel Nodes" + } + ], "https://github.com/discus0434/comfyui-caching-embeddings": [ [ "CachingCLIPTextEncode" @@ -8060,8 +8084,87 @@ "title_aux": "ComfyUI_rotate_image" } ], + "https://github.com/jamesWalker55/comfyui-p2ldgan": [ + [ + "P2LDGAN" + ], + { + "title_aux": "ComfyUI - P2LDGAN Node" + } + ], "https://github.com/jamesWalker55/comfyui-various": [ - [], + [ + "BatchLoadImage", + "BatchSaveImage", + "GroupInfoExtractFloat", + "GroupInfoExtractInt", + "GroupLoadBatchImages", + "GroupLoadImage", + "JWDatetimeString", + "JWImageBatchCount", + "JWImageContrast", + "JWImageExtractFromBatch", + "JWImageFlip", + "JWImageLevels", + "JWImageLoadRGB", + "JWImageLoadRGBA", + "JWImageLoadRGBIfExists", + "JWImageMix", + "JWImageResize", + "JWImageResizeByFactor", + "JWImageResizeByLongerSide", + "JWImageResizeByShorterSide", + "JWImageResizeToSquare", + "JWImageSaturation", + "JWImageSaveToPath", + "JWImageSequenceExtractFromBatch", + "JWImageStackChannels", + "JWInfoHashExtractFloat", + "JWInfoHashExtractInteger", + "JWInfoHashExtractString", + "JWInfoHashFromInfoHashList", + "JWInfoHashFromRangedInfo", + "JWInfoHashListExtractStringList", + "JWInfoHashListFromRangedInfo", + "JWInfoHashPrint", + "JWLoadImageSequence", + "JWLoadImagesFromString", + "JWLoopImageSequence", + "JWMaskLikeImageSize", + "JWMaskResize", + "JWMaskSequenceApplyToLatent", + "JWMaskSequenceFromMask", + "JWMaskSequenceJoin", + "JWPrintFloat", + "JWPrintImage", + "JWPrintInteger", + "JWPrintLatent", + "JWPrintMask", + "JWPrintString", + "JWRangedInfoCalculateSubBatch", + "JWReferenceOnly", + "JWSaveImageSequence", + "JWStringListCLIPEncode", + "JWStringListFromString", + "JWStringListFromStrings", + "JWStringListJoin", + "JWStringListRepeat", + "JWStringListToFormatedString", + "JWStringListToString", + "JWUncropCrop", + "JWUncropNewRect", + "JWUncropUncrop", + "JamesLoadImageGroup", + "RAFTEstimate", + "RAFTFlowToImage", + "RAFTLoadFlowFromEXRChannels", + "RCReceiveFloat", + "RCReceiveFloatList", + "RCReceiveInt", + "RCReceiveIntList", + "RCReceiveLatent", + "RCSendLatent" + ], { "nodename_pattern": "^JW", "title_aux": "Various ComfyUI Nodes by Type" diff --git a/node_db/new/extension-node-map.json b/node_db/new/extension-node-map.json index b034bb88..0c351419 100644 --- a/node_db/new/extension-node-map.json +++ b/node_db/new/extension-node-map.json @@ -4961,6 +4961,18 @@ "title_aux": "StableCascadeResizer" } ], + "https://github.com/ansonkao/comfyui-geometry": [ + [ + "TransformTemplateOntoFaceMask" + ], + { + "author": "Anson Kao", + "description": "A small collection of custom nodes for use with ComfyUI, by Discopixel", + "nickname": "ComfyUI Discopixel", + "title": "ComfyUI Discopixel", + "title_aux": "comfyui-geometry" + } + ], "https://github.com/antrobot1234/antrobots-comfyUI-nodepack": [ [ "composite", @@ -6954,6 +6966,18 @@ "title_aux": "ComfyUI-Vextra-Nodes" } ], + "https://github.com/discopixel-studio/comfyui-discopixel": [ + [ + "TransformTemplateOntoFaceMask" + ], + { + "author": "Anson Kao", + "description": "A small collection of custom nodes for use with ComfyUI, by Discopixel", + "nickname": "ComfyUI Discopixel", + "title": "ComfyUI Discopixel", + "title_aux": "ComfyUI Discopixel Nodes" + } + ], "https://github.com/discus0434/comfyui-caching-embeddings": [ [ "CachingCLIPTextEncode" @@ -8060,8 +8084,87 @@ "title_aux": "ComfyUI_rotate_image" } ], + "https://github.com/jamesWalker55/comfyui-p2ldgan": [ + [ + "P2LDGAN" + ], + { + "title_aux": "ComfyUI - P2LDGAN Node" + } + ], "https://github.com/jamesWalker55/comfyui-various": [ - [], + [ + "BatchLoadImage", + "BatchSaveImage", + "GroupInfoExtractFloat", + "GroupInfoExtractInt", + "GroupLoadBatchImages", + "GroupLoadImage", + "JWDatetimeString", + "JWImageBatchCount", + "JWImageContrast", + "JWImageExtractFromBatch", + "JWImageFlip", + "JWImageLevels", + "JWImageLoadRGB", + "JWImageLoadRGBA", + "JWImageLoadRGBIfExists", + "JWImageMix", + "JWImageResize", + "JWImageResizeByFactor", + "JWImageResizeByLongerSide", + "JWImageResizeByShorterSide", + "JWImageResizeToSquare", + "JWImageSaturation", + "JWImageSaveToPath", + "JWImageSequenceExtractFromBatch", + "JWImageStackChannels", + "JWInfoHashExtractFloat", + "JWInfoHashExtractInteger", + "JWInfoHashExtractString", + "JWInfoHashFromInfoHashList", + "JWInfoHashFromRangedInfo", + "JWInfoHashListExtractStringList", + "JWInfoHashListFromRangedInfo", + "JWInfoHashPrint", + "JWLoadImageSequence", + "JWLoadImagesFromString", + "JWLoopImageSequence", + "JWMaskLikeImageSize", + "JWMaskResize", + "JWMaskSequenceApplyToLatent", + "JWMaskSequenceFromMask", + "JWMaskSequenceJoin", + "JWPrintFloat", + "JWPrintImage", + "JWPrintInteger", + "JWPrintLatent", + "JWPrintMask", + "JWPrintString", + "JWRangedInfoCalculateSubBatch", + "JWReferenceOnly", + "JWSaveImageSequence", + "JWStringListCLIPEncode", + "JWStringListFromString", + "JWStringListFromStrings", + "JWStringListJoin", + "JWStringListRepeat", + "JWStringListToFormatedString", + "JWStringListToString", + "JWUncropCrop", + "JWUncropNewRect", + "JWUncropUncrop", + "JamesLoadImageGroup", + "RAFTEstimate", + "RAFTFlowToImage", + "RAFTLoadFlowFromEXRChannels", + "RCReceiveFloat", + "RCReceiveFloatList", + "RCReceiveInt", + "RCReceiveIntList", + "RCReceiveLatent", + "RCSendLatent" + ], { "nodename_pattern": "^JW", "title_aux": "Various ComfyUI Nodes by Type" diff --git a/scanner.py b/scanner.py index c23bb5fc..d3326c8d 100644 --- a/scanner.py +++ b/scanner.py @@ -88,6 +88,11 @@ def scan_in_file(filename, is_builtin=False): for key in keys: nodes.add(key.strip()) + pattern4 = r'@register_node\("(.+)",\s*\".+"\)' + keys = re.findall(pattern4, code) + for key in keys: + nodes.add(key.strip()) + matches = regex.findall(code) for match in matches: dict_text = match