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

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