Browse Source

feat: message collapse register

update DB
pull/140/head
Dr.Lt.Data 1 year ago
parent
commit
5d00f47eef
  1. 2
      __init__.py
  2. 3
      extension-node-map.json
  3. 3
      node_db/new/extension-node-map.json
  4. 15
      prestartup_script.py

2
__init__.py

@ -56,7 +56,7 @@ sys.path.append('../..')
from torchvision.datasets.utils import download_url
# ensure .js
print("### Loading: ComfyUI-Manager (V0.34)")
print("### Loading: ComfyUI-Manager (V0.35)")
comfy_ui_required_revision = 1240
comfy_ui_revision = "Unknown"

3
extension-node-map.json

@ -2754,6 +2754,8 @@
"ImpactMakeImageList",
"ImpactMinMax",
"ImpactNeg",
"ImpactNodeSetMuteState",
"ImpactQueueTrigger",
"ImpactSEGSConcat",
"ImpactSEGSLabelFilter",
"ImpactSEGSOrderedFilter",
@ -2762,6 +2764,7 @@
"ImpactSEGSToMaskBatch",
"ImpactSEGSToMaskList",
"ImpactScaleBy_BBOX_SEG_ELT",
"ImpactSetWidgetValue",
"ImpactSimpleDetectorSEGS",
"ImpactSimpleDetectorSEGSPipe",
"ImpactStringSelector",

3
node_db/new/extension-node-map.json

@ -2754,6 +2754,8 @@
"ImpactMakeImageList",
"ImpactMinMax",
"ImpactNeg",
"ImpactNodeSetMuteState",
"ImpactQueueTrigger",
"ImpactSEGSConcat",
"ImpactSEGSLabelFilter",
"ImpactSEGSOrderedFilter",
@ -2762,6 +2764,7 @@
"ImpactSEGSToMaskBatch",
"ImpactSEGSToMaskList",
"ImpactScaleBy_BBOX_SEG_ELT",
"ImpactSetWidgetValue",
"ImpactSimpleDetectorSEGS",
"ImpactSimpleDetectorSEGSPipe",
"ImpactStringSelector",

15
prestartup_script.py

@ -6,6 +6,18 @@ import atexit
import threading
import re
message_collapses = []
def register_message_collapse(f):
global message_collapses
message_collapses.append(f)
sys.__comfyui_manager_register_message_collapse = register_message_collapse
try:
if '--port' in sys.argv:
port_index = sys.argv.index('--port')
@ -46,6 +58,9 @@ try:
raise ValueError("The object does not have a fileno method")
def write(self, message):
if any(f(message) for f in message_collapses):
return
if not self.is_stdout:
match = re.search(tqdm, message)
if match:

Loading…
Cancel
Save