From bf25d0208a63d947f5e3ec70296e6abd3220cdc9 Mon Sep 17 00:00:00 2001
From: "dr.lt.data" <Dr.Lt.Data@gmail.com>
Date: Thu, 4 Jan 2024 15:49:47 +0900
Subject: [PATCH] feat: add timestamp to log file

---
 __init__.py          | 4 ++--
 prestartup_script.py | 8 +++++++-
 2 files changed, 9 insertions(+), 3 deletions(-)

diff --git a/__init__.py b/__init__.py
index 0e6a02e..439bfd5 100644
--- a/__init__.py
+++ b/__init__.py
@@ -27,7 +27,7 @@ except:
     print(f"[WARN] ComfyUI-Manager: Your ComfyUI version is outdated. Please update to the latest version.")
 
 
-version = [1, 21]
+version = [1, 23]
 version_str = f"V{version[0]}.{version[1]}" + (f'.{version[2]}' if len(version) > 2 else '')
 print(f"### Loading: ComfyUI-Manager ({version_str})")
 
@@ -35,9 +35,9 @@ print(f"### Loading: ComfyUI-Manager ({version_str})")
 required_comfyui_revision = 1793
 comfy_ui_hash = "-"
 
-
 cache_lock = threading.Lock()
 
+
 def handle_stream(stream, prefix):
     stream.reconfigure(encoding=locale.getpreferredencoding(), errors='replace')
     for msg in stream:
diff --git a/prestartup_script.py b/prestartup_script.py
index 5869237..f06543d 100644
--- a/prestartup_script.py
+++ b/prestartup_script.py
@@ -142,6 +142,7 @@ try:
         def __init__(self, is_stdout):
             self.is_stdout = is_stdout
             self.encoding = "utf-8"
+            self.last_char = ''
 
         def fileno(self):
             try:
@@ -193,8 +194,13 @@ try:
 
         def sync_write(self, message):
             with log_lock:
-                log_file.write(message)
+                timestamp = datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')[:-3]
+                if self.last_char != '\n':
+                    log_file.write(message)
+                else:
+                    log_file.write(f"[{timestamp}] {message}")
                 log_file.flush()
+                self.last_char = message[-1]
 
             with std_log_lock:
                 if self.is_stdout: