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

fix: exception handling for handle_stream

Browse Source
This commit is contained in:
Dr.Lt.Data 2023-11-05 00:44:52 +09:00
parent 59a2ce177d
commit 458f1f6817
2 changed files with 19 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
__init__.py
View File

@ -9,9 +9,13 @@ import re
from tqdm.auto import tqdm
from git.remote import RemoteProgress
def handle_stream(stream, prefix):
for line in stream:
print(prefix, line, end="")
try:
for line in stream:
print(prefix, line, end="")
except Exception:
print("[!] ??? log decoding error ???")
def run_script(cmd, cwd='.'):
@ -58,7 +62,7 @@ sys.path.append('../..')
from torchvision.datasets.utils import download_url
# ensure .js
print("### Loading: ComfyUI-Manager (V0.38)")
print("### Loading: ComfyUI-Manager (V0.38.1)")
comfy_ui_required_revision = 1240
comfy_ui_revision = "Unknown"

18
prestartup_script.py
View File

@ -25,8 +25,11 @@ git_script_path = os.path.join(comfyui_manager_path, "git_helper.py")
def handle_stream(stream, prefix):
for msg in stream:
print(prefix, msg, end="")
try:
for msg in stream:
print(prefix, msg, end="")
except Exception:
print("[!] ??? log decoding error ???")
def process_wrap(cmd_str, cwd_path, handler=None):
@ -147,11 +150,14 @@ if os.path.exists(restore_snapshot_path):
cloned_repos = []
def msg_capture(stream, prefix):
for msg in stream:
if msg.startswith("CLONE: "):
cloned_repos.append(msg[7:])
try:
for msg in stream:
if msg.startswith("CLONE: "):
cloned_repos.append(msg[7:])
print(prefix, msg, end="")
print(prefix, msg, end="")
except Exception:
print("[!] [snapshot restore] ??? log decoding error ???")
print(f"[ComfyUI-Manager] Restore snapshot.")
cmd_str = [sys.executable, git_script_path, '--apply-snapshot', restore_snapshot_path]