refactor: Logger -> ComfyUIManagerLogger

fix: prestartup Logger - add close method
This commit is contained in:
Dr.Lt.Data 2023-11-21 18:10:21 +09:00
parent 1d21359a5a
commit 4feaa2d239
2 changed files with 8 additions and 4 deletions

View File

@ -14,7 +14,7 @@ import concurrent
import ssl
from urllib.parse import urlparse
version = "V1.1.2"
version = "V1.1.3"
print(f"### Loading: ComfyUI-Manager ({version})")

View File

@ -83,7 +83,7 @@ try:
log_file = open(f"comfyui{postfix}.log", "w", encoding="utf-8")
log_lock = threading.Lock()
class Logger:
class ComfyUIManagerLogger:
def __init__(self, is_stdout):
self.is_stdout = is_stdout
self.encoding = "utf-8"
@ -136,6 +136,10 @@ try:
else:
original_stderr.flush()
def close(self):
self.flush()
pass
def reconfigure(self, *args, **kwargs):
pass
@ -143,8 +147,8 @@ try:
log_file.close()
sys.stdout = Logger(True)
sys.stderr = Logger(False)
sys.stdout = ComfyUIManagerLogger(True)
sys.stderr = ComfyUIManagerLogger(False)
atexit.register(close_log)
except Exception as e: