mirror of
https://github.com/Comfy-Org/ComfyUI-Manager.git
synced 2025-12-16 01:57:04 +08:00
update DB
robust scanner.py
This commit is contained in:
parent
f65e13ba36
commit
6b7dd1b82f
@ -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
10
scanner.py
@ -26,10 +26,18 @@ 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)
|
||||||
|
|
||||||
|
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
|
return nodes
|
||||||
|
|
||||||
|
|
||||||
def get_py_file_paths(dirname):
|
def get_py_file_paths(dirname):
|
||||||
file_paths = []
|
file_paths = []
|
||||||
|
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user