feat: message collapse register

update DB
This commit is contained in:
Dr.Lt.Data 2023-10-15 17:45:30 +09:00
parent 7255f89dfb
commit 5d00f47eef
4 changed files with 22 additions and 1 deletions

View File

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

View File

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

View File

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

View File

@ -6,6 +6,18 @@ import atexit
import threading import threading
import re 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: try:
if '--port' in sys.argv: if '--port' in sys.argv:
port_index = sys.argv.index('--port') port_index = sys.argv.index('--port')
@ -46,6 +58,9 @@ try:
raise ValueError("The object does not have a fileno method") raise ValueError("The object does not have a fileno method")
def write(self, message): def write(self, message):
if any(f(message) for f in message_collapses):
return
if not self.is_stdout: if not self.is_stdout:
match = re.search(tqdm, message) match = re.search(tqdm, message)
if match: if match: