Browse Source

update DB

robust scanner.py
pull/25/head
Dr.Lt.Data 1 year ago
parent
commit
6b7dd1b82f
  1. 5
      extension-node-map.json
  2. 10
      scanner.py

5
extension-node-map.json

@ -401,6 +401,7 @@
"CLIP Vision Input Switch", "CLIP Vision Input Switch",
"CLIPSeg Batch Masking", "CLIPSeg Batch Masking",
"CLIPSeg Masking", "CLIPSeg Masking",
"CLIPTextEncode (BlenderNeko Advanced + NSP)",
"CLIPTextEncode (NSP)", "CLIPTextEncode (NSP)",
"Cache Node", "Cache Node",
"Checkpoint Loader", "Checkpoint Loader",
@ -603,6 +604,10 @@
"Single Text String", "Single Text String",
"String to ASCII" "String to ASCII"
], ],
"https://github.com/city96/ComfyUI_NetDist": [
"FetchRemote",
"QueueRemote"
],
"https://github.com/comfyanonymous/ComfyUI_experiments/raw/master/sampler_rescalecfg.py": [ "https://github.com/comfyanonymous/ComfyUI_experiments/raw/master/sampler_rescalecfg.py": [
"RescaleClassifierFreeGuidanceTest" "RescaleClassifierFreeGuidanceTest"
], ],

10
scanner.py

@ -27,8 +27,16 @@ def scan_in_file(filename):
for key, value in class_dict.items(): for key, value in class_dict.items():
nodes.add(key) nodes.add(key)
return nodes update_pattern = r"NODE_CLASS_MAPPINGS.update\s*\({([^}]*)}\)"
update_match = re.search(update_pattern, code)
if update_match:
update_dict_text = update_match.group(1)
update_key_value_pairs = re.findall(r"\"([^\"]*)\"\s*:\s*([^,\n]*)", update_dict_text)
for key, value in update_key_value_pairs:
class_dict[key] = value.strip()
nodes.add(key)
return nodes
def get_py_file_paths(dirname): def get_py_file_paths(dirname):
file_paths = [] file_paths = []

Loading…
Cancel
Save