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

improve: logging supports multi port

Browse Source
This commit is contained in:
dr.lt.data 2023-10-04 13:02:28 +09:00
parent 58dd5ae386
commit dee6cd8969
2 changed files with 16 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
__init__.py
View File

@ -56,7 +56,7 @@ sys.path.append('../..')
from torchvision.datasets.utils import download_url from torchvision.datasets.utils import download_url
# ensure .js # ensure .js
print("### Loading: ComfyUI-Manager (V0.32)") print("### Loading: ComfyUI-Manager (V0.33)")
comfy_ui_required_revision = 1240 comfy_ui_required_revision = 1240
comfy_ui_revision = "Unknown" comfy_ui_revision = "Unknown"

22
prestartup_script.py
View File

@ -7,20 +7,28 @@ import threading
import re import re
try: try:
if '--port' in sys.argv:
port_index = sys.argv.index('--port')
if port_index + 1 < len(sys.argv):
port = int(sys.argv[port_index + 1])
postfix = f"_{port}"
else:
postfix = ""
# Logger setup # Logger setup
if os.path.exists("comfyui.log"): if os.path.exists(f"comfyui{postfix}.log"):
if os.path.exists("comfyui.prev.log"): if os.path.exists(f"comfyui{postfix}.prev.log"):
if os.path.exists("comfyui.prev2.log"): if os.path.exists(f"comfyui{postfix}.prev2.log"):
os.remove("comfyui.prev2.log") os.remove(f"comfyui{postfix}.prev2.log")
os.rename("comfyui.prev.log", "comfyui.prev2.log") os.rename(f"comfyui{postfix}.prev.log", f"comfyui{postfix}.prev2.log")
os.rename("comfyui.log", "comfyui.prev.log") os.rename(f"comfyui{postfix}.log", f"comfyui{postfix}.prev.log")
original_stdout = sys.stdout original_stdout = sys.stdout
original_stderr = sys.stderr original_stderr = sys.stderr
tqdm = r'\d+%.*\[(.*?)\]' tqdm = r'\d+%.*\[(.*?)\]'
log_file = open("comfyui.log", "w", encoding="utf-8") log_file = open(f"comfyui{postfix}.log", "w", encoding="utf-8")
log_lock = threading.Lock() log_lock = threading.Lock()
class Logger: class Logger: