浏览代码

Fix: Add idiot check to previous commit and fix Get_Socket_Maps

Joseph Brandenburg 6 月之前
父节点
当前提交
b0c7895b95
共有 2 个文件被更改,包括 3 次插入3 次删除
  1. 2 2
      base_definitions.py
  2. 1 1
      utilities.py

+ 2 - 2
base_definitions.py

@@ -374,7 +374,7 @@ def node_group_update(node, force = False):
             for i, socket in enumerate(node.inputs):
                 if socket.identifier == "Schema Length" and i == 0:
                     continue
-                elif socket.identifier in socket_map_in.keys():
+                elif (socket_map_in is None) or socket.identifier in socket_map_in.keys():
                     remove_me.append(socket)
             while remove_me:
                 node.inputs.remove(remove_me.pop())
@@ -382,7 +382,7 @@ def node_group_update(node, force = False):
         if update_output:
             remove_me=[]
             for socket in node.outputs:
-                if socket.identifier in socket_map_out.keys():
+                if (socket_map_out is None) or socket.identifier in socket_map_out.keys():
                     remove_me.append(socket)
             while remove_me:
                 node.inputs.remove(remove_me.pop())

+ 1 - 1
utilities.py

@@ -101,7 +101,7 @@ def get_socket_maps(node, force=False):
                     if val is None:
                         raise RuntimeError(f"ERROR: Could not get socket data for socket of type: {sock.bl_idname}")
                 else:
-                    if not force: return
+                    if not force: continue
                 map[sock.identifier]=val
             else:
                 from .socket_definitions import no_default_value