From 6b7dd1b82f854c3628e52cae4d99a14cdc86db4f Mon Sep 17 00:00:00 2001
From: "Dr.Lt.Data" <dr.lt.data@gmail.com>
Date: Mon, 19 Jun 2023 16:23:11 +0900
Subject: [PATCH] update DB robust scanner.py

---
 extension-node-map.json |  5 +++++
 scanner.py              | 10 +++++++++-
 2 files changed, 14 insertions(+), 1 deletion(-)

diff --git a/extension-node-map.json b/extension-node-map.json
index e0e4b73..ef5257c 100644
--- a/extension-node-map.json
+++ b/extension-node-map.json
@@ -401,6 +401,7 @@
         "CLIP Vision Input Switch",
         "CLIPSeg Batch Masking",
         "CLIPSeg Masking",
+        "CLIPTextEncode (BlenderNeko Advanced + NSP)",
         "CLIPTextEncode (NSP)",
         "Cache Node",
         "Checkpoint Loader",
@@ -603,6 +604,10 @@
         "Single Text String",
         "String to ASCII"
     ],
+    "https://github.com/city96/ComfyUI_NetDist": [
+        "FetchRemote",
+        "QueueRemote"
+    ],
     "https://github.com/comfyanonymous/ComfyUI_experiments/raw/master/sampler_rescalecfg.py": [
         "RescaleClassifierFreeGuidanceTest"
     ],
diff --git a/scanner.py b/scanner.py
index 2866d28..88e4288 100644
--- a/scanner.py
+++ b/scanner.py
@@ -26,10 +26,18 @@ def scan_in_file(filename):
 
     for key, value in class_dict.items():
         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
 
-
 def get_py_file_paths(dirname):
     file_paths = []