wangbo/ComfyUI-Manager
1
0
Fork 0
mirror of https://github.com/Comfy-Org/ComfyUI-Manager.git synced 2025-12-20 11:52:59 +08:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

fix: logging hook - robust patch

Browse Source 
fix: potential crash - extracting port info
This commit is contained in:
Dr.Lt.Data 2024-07-06 21:35:58 +09:00
parent e5e71b20bb
commit 7adafa453d
3 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
glob/manager_core.py
View File

@ -23,7 +23,7 @@ sys.path.append(glob_path)
import cm_global import cm_global
from manager_util import * from manager_util import *
version = [2, 44, 2] version = [2, 44, 3]
version_str = f"V{version[0]}.{version[1]}" + (f'.{version[2]}' if len(version) > 2 else '') version_str = f"V{version[0]}.{version[1]}" + (f'.{version[2]}' if len(version) > 2 else '')

10
prestartup_script.py
View File

@ -167,6 +167,8 @@ try:
postfix = f"_{port}" postfix = f"_{port}"
else: else:
postfix = "" postfix = ""
else:
postfix = ""
# Logger setup # Logger setup
if enable_file_logging: if enable_file_logging:
@ -301,11 +303,13 @@ try:
if enable_file_logging: if enable_file_logging:
sys.stdout = ComfyUIManagerLogger(True) sys.stdout = ComfyUIManagerLogger(True)
sys.stderr = ComfyUIManagerLogger(False) stderr_wrapper = ComfyUIManagerLogger(False)
sys.stderr = stderr_wrapper
atexit.register(close_log) atexit.register(close_log)
else: else:
sys.stdout.close_log = lambda: None sys.stdout.close_log = lambda: None
stderr_wrapper = None
class LoggingHandler(logging.Handler): class LoggingHandler(logging.Handler):
@ -322,8 +326,8 @@ try:
if 'Starting server' in message: if 'Starting server' in message:
is_start_mode = False is_start_mode = False
if enable_file_logging: if stderr_wrapper:
sys.stderr.sync_write(message+'\n', file_only=True) stderr_wrapper.sync_write(message+'\n', file_only=True)
logging.getLogger().addHandler(LoggingHandler()) logging.getLogger().addHandler(LoggingHandler())

2
pyproject.toml
View File

@ -1,7 +1,7 @@
[project] [project]
name = "comfyui-manager" name = "comfyui-manager"
description = "ComfyUI-Manager provides features to install and manage custom nodes for ComfyUI, as well as various functionalities to assist with ComfyUI." description = "ComfyUI-Manager provides features to install and manage custom nodes for ComfyUI, as well as various functionalities to assist with ComfyUI."
version = "2.44.2" version = "2.44.3"
license = "LICENSE" license = "LICENSE"
dependencies = ["GitPython", "PyGithub", "matrix-client==0.4.0", "transformers", "huggingface-hub>0.20", "typer", "rich", "typing-extensions"] dependencies = ["GitPython", "PyGithub", "matrix-client==0.4.0", "transformers", "huggingface-hub>0.20", "typer", "rich", "typing-extensions"]