nodes/LAOGOU-666-ComfyUI-LG_HotReload
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix_bugs

Browse Source
This commit is contained in:
LAOGOU-666
2025-03-02 00:33:28 +08:00
parent e2c6998ce1
commit 35136e5ae7
2 changed files with 9 additions and 43 deletions
Download Patch File Download Diff File Expand all files Collapse all files
__init__.py
+4 -14
View File
@@ -260,12 +260,7 @@ class DebouncedHotReloader(FileSystemEventHandler):
# 重新加载自定义节点
load_custom_node(module_path)
print(f'\033[92m[LG_HotReload] Successfully reloaded module: {module_name}\033[0m')
if registered_routes:
print(f'\033[92m[LG_HotReload] Registered routes: {len(registered_routes)}\033[0m')
print(f'\033[92m[LG_HotReload] Loaded nodes: {list(getattr(module, "NODE_CLASS_MAPPINGS", {}).keys())}\033[0m')
return web.Response(text='OK')
except Exception as e:
@@ -421,14 +416,7 @@ class DebouncedHotReloader(FileSystemEventHandler):
removed_nodes = set(route['handler'] for route in old_routes) - new_nodes
updated_nodes = new_nodes & set(route['handler'] for route in old_routes)
action = "deleted" if not os.path.exists(file_path) else "added" if file_path not in self.__hashes else "modified"
print(f'\033[92m[LG_HotReload] Reloaded module: {module_name}\033[0m')
if added_nodes or removed_nodes or updated_nodes:
if added_nodes:
print(f'\033[92m[LG_HotReload] Added nodes: {added_nodes}\033[0m')
if removed_nodes:
print(f'\033[92m[LG_HotReload] Removed nodes: {removed_nodes}\033[0m')
if updated_nodes:
print(f'\033[92m[LG_HotReload] Updated nodes: {updated_nodes}\033[0m')
update_message = {
"type": "hot_reload_update",
"data": {
@@ -457,6 +445,8 @@ class DebouncedHotReloader(FileSystemEventHandler):
web_dir = os.path.join(module_path, module.WEB_DIRECTORY)
if os.path.exists(web_dir) and os.path.isdir(web_dir):
self.copy_web_files(module_name, web_dir)
print(f'\033[92m[LG_HotReload] Successfully reloaded module: {module_name}\033[0m')
except requests.RequestException as e:
print(f'\033[91m[LG_HotReload] Reload failed: {e}\033[0m')
except Exception as e:
web/hot_reload.js
+5 -29
View File
@@ -23,26 +23,12 @@ app.registerExtension({
acc[w.name] = w.serializeValue ? w.serializeValue() : w.value;
return acc;
}, {}),
properties: {...node.properties},
inputs: node.inputs?.map(i => ({...i})),
outputs: node.outputs?.map(o => ({...o})),
connections: {
inputs: node.inputs?.map((input, slot) =>
input.link != null ? {
slot,
target_id: app.graph.links[input.link].origin_id,
target_slot: app.graph.links[input.link].origin_slot
} : null).filter(x => x),
outputs: node.outputs?.map((output, slot) =>
output.links?.map(link => ({
slot,
target_id: app.graph.links[link].target_id,
target_slot: app.graph.links[link].target_slot
}))).flat().filter(x => x)
}
properties: {...node.properties}
});
});
}
// 更新节点定义
for (const nodeClass of nodesToUpdate) {
const response = await api.fetchApi(`/object_info/${nodeClass}`);
const nodeData = await response.json();
@@ -62,6 +48,8 @@ app.registerExtension({
});
}
}
// 恢复节点状态，但不处理连接
for (const state of savedStates.values()) {
const node = app.graph.getNodeById(state.id);
if (node) {
@@ -79,18 +67,6 @@ app.registerExtension({
}
});
}
state.connections.inputs.forEach(conn => {
const targetNode = app.graph.getNodeById(conn.target_id);
if (targetNode) {
node.connect(conn.slot, targetNode, conn.target_slot);
}
});
state.connections.outputs.forEach(conn => {
const targetNode = app.graph.getNodeById(conn.target_id);
if (targetNode) {
node.connect(conn.slot, targetNode, conn.target_slot);
}
});
}
}
app.graph.setDirtyCanvas(true);