update scanner.py

update DB
This commit is contained in:
Dr.Lt.Data 2024-01-28 17:24:11 +09:00
parent b1f613e649
commit 6e06828aa4
5 changed files with 148 additions and 4 deletions

View File

@ -4664,7 +4664,36 @@
"install_type": "git-clone", "install_type": "git-clone",
"description": "Transcribe audio and add subtitles to videos using Whisper in ComfyUI" "description": "Transcribe audio and add subtitles to videos using Whisper in ComfyUI"
}, },
{
"author": "blepping",
"title": "ComfyUI-bleh",
"reference": "https://github.com/blepping/ComfyUI-bleh",
"files": [
"https://github.com/blepping/ComfyUI-bleh"
],
"install_type": "git-clone",
"description": "Better TAESD previews, BlehHyperTile."
},
{
"author": "JerryOrbachJr",
"title": "ComfyUI-RandomSize",
"reference": "https://github.com/JerryOrbachJr/ComfyUI-RandomSize",
"files": [
"https://github.com/JerryOrbachJr/ComfyUI-RandomSize"
],
"install_type": "git-clone",
"description": "A ComfyUI custom node that randomly selects a height and width pair from a list in a config file"
},
{
"author": "jamal-alkharrat",
"title": "ComfyUI_rotate_image",
"reference": "https://github.com/jamal-alkharrat/ComfyUI_rotate_image",
"files": [
"https://github.com/jamal-alkharrat/ComfyUI_rotate_image"
],
"install_type": "git-clone",
"description": "ComfyUI Custom Node to Rotate Images, Img2Img node."
},

View File

@ -886,6 +886,7 @@
[ [
"ReActorFaceSwap", "ReActorFaceSwap",
"ReActorLoadFaceModel", "ReActorLoadFaceModel",
"ReActorRestoreFace",
"ReActorSaveFaceModel" "ReActorSaveFaceModel"
], ],
{ {
@ -1090,6 +1091,18 @@
"title_aux": "Rembg Background Removal Node for ComfyUI" "title_aux": "Rembg Background Removal Node for ComfyUI"
} }
], ],
"https://github.com/JerryOrbachJr/ComfyUI-RandomSize": [
[
"JOJR_RandomSize"
],
{
"author": "JerryOrbachJr",
"description": "A ComfyUI custom node that randomly selects a height and width pair from a list in a config file",
"nickname": "Random Size",
"title": "Random Size",
"title_aux": "ComfyUI-RandomSize"
}
],
"https://github.com/Jordach/comfy-plasma": [ "https://github.com/Jordach/comfy-plasma": [
[ [
"JDC_AutoContrast", "JDC_AutoContrast",
@ -3449,6 +3462,14 @@
"title_aux": "CLIPSeg" "title_aux": "CLIPSeg"
} }
], ],
"https://github.com/blepping/ComfyUI-bleh": [
[
"BlehHyperTile"
],
{
"title_aux": "ComfyUI-bleh"
}
],
"https://github.com/bmad4ever/comfyui_ab_samplercustom": [ "https://github.com/bmad4ever/comfyui_ab_samplercustom": [
[ [
"AB SamplerCustom (experimental)" "AB SamplerCustom (experimental)"
@ -5073,6 +5094,14 @@
"title_aux": "Efficiency Nodes for ComfyUI Version 2.0+" "title_aux": "Efficiency Nodes for ComfyUI Version 2.0+"
} }
], ],
"https://github.com/jamal-alkharrat/ComfyUI_rotate_image": [
[
"RotateImage"
],
{
"title_aux": "ComfyUI_rotate_image"
}
],
"https://github.com/jamesWalker55/comfyui-various": [ "https://github.com/jamesWalker55/comfyui-various": [
[], [],
{ {
@ -5342,6 +5371,7 @@
], ],
"https://github.com/komojini/komojini-comfyui-nodes": [ "https://github.com/komojini/komojini-comfyui-nodes": [
[ [
"BatchCreativeInterpolationNodeDynamicSettings",
"CachedGetter", "CachedGetter",
"DragNUWAImageCanvas", "DragNUWAImageCanvas",
"FlowBuilder", "FlowBuilder",
@ -6407,6 +6437,7 @@
"AreaToMask", "AreaToMask",
"CLIPSeg", "CLIPSeg",
"CLIPSeg_", "CLIPSeg_",
"CenterImage",
"CharacterInText", "CharacterInText",
"ChatGPTOpenAI", "ChatGPTOpenAI",
"CkptNames_", "CkptNames_",
@ -6473,7 +6504,6 @@
"https://github.com/shiimizu/ComfyUI-PhotoMaker-Plus": [ "https://github.com/shiimizu/ComfyUI-PhotoMaker-Plus": [
[ [
"PhotoMakerEncodePlus", "PhotoMakerEncodePlus",
"PhotoMakerLoader",
"PhotoMakerStyles", "PhotoMakerStyles",
"PrepImagesForClipVisionFromPath" "PrepImagesForClipVisionFromPath"
], ],

View File

@ -10,6 +10,36 @@
}, },
{
"author": "jamal-alkharrat",
"title": "ComfyUI_rotate_image",
"reference": "https://github.com/jamal-alkharrat/ComfyUI_rotate_image",
"files": [
"https://github.com/jamal-alkharrat/ComfyUI_rotate_image"
],
"install_type": "git-clone",
"description": "ComfyUI Custom Node to Rotate Images, Img2Img node."
},
{
"author": "JerryOrbachJr",
"title": "ComfyUI-RandomSize",
"reference": "https://github.com/JerryOrbachJr/ComfyUI-RandomSize",
"files": [
"https://github.com/JerryOrbachJr/ComfyUI-RandomSize"
],
"install_type": "git-clone",
"description": "A ComfyUI custom node that randomly selects a height and width pair from a list in a config file"
},
{
"author": "blepping",
"title": "ComfyUI-bleh",
"reference": "https://github.com/blepping/ComfyUI-bleh",
"files": [
"https://github.com/blepping/ComfyUI-bleh"
],
"install_type": "git-clone",
"description": "Better TAESD previews, BlehHyperTile."
},
{ {
"author": "yuvraj108c", "author": "yuvraj108c",
"title": "ComfyUI Whisper", "title": "ComfyUI Whisper",

View File

@ -886,6 +886,7 @@
[ [
"ReActorFaceSwap", "ReActorFaceSwap",
"ReActorLoadFaceModel", "ReActorLoadFaceModel",
"ReActorRestoreFace",
"ReActorSaveFaceModel" "ReActorSaveFaceModel"
], ],
{ {
@ -1090,6 +1091,18 @@
"title_aux": "Rembg Background Removal Node for ComfyUI" "title_aux": "Rembg Background Removal Node for ComfyUI"
} }
], ],
"https://github.com/JerryOrbachJr/ComfyUI-RandomSize": [
[
"JOJR_RandomSize"
],
{
"author": "JerryOrbachJr",
"description": "A ComfyUI custom node that randomly selects a height and width pair from a list in a config file",
"nickname": "Random Size",
"title": "Random Size",
"title_aux": "ComfyUI-RandomSize"
}
],
"https://github.com/Jordach/comfy-plasma": [ "https://github.com/Jordach/comfy-plasma": [
[ [
"JDC_AutoContrast", "JDC_AutoContrast",
@ -3449,6 +3462,14 @@
"title_aux": "CLIPSeg" "title_aux": "CLIPSeg"
} }
], ],
"https://github.com/blepping/ComfyUI-bleh": [
[
"BlehHyperTile"
],
{
"title_aux": "ComfyUI-bleh"
}
],
"https://github.com/bmad4ever/comfyui_ab_samplercustom": [ "https://github.com/bmad4ever/comfyui_ab_samplercustom": [
[ [
"AB SamplerCustom (experimental)" "AB SamplerCustom (experimental)"
@ -5073,6 +5094,14 @@
"title_aux": "Efficiency Nodes for ComfyUI Version 2.0+" "title_aux": "Efficiency Nodes for ComfyUI Version 2.0+"
} }
], ],
"https://github.com/jamal-alkharrat/ComfyUI_rotate_image": [
[
"RotateImage"
],
{
"title_aux": "ComfyUI_rotate_image"
}
],
"https://github.com/jamesWalker55/comfyui-various": [ "https://github.com/jamesWalker55/comfyui-various": [
[], [],
{ {
@ -5342,6 +5371,7 @@
], ],
"https://github.com/komojini/komojini-comfyui-nodes": [ "https://github.com/komojini/komojini-comfyui-nodes": [
[ [
"BatchCreativeInterpolationNodeDynamicSettings",
"CachedGetter", "CachedGetter",
"DragNUWAImageCanvas", "DragNUWAImageCanvas",
"FlowBuilder", "FlowBuilder",
@ -6407,6 +6437,7 @@
"AreaToMask", "AreaToMask",
"CLIPSeg", "CLIPSeg",
"CLIPSeg_", "CLIPSeg_",
"CenterImage",
"CharacterInText", "CharacterInText",
"ChatGPTOpenAI", "ChatGPTOpenAI",
"CkptNames_", "CkptNames_",
@ -6473,7 +6504,6 @@
"https://github.com/shiimizu/ComfyUI-PhotoMaker-Plus": [ "https://github.com/shiimizu/ComfyUI-PhotoMaker-Plus": [
[ [
"PhotoMakerEncodePlus", "PhotoMakerEncodePlus",
"PhotoMakerLoader",
"PhotoMakerStyles", "PhotoMakerStyles",
"PrepImagesForClipVisionFromPath" "PrepImagesForClipVisionFromPath"
], ],

View File

@ -1,3 +1,4 @@
import ast
import re import re
import os import os
import json import json
@ -22,6 +23,28 @@ if not os.path.exists(temp_dir):
print(f"TEMP DIR: {temp_dir}") print(f"TEMP DIR: {temp_dir}")
def extract_nodes(code_text):
try:
parsed_code = ast.parse(code_text)
assignments = (node for node in parsed_code.body if isinstance(node, ast.Assign))
for assignment in assignments:
if isinstance(assignment.targets[0], ast.Name) and assignment.targets[0].id == 'NODE_CLASS_MAPPINGS':
node_class_mappings = assignment.value
break
else:
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])
return s
else:
return set()
except:
return set()
# scan # scan
def scan_in_file(filename, is_builtin=False): def scan_in_file(filename, is_builtin=False):
global builtin_nodes global builtin_nodes
@ -39,6 +62,8 @@ def scan_in_file(filename, is_builtin=False):
nodes = set() nodes = set()
class_dict = {} class_dict = {}
nodes |= extract_nodes(code)
pattern2 = r'^[^=]*_CLASS_MAPPINGS\["(.*?)"\]' pattern2 = r'^[^=]*_CLASS_MAPPINGS\["(.*?)"\]'
keys = re.findall(pattern2, code) keys = re.findall(pattern2, code)
for key in keys: for key in keys: