xinyun/ComfyUI-Manager
1
0
Fork 0
mirror of https://git.datalinker.icu/ltdrdata/ComfyUI-Manager synced 2025-12-10 06:34:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: drawbadge crashes to several extension

Browse Source
This commit is contained in:
Dr.Lt.Data 2023-12-09 10:00:18 +09:00
parent 92ac9ef51f
commit 71fcc35acd
2 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
__init__.py
View File

@ -18,7 +18,7 @@ import re
import signal import signal
import nodes import nodes
version = "V1.10.2" version = "V1.10.3"
print(f"### Loading: ComfyUI-Manager ({version})") print(f"### Loading: ComfyUI-Manager ({version})")
required_comfyui_revision = 1793 required_comfyui_revision = 1793

11
js/comfyui-manager.js
View File

@ -279,8 +279,9 @@ function getNickname(node, nodename) {
} }
} }
function drawBadge(node, orig, ctx) { function drawBadge(node, orig, restArgs) {
const r = orig?.apply?.(node, arguments); let ctx = restArgs[0];
const r = orig?.apply?.(node, restArgs);
if (!node.flags.collapsed && badge_mode != 'none' && node.constructor.title_mode != LiteGraph.NO_TITLE) { if (!node.flags.collapsed && badge_mode != 'none' && node.constructor.title_mode != LiteGraph.NO_TITLE) {
let text = ""; let text = "";
@ -953,7 +954,9 @@ app.registerExtension({
if(!node.badge_enabled) { if(!node.badge_enabled) {
node.getNickname = function () { return getNickname(node, node.comfyClass.trim()) }; node.getNickname = function () { return getNickname(node, node.comfyClass.trim()) };
const orig = node.__proto__.onDrawForeground; const orig = node.__proto__.onDrawForeground;
node.onDrawForeground = function (ctx) { drawBadge(node, orig, ctx) }; node.onDrawForeground = function (ctx) {
drawBadge(node, orig, arguments)
};
node.badge_enabled = true; node.badge_enabled = true;
} }
}, },
@ -961,7 +964,7 @@ app.registerExtension({
if(!node.badge_enabled) { if(!node.badge_enabled) {
const orig = node.onDrawForeground; const orig = node.onDrawForeground;
node.getNickname = function () { return getNickname(node, node.type.trim()) }; node.getNickname = function () { return getNickname(node, node.type.trim()) };
node.onDrawForeground = function (ctx) { drawBadge(node, orig, ctx) }; node.onDrawForeground = function (ctx) { drawBadge(node, orig, arguments) };
} }
}, },