From dee6cd89692ee9b8253c56ba7386b4a5a84fffd8 Mon Sep 17 00:00:00 2001
From: "dr.lt.data" <Dr.Lt.Data@gmail.com>
Date: Wed, 4 Oct 2023 13:02:28 +0900
Subject: [PATCH] improve: logging supports multi port

---
 __init__.py          |  2 +-
 prestartup_script.py | 22 +++++++++++++++-------
 2 files changed, 16 insertions(+), 8 deletions(-)

diff --git a/__init__.py b/__init__.py
index 2d5c183c..bb50b24f 100644
--- a/__init__.py
+++ b/__init__.py
@@ -56,7 +56,7 @@ sys.path.append('../..')
 from torchvision.datasets.utils import download_url
 
 # ensure .js
-print("### Loading: ComfyUI-Manager (V0.32)")
+print("### Loading: ComfyUI-Manager (V0.33)")
 
 comfy_ui_required_revision = 1240
 comfy_ui_revision = "Unknown"
diff --git a/prestartup_script.py b/prestartup_script.py
index 9eed64c6..f1d2acfd 100644
--- a/prestartup_script.py
+++ b/prestartup_script.py
@@ -7,20 +7,28 @@ import threading
 import re
 
 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
-    if os.path.exists("comfyui.log"):
-        if os.path.exists("comfyui.prev.log"):
-            if os.path.exists("comfyui.prev2.log"):
-                os.remove("comfyui.prev2.log")
-            os.rename("comfyui.prev.log", "comfyui.prev2.log")
-        os.rename("comfyui.log", "comfyui.prev.log")
+    if os.path.exists(f"comfyui{postfix}.log"):
+        if os.path.exists(f"comfyui{postfix}.prev.log"):
+            if os.path.exists(f"comfyui{postfix}.prev2.log"):
+                os.remove(f"comfyui{postfix}.prev2.log")
+            os.rename(f"comfyui{postfix}.prev.log", f"comfyui{postfix}.prev2.log")
+        os.rename(f"comfyui{postfix}.log", f"comfyui{postfix}.prev.log")
 
     original_stdout = sys.stdout
     original_stderr = sys.stderr
 
     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()
 
     class Logger: