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

improve: tqdm in process

Browse Source 
fix: encoding error of process output
This commit is contained in:
Dr.Lt.Data 2023-11-05 10:05:39 +09:00
parent c78c7d7e8f
commit 92ce2d706b
2 changed files with 37 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
__init__.py
View File

@ -6,15 +6,15 @@ import sys
import threading
import datetime
import re
import locale
from tqdm.auto import tqdm
version = "V0.39"
def handle_stream(stream, prefix):
try:
for line in stream:
print(prefix, line, end="")
except Exception:
print("[!] ??? log decoding error ???")
stream.reconfigure(encoding=locale.getpreferredencoding(), errors='replace')
for line in stream:
print(prefix, line, end="")
def run_script(cmd, cwd='.'):
@ -63,7 +63,7 @@ sys.path.append('../..')
from torchvision.datasets.utils import download_url
# ensure .js
print("### Loading: ComfyUI-Manager (V0.38.3)")
print(f"### Loading: ComfyUI-Manager ({version})")
comfy_ui_required_revision = 1240
comfy_ui_revision = "Unknown"

43
prestartup_script.py
View File

@ -5,6 +5,7 @@ import sys
import atexit
import threading
import re
import locale
message_collapses = []
@ -25,11 +26,18 @@ git_script_path = os.path.join(comfyui_manager_path, "git_helper.py")
def handle_stream(stream, prefix):
try:
for msg in stream:
print(prefix, msg, end="")
except Exception:
print("[!] ??? log decoding error ???")
stream.reconfigure(encoding=locale.getpreferredencoding(), errors='replace')
for msg in stream:
if prefix == '[!]' and ('it/s]' or 's/it]') in msg and ('%|' in msg or 'it [' in msg):
if msg.startswith('100%'):
print('\r' + msg, end="", file=sys.stderr),
else:
print('\r' + msg[:-1], end="", file=sys.stderr),
else:
if prefix == '[!]':
print(prefix, msg, end="", file=sys.stderr)
else:
print(prefix, msg, end="")
def process_wrap(cmd_str, cwd_path, handler=None):
@ -150,14 +158,25 @@ if os.path.exists(restore_snapshot_path):
cloned_repos = []
def msg_capture(stream, prefix):
try:
for msg in stream:
if msg.startswith("CLONE: "):
cloned_repos.append(msg[7:])
stream.reconfigure(encoding=locale.getpreferredencoding(), errors='replace')
for msg in stream:
if msg.startswith("CLONE: "):
cloned_repos.append(msg[7:])
if prefix == '[!]':
print(prefix, msg, end="", file=sys.stderr)
else:
print(prefix, msg, end="")
print(prefix, msg, end="")
except Exception:
print("[!] [snapshot restore] ??? log decoding error ???")
elif prefix == '[!]' and ('it/s]' in msg or 's/it]' in msg) and ('%|' in msg or 'it [' in msg):
if msg.startswith('100%'):
print('\r' + msg, end="", file=sys.stderr),
else:
print('\r'+msg[:-1], end="", file=sys.stderr),
else:
if prefix == '[!]':
print(prefix, msg, end="", file=sys.stderr)
else:
print(prefix, msg, end="")
print(f"[ComfyUI-Manager] Restore snapshot.")
cmd_str = [sys.executable, git_script_path, '--apply-snapshot', restore_snapshot_path]