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
# ensure .js
print("### Loading: ComfyUI-Manager (V0.34)")
print("### Loading: ComfyUI-Manager (V0.35)")
comfy_ui_required_revision = 1240
comfy_ui_revision = "Unknown"

View File

@ -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",

View File

@ -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",

View File

@ -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: