Browse Source

fix: scanner.py for python 3.12

update DB
pull/664/head
Dr.Lt.Data 6 months ago
parent
commit
6a378052f2
  1. 51
      custom-node-list.json
  2. 178
      extension-node-map.json
  3. 408
      github-stats.json
  4. 43
      node_db/new/custom-node-list.json
  5. 178
      node_db/new/extension-node-map.json
  6. 45
      scanner.py

51
custom-node-list.json

@ -8087,6 +8087,7 @@
{
"author": "JettHu",
"title": "ComfyUI-TCD",
"id": "tcd",
"reference": "https://github.com/JettHu/ComfyUI-TCD",
"files": [
"https://github.com/JettHu/ComfyUI-TCD"
@ -8097,6 +8098,7 @@
{
"author": "sugarkwork",
"title": "comfyui_tag_filter",
"id": "tag-filter",
"reference": "https://github.com/sugarkwork/comfyui_tag_fillter",
"files": [
"https://github.com/sugarkwork/comfyui_tag_fillter"
@ -8117,6 +8119,7 @@
{
"author": "TencentQQGYLab",
"title": "ComfyUI-ELLA",
"id": "ella",
"reference": "https://github.com/TencentQQGYLab/ComfyUI-ELLA",
"files": [
"https://github.com/TencentQQGYLab/ComfyUI-ELLA"
@ -8127,6 +8130,7 @@
{
"author": "DarKDinDoN",
"title": "ComfyUI Checkpoint Automatic Config",
"id": "checkpoint-autoconfig",
"reference": "https://github.com/DarKDinDoN/comfyui-checkpoint-automatic-config",
"files": [
"https://github.com/DarKDinDoN/comfyui-checkpoint-automatic-config"
@ -8137,6 +8141,7 @@
{
"author": "aburahamu",
"title": "ComfyUI-RequestPoster",
"id": "request-poster",
"reference": "https://github.com/aburahamu/ComfyUI-RequestsPoster",
"files": [
"https://github.com/aburahamu/ComfyUI-RequestsPoster"
@ -8147,6 +8152,7 @@
{
"author": "MinusZoneAI",
"title": "ComfyUI-Prompt-MZ",
"id": "prompt-mz",
"reference": "https://github.com/MinusZoneAI/ComfyUI-Prompt-MZ",
"files": [
"https://github.com/MinusZoneAI/ComfyUI-Prompt-MZ"
@ -8154,9 +8160,21 @@
"install_type": "git-clone",
"description": "Use llama.cpp to help generate some nodes for prompt word related work"
},
{
"author": "MinusZoneAI",
"title": "ComfyUI-StylizePhoto-MZ",
"id": "stylizephoto",
"reference": "https://github.com/MinusZoneAI/ComfyUI-StylizePhoto-MZ",
"files": [
"https://github.com/MinusZoneAI/ComfyUI-StylizePhoto-MZ"
],
"install_type": "git-clone",
"description": "A stylized node with simple operation. The effect is achieved by I2I and lora. The clay style is currently implemented.Comes with watermark function."
},
{
"author": "blueraincoatli",
"title": "comfyUI_SillyNodes",
"id": "silly",
"reference": "https://github.com/blueraincoatli/comfyUI_SillyNodes",
"files": [
"https://github.com/blueraincoatli/comfyUI_SillyNodes"
@ -8167,6 +8185,7 @@
{
"author": "ty0x2333",
"title": "ComfyUI-Dev-Utils",
"id": "dev-utils",
"reference": "https://github.com/ty0x2333/ComfyUI-Dev-Utils",
"files": [
"https://github.com/ty0x2333/ComfyUI-Dev-Utils"
@ -8520,6 +8539,28 @@
"install_type": "git-clone",
"description": "Misc Nodes: ControlNet Selector Node, Load Optional ControlNet Model"
},
{
"author": "mephisto83",
"title": "petty-paint-comfyui-node",
"id": "petty-paint",
"reference": "https://github.com/mephisto83/petty-paint-comfyui-node",
"files": [
"https://github.com/mephisto83/petty-paint-comfyui-node"
],
"install_type": "git-clone",
"description": "An integration between comfy ui and petty paint"
},
{
"author": "fsdymy1024",
"title": "ComfyUI_fsdymy",
"id": "fsdymy",
"reference": "https://github.com/fsdymy1024/ComfyUI_fsdymy",
"files": [
"https://github.com/fsdymy1024/ComfyUI_fsdymy"
],
"install_type": "git-clone",
"description": "Nodes:Save Image Without Metadata"
},
@ -8846,6 +8887,16 @@
"install_type": "copy",
"description": "Nodes:Send Video to Discord"
},
{
"author": "xliry",
"title": "color2rgb",
"reference": "https://github.com/vxinhao/color2rgb",
"files": [
"https://github.com/vxinhao/color2rgb/raw/main/color2rgb.py"
],
"install_type": "copy",
"description": "Nodes:color2RGB"
},
{

178
extension-node-map.json

@ -475,6 +475,22 @@
"title_aux": "ComfyUI-Aimidi-nodes"
}
],
"https://github.com/AlekPet/ComfyUI_Custom_Nodes_AlekPet": [
[
"ArgosTranslateCLIPTextEncodeNode",
"ArgosTranslateTextNode",
"DeepTranslatorCLIPTextEncodeNode",
"DeepTranslatorTextNode",
"GoogleTranslateCLIPTextEncodeNode",
"GoogleTranslateTextNode",
"PainterNode",
"PoseNode",
"PreviewTextNode"
],
{
"title_aux": "AlekPet/ComfyUI_Custom_Nodes_AlekPet"
}
],
"https://github.com/Alysondao/Comfyui-Yolov8-JSON": [
[
"Apply Yolov8 Model",
@ -4085,6 +4101,7 @@
"https://github.com/TinyTerra/ComfyUI_tinyterraNodes": [
[
"ttN KSampler_v2",
"ttN advPlot merge",
"ttN advPlot range",
"ttN advanced xyPlot",
"ttN compareInput",
@ -7924,6 +7941,14 @@
"title_aux": "ComfyUI_MagicClothing"
}
],
"https://github.com/fsdymy1024/ComfyUI_fsdymy": [
[
"Save Image Without Metadata"
],
{
"title_aux": "ComfyUI_fsdymy"
}
],
"https://github.com/gemell1/ComfyUI_GMIC": [
[
"GmicCliWrapper",
@ -7990,6 +8015,92 @@
"title_aux": "SaltAI-Open-Resources"
}
],
"https://github.com/get-salt-AI/SaltAI_LlamaIndex": [
[
"AddTool",
"ChangeSystemMessage",
"ClearMemory",
"ConversableAgentCreator",
"ConversableAgentCreatorAdvanced",
"ConvertAgentAsTool",
"ConvertAgentToLlamaindex",
"CreateTavilySearchTool",
"GenerateReply",
"GroupChat",
"GroupChatAdvanced",
"GroupChatManagerCreator",
"LLMCSVReader",
"LLMChat",
"LLMChatBot",
"LLMChatEngine",
"LLMChatMessageConcat",
"LLMChatMessages",
"LLMChatMessagesAdv",
"LLMComplete",
"LLMDirectoryReader",
"LLMDocumentListAppend",
"LLMDocxReader",
"LLMEpubReader",
"LLMFlatReader",
"LLMGoogleDocsReader",
"LLMHTMLTagReader",
"LLMHWPReader",
"LLMHtmlComposer",
"LLMHtmlRepair",
"LLMIPYNBReader",
"LLMImageCaptionReader",
"LLMImageTabularChartReader",
"LLMImageTextReader",
"LLMImageVisionLLMReader",
"LLMInputToDocuments",
"LLMJsonComposer",
"LLMJsonRepair",
"LLMMarkdownComposer",
"LLMMarkdownReader",
"LLMMarkdownRepair",
"LLMMboxReader",
"LLMNotionReader",
"LLMOpenAIModel",
"LLMOpenAIModelOpts",
"LLMPDFReader",
"LLMPagedCSVReader",
"LLMPandasCSVReader",
"LLMPostProcessDocuments",
"LLMPptxReader",
"LLMPyMuPDFReader",
"LLMQueryEngine",
"LLMQueryEngineAdv",
"LLMQueryEngineAsTool",
"LLMRTFReader",
"LLMRegexCreator",
"LLMRegexRepair",
"LLMRssReaderNode",
"LLMSaltWebCrawler",
"LLMSemanticSplitterNodeParser",
"LLMSentenceSplitterNodeCreator",
"LLMServiceContextAdv",
"LLMServiceContextDefault",
"LLMSimpleWebPageReader",
"LLMSimpleWebPageReaderAdv",
"LLMSummaryIndex",
"LLMTavilyResearch",
"LLMTrafilaturaWebReader",
"LLMTrafilaturaWebReaderAdv",
"LLMTreeIndex",
"LLMUnstructuredReader",
"LLMVectorStoreIndex",
"LLMVideoAudioReader",
"LLMXMLReader",
"LLMYamlComposer",
"LLMYamlRepair",
"SaltJSONQueryEngine",
"SendMessage",
"SimpleChat"
],
{
"title_aux": "SaltAI_LlamaIndex"
}
],
"https://github.com/giriss/comfy-image-saver": [
[
"Cfg Literal",
@ -8151,6 +8262,7 @@
"ACE_Float",
"ACE_ImageColorFix",
"ACE_ImageConstrain",
"ACE_ImageFaceCrop",
"ACE_ImageGetSize",
"ACE_ImageLoadFromCloud",
"ACE_ImageQA",
@ -8651,8 +8763,11 @@
"FacelessFaceRestore",
"FacelessFaceSwap",
"FacelessLoadFrames",
"FacelessLoadImageUrl",
"FacelessLoadVideo",
"FacelessLoadVideoUrl",
"FacelessSaveVideo",
"FacelessUploadVideo",
"FacelessVideoFaceRestore",
"FacelessVideoFaceSwap"
],
@ -9035,6 +9150,16 @@
"title_aux": "Geowizard depth and normal estimation in ComfyUI"
}
],
"https://github.com/kijai/ComfyUI-IC-Light": [
[
"ICLightConditioning",
"LightSource",
"LoadAndApplyICLightUnet"
],
{
"title_aux": "ComfyUI-IC-Light"
}
],
"https://github.com/kijai/ComfyUI-KJNodes": [
[
"AddLabel",
@ -9080,6 +9205,7 @@
"GenerateNoise",
"GetImageRangeFromBatch",
"GetImageSizeAndCount",
"GetImagesFromBatchIndexed",
"GetLatentsFromBatchIndexed",
"GetMaskSizeAndCount",
"GrowMaskWithBlur",
@ -9109,6 +9235,7 @@
"MaskBatchMulti",
"MaskOrImageToWeight",
"MergeImageChannels",
"ModelPassThrough",
"NormalizedAmplitudeToFloatList",
"NormalizedAmplitudeToMask",
"OffsetMask",
@ -10107,6 +10234,32 @@
"title_aux": "MTB Nodes"
}
],
"https://github.com/mephisto83/petty-paint-comfyui-node": [
[
"PettyPaintAppend",
"PettyPaintComponent",
"PettyPaintConditioningSetMaskAndCombine",
"PettyPaintConvert",
"PettyPaintExec",
"PettyPaintImageCompositeMasked",
"PettyPaintImageSave",
"PettyPaintImageStore",
"PettyPaintImageToMask",
"PettyPaintJsonMap",
"PettyPaintJsonRead",
"PettyPaintJsonReadArray",
"PettyPaintLoadImages",
"PettyPaintMap",
"PettyPaintRemoveAddText",
"PettyPaintSDTurboScheduler",
"PettyPaintText",
"PettyPaintTexts_to_Conditioning",
"PettyPaintToJson"
],
{
"title_aux": "petty-paint-comfyui-node"
}
],
"https://github.com/meshmesh-io/ComfyUI-MeshMesh": [
[
"ColorPicker",
@ -10191,6 +10344,7 @@
"IntSubtraction",
"IntToFloatMultiplication",
"LogicNot",
"NoneToZero",
"NumeralToString",
"OneFloat",
"PngColorMasksToMaskList",
@ -10201,6 +10355,9 @@
"PngRectanglesToMaskList",
"RandomNestedLayouts",
"RandomTillingLayouts",
"SN74HC1G86",
"SN74HC86",
"SN74LVC1G125",
"SeedGenerator",
"SingleBooleanTrigger",
"SixBooleanTrigger",
@ -10903,6 +11060,15 @@
"title_aux": "comfy_meh"
}
],
"https://github.com/saftle/suplex_comfy_nodes": [
[
"ControlNet Selector",
"ControlNetOptionalLoader"
],
{
"title_aux": "Suplex Misc ComfyUI Nodes"
}
],
"https://github.com/sdfxai/SDFXBridgeForComfyUI": [
[
"SDFXClipTextEncode"
@ -11171,6 +11337,8 @@
],
"https://github.com/sipherxyz/comfyui-art-venture": [
[
"AV_AwsBedrockClaudeApi",
"AV_AwsBedrockMistralApi",
"AV_CheckpointMerge",
"AV_CheckpointModelsToParametersPipe",
"AV_CheckpointSave",
@ -11183,6 +11351,7 @@
"AV_ControlNetPreprocessor",
"AV_LLMApiConfig",
"AV_LLMChat",
"AV_LLMCompletion",
"AV_LLMMessage",
"AV_LoraListLoader",
"AV_LoraListStacker",
@ -11998,6 +12167,14 @@
"title_aux": "Kandinsky 2.2 ComfyUI Plugin"
}
],
"https://github.com/vxinhao/color2rgb/raw/main/color2rgb.py": [
[
"color2RGB"
],
{
"title_aux": "color2rgb"
}
],
"https://github.com/wallish77/wlsh_nodes": [
[
"Alternating KSampler (WLSH)",
@ -12355,6 +12532,7 @@
"easy preSamplingLayerDiffusionADDTL",
"easy preSamplingNoiseIn",
"easy preSamplingSdTurbo",
"easy prompt",
"easy promptConcat",
"easy promptLine",
"easy promptList",

408
github-stats.json

File diff suppressed because it is too large Load Diff

43
node_db/new/custom-node-list.json

@ -22,6 +22,28 @@
"install_type": "git-clone",
"description": "ComfyUI native nodes for IC-Light"
},
{
"author": "MinusZoneAI",
"title": "ComfyUI-StylizePhoto-MZ",
"id": "stylizephoto",
"reference": "https://github.com/MinusZoneAI/ComfyUI-StylizePhoto-MZ",
"files": [
"https://github.com/MinusZoneAI/ComfyUI-StylizePhoto-MZ"
],
"install_type": "git-clone",
"description": "A stylized node with simple operation. The effect is achieved by I2I and lora. The clay style is currently implemented.Comes with watermark function."
},
{
"author": "mephisto83",
"title": "petty-paint-comfyui-node",
"id": "petty-paint",
"reference": "https://github.com/mephisto83/petty-paint-comfyui-node",
"files": [
"https://github.com/mephisto83/petty-paint-comfyui-node"
],
"install_type": "git-clone",
"description": "An integration between comfy ui and petty paint"
},
{
"author": "Suplex",
"title": "Suplex Misc ComfyUI Nodes",
@ -33,6 +55,27 @@
"install_type": "git-clone",
"description": "Misc Nodes: ControlNet Selector Node, Load Optional ControlNet Model"
},
{
"author": "fsdymy1024",
"title": "ComfyUI_fsdymy",
"id": "fsdymy",
"reference": "https://github.com/fsdymy1024/ComfyUI_fsdymy",
"files": [
"https://github.com/fsdymy1024/ComfyUI_fsdymy"
],
"install_type": "git-clone",
"description": "Nodes:Save Image Without Metadata"
},
{
"author": "xliry",
"title": "color2rgb",
"reference": "https://github.com/vxinhao/color2rgb",
"files": [
"https://github.com/vxinhao/color2rgb/raw/main/color2rgb.py"
],
"install_type": "copy",
"description": "Nodes:color2RGB"
},
{
"author": "SaltAI",
"title": "SaltAI_LlamaIndex",

178
node_db/new/extension-node-map.json

@ -475,6 +475,22 @@
"title_aux": "ComfyUI-Aimidi-nodes"
}
],
"https://github.com/AlekPet/ComfyUI_Custom_Nodes_AlekPet": [
[
"ArgosTranslateCLIPTextEncodeNode",
"ArgosTranslateTextNode",
"DeepTranslatorCLIPTextEncodeNode",
"DeepTranslatorTextNode",
"GoogleTranslateCLIPTextEncodeNode",
"GoogleTranslateTextNode",
"PainterNode",
"PoseNode",
"PreviewTextNode"
],
{
"title_aux": "AlekPet/ComfyUI_Custom_Nodes_AlekPet"
}
],
"https://github.com/Alysondao/Comfyui-Yolov8-JSON": [
[
"Apply Yolov8 Model",
@ -4085,6 +4101,7 @@
"https://github.com/TinyTerra/ComfyUI_tinyterraNodes": [
[
"ttN KSampler_v2",
"ttN advPlot merge",
"ttN advPlot range",
"ttN advanced xyPlot",
"ttN compareInput",
@ -7924,6 +7941,14 @@
"title_aux": "ComfyUI_MagicClothing"
}
],
"https://github.com/fsdymy1024/ComfyUI_fsdymy": [
[
"Save Image Without Metadata"
],
{
"title_aux": "ComfyUI_fsdymy"
}
],
"https://github.com/gemell1/ComfyUI_GMIC": [
[
"GmicCliWrapper",
@ -7990,6 +8015,92 @@
"title_aux": "SaltAI-Open-Resources"
}
],
"https://github.com/get-salt-AI/SaltAI_LlamaIndex": [
[
"AddTool",
"ChangeSystemMessage",
"ClearMemory",
"ConversableAgentCreator",
"ConversableAgentCreatorAdvanced",
"ConvertAgentAsTool",
"ConvertAgentToLlamaindex",
"CreateTavilySearchTool",
"GenerateReply",
"GroupChat",
"GroupChatAdvanced",
"GroupChatManagerCreator",
"LLMCSVReader",
"LLMChat",
"LLMChatBot",
"LLMChatEngine",
"LLMChatMessageConcat",
"LLMChatMessages",
"LLMChatMessagesAdv",
"LLMComplete",
"LLMDirectoryReader",
"LLMDocumentListAppend",
"LLMDocxReader",
"LLMEpubReader",
"LLMFlatReader",
"LLMGoogleDocsReader",
"LLMHTMLTagReader",
"LLMHWPReader",
"LLMHtmlComposer",
"LLMHtmlRepair",
"LLMIPYNBReader",
"LLMImageCaptionReader",
"LLMImageTabularChartReader",
"LLMImageTextReader",
"LLMImageVisionLLMReader",
"LLMInputToDocuments",
"LLMJsonComposer",
"LLMJsonRepair",
"LLMMarkdownComposer",
"LLMMarkdownReader",
"LLMMarkdownRepair",
"LLMMboxReader",
"LLMNotionReader",
"LLMOpenAIModel",
"LLMOpenAIModelOpts",
"LLMPDFReader",
"LLMPagedCSVReader",
"LLMPandasCSVReader",
"LLMPostProcessDocuments",
"LLMPptxReader",
"LLMPyMuPDFReader",
"LLMQueryEngine",
"LLMQueryEngineAdv",
"LLMQueryEngineAsTool",
"LLMRTFReader",
"LLMRegexCreator",
"LLMRegexRepair",
"LLMRssReaderNode",
"LLMSaltWebCrawler",
"LLMSemanticSplitterNodeParser",
"LLMSentenceSplitterNodeCreator",
"LLMServiceContextAdv",
"LLMServiceContextDefault",
"LLMSimpleWebPageReader",
"LLMSimpleWebPageReaderAdv",
"LLMSummaryIndex",
"LLMTavilyResearch",
"LLMTrafilaturaWebReader",
"LLMTrafilaturaWebReaderAdv",
"LLMTreeIndex",
"LLMUnstructuredReader",
"LLMVectorStoreIndex",
"LLMVideoAudioReader",
"LLMXMLReader",
"LLMYamlComposer",
"LLMYamlRepair",
"SaltJSONQueryEngine",
"SendMessage",
"SimpleChat"
],
{
"title_aux": "SaltAI_LlamaIndex"
}
],
"https://github.com/giriss/comfy-image-saver": [
[
"Cfg Literal",
@ -8151,6 +8262,7 @@
"ACE_Float",
"ACE_ImageColorFix",
"ACE_ImageConstrain",
"ACE_ImageFaceCrop",
"ACE_ImageGetSize",
"ACE_ImageLoadFromCloud",
"ACE_ImageQA",
@ -8651,8 +8763,11 @@
"FacelessFaceRestore",
"FacelessFaceSwap",
"FacelessLoadFrames",
"FacelessLoadImageUrl",
"FacelessLoadVideo",
"FacelessLoadVideoUrl",
"FacelessSaveVideo",
"FacelessUploadVideo",
"FacelessVideoFaceRestore",
"FacelessVideoFaceSwap"
],
@ -9035,6 +9150,16 @@
"title_aux": "Geowizard depth and normal estimation in ComfyUI"
}
],
"https://github.com/kijai/ComfyUI-IC-Light": [
[
"ICLightConditioning",
"LightSource",
"LoadAndApplyICLightUnet"
],
{
"title_aux": "ComfyUI-IC-Light"
}
],
"https://github.com/kijai/ComfyUI-KJNodes": [
[
"AddLabel",
@ -9080,6 +9205,7 @@
"GenerateNoise",
"GetImageRangeFromBatch",
"GetImageSizeAndCount",
"GetImagesFromBatchIndexed",
"GetLatentsFromBatchIndexed",
"GetMaskSizeAndCount",
"GrowMaskWithBlur",
@ -9109,6 +9235,7 @@
"MaskBatchMulti",
"MaskOrImageToWeight",
"MergeImageChannels",
"ModelPassThrough",
"NormalizedAmplitudeToFloatList",
"NormalizedAmplitudeToMask",
"OffsetMask",
@ -10107,6 +10234,32 @@
"title_aux": "MTB Nodes"
}
],
"https://github.com/mephisto83/petty-paint-comfyui-node": [
[
"PettyPaintAppend",
"PettyPaintComponent",
"PettyPaintConditioningSetMaskAndCombine",
"PettyPaintConvert",
"PettyPaintExec",
"PettyPaintImageCompositeMasked",
"PettyPaintImageSave",
"PettyPaintImageStore",
"PettyPaintImageToMask",
"PettyPaintJsonMap",
"PettyPaintJsonRead",
"PettyPaintJsonReadArray",
"PettyPaintLoadImages",
"PettyPaintMap",
"PettyPaintRemoveAddText",
"PettyPaintSDTurboScheduler",
"PettyPaintText",
"PettyPaintTexts_to_Conditioning",
"PettyPaintToJson"
],
{
"title_aux": "petty-paint-comfyui-node"
}
],
"https://github.com/meshmesh-io/ComfyUI-MeshMesh": [
[
"ColorPicker",
@ -10191,6 +10344,7 @@
"IntSubtraction",
"IntToFloatMultiplication",
"LogicNot",
"NoneToZero",
"NumeralToString",
"OneFloat",
"PngColorMasksToMaskList",
@ -10201,6 +10355,9 @@
"PngRectanglesToMaskList",
"RandomNestedLayouts",
"RandomTillingLayouts",
"SN74HC1G86",
"SN74HC86",
"SN74LVC1G125",
"SeedGenerator",
"SingleBooleanTrigger",
"SixBooleanTrigger",
@ -10903,6 +11060,15 @@
"title_aux": "comfy_meh"
}
],
"https://github.com/saftle/suplex_comfy_nodes": [
[
"ControlNet Selector",
"ControlNetOptionalLoader"
],
{
"title_aux": "Suplex Misc ComfyUI Nodes"
}
],
"https://github.com/sdfxai/SDFXBridgeForComfyUI": [
[
"SDFXClipTextEncode"
@ -11171,6 +11337,8 @@
],
"https://github.com/sipherxyz/comfyui-art-venture": [
[
"AV_AwsBedrockClaudeApi",
"AV_AwsBedrockMistralApi",
"AV_CheckpointMerge",
"AV_CheckpointModelsToParametersPipe",
"AV_CheckpointSave",
@ -11183,6 +11351,7 @@
"AV_ControlNetPreprocessor",
"AV_LLMApiConfig",
"AV_LLMChat",
"AV_LLMCompletion",
"AV_LLMMessage",
"AV_LoraListLoader",
"AV_LoraListStacker",
@ -11998,6 +12167,14 @@
"title_aux": "Kandinsky 2.2 ComfyUI Plugin"
}
],
"https://github.com/vxinhao/color2rgb/raw/main/color2rgb.py": [
[
"color2RGB"
],
{
"title_aux": "color2rgb"
}
],
"https://github.com/wallish77/wlsh_nodes": [
[
"Alternating KSampler (WLSH)",
@ -12355,6 +12532,7 @@
"easy preSamplingLayerDiffusionADDTL",
"easy preSamplingNoiseIn",
"easy preSamplingSdTurbo",
"easy prompt",
"easy promptConcat",
"easy promptLine",
"easy promptList",

45
scanner.py

@ -37,8 +37,17 @@ else:
print(f"TEMP DIR: {temp_dir}")
parse_cnt = 0
def extract_nodes(code_text):
global parse_cnt
try:
if parse_cnt % 100 == 0:
print(f".", end="", flush=True)
parse_cnt += 1
code_text = re.sub(r'\\[^"\']', '', code_text)
parsed_code = ast.parse(code_text)
assignments = (node for node in parsed_code.body if isinstance(node, ast.Assign))
@ -51,7 +60,12 @@ def extract_nodes(code_text):
node_class_mappings = None
if node_class_mappings:
s = set([key.s.strip() for key in node_class_mappings.keys if key is not None])
s = set()
for key in node_class_mappings.keys:
if key is not None and isinstance(key.value, str):
s.add(key.value.strip())
return s
else:
return set()
@ -78,20 +92,24 @@ def scan_in_file(filename, is_builtin=False):
nodes |= extract_nodes(code)
pattern2 = r'^[^=]*_CLASS_MAPPINGS\["(.*?)"\]'
keys = re.findall(pattern2, code)
for key in keys:
nodes.add(key.strip())
def extract_keys(pattern, code):
keys = re.findall(pattern, code)
return {key.strip() for key in keys}
pattern3 = r'^[^=]*_CLASS_MAPPINGS\[\'(.*?)\'\]'
keys = re.findall(pattern3, code)
for key in keys:
nodes.add(key.strip())
def update_nodes(nodes, new_keys):
nodes |= new_keys
pattern4 = r'@register_node\("(.+)",\s*\".+"\)'
keys = re.findall(pattern4, code)
for key in keys:
nodes.add(key.strip())
patterns = [
r'^[^=]*_CLASS_MAPPINGS\["(.*?)"\]',
r'^[^=]*_CLASS_MAPPINGS\[\'(.*?)\'\]',
r'@register_node\("(.+)",\s*\".+"\)',
r'"(\w+)"\s*:\s*{"class":\s*\w+\s*'
]
with concurrent.futures.ThreadPoolExecutor() as executor:
futures = {executor.submit(extract_keys, pattern, code): pattern for pattern in patterns}
for future in concurrent.futures.as_completed(futures):
update_nodes(nodes, future.result())
matches = regex.findall(code)
for match in matches:
@ -446,3 +464,4 @@ updated_node_info = update_custom_nodes()
print("\n# 'extension-node-map.json' file is generated.\n")
gen_json(updated_node_info)
print("\nDONE.\n")
Loading…
Cancel
Save