118 lines
4.5 KiB
Python
118 lines
4.5 KiB
Python
from server import PromptServer # type:ignore
|
|
import folder_paths # type:ignore
|
|
import os
|
|
import threading
|
|
import time
|
|
import datetime
|
|
|
|
|
|
CATEGORY = "HotReload"
|
|
|
|
|
|
# ------------ Terminal ------------
|
|
class HotReload_Terminal:
|
|
|
|
CATEGORY = CATEGORY
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(s):
|
|
return {"optional": {}}
|
|
|
|
RETURN_TYPES = ()
|
|
FUNCTION = "Func"
|
|
|
|
def Func(self):
|
|
pass
|
|
|
|
|
|
# file terminal checker
|
|
CHECKER_THREAD_NAME = "HotReload.LogFileChecker"
|
|
|
|
|
|
# func checker
|
|
def LogFileChecker(selfStamp):
|
|
|
|
CheckedLine = 0
|
|
CheckedDateSec = 0.0
|
|
|
|
while True:
|
|
try:
|
|
|
|
# check for stop
|
|
for thread in threading.enumerate():
|
|
if thread.name == CHECKER_THREAD_NAME:
|
|
stamp = thread.__dict__.get("stamp")
|
|
if stamp > selfStamp:
|
|
return
|
|
|
|
# check file
|
|
port = PromptServer.instance.port if hasattr(PromptServer.instance, "port") else 8000
|
|
|
|
logFilePath = os.path.join(folder_paths.get_user_directory(), f"comfyui_{port}.log")
|
|
if os.path.isfile(logFilePath):
|
|
fileDateSec = os.path.getmtime(logFilePath)
|
|
if fileDateSec > CheckedDateSec + 0.1:
|
|
requareClear = CheckedLine == 0
|
|
CheckedDateSec = fileDateSec
|
|
resultText = ""
|
|
try:
|
|
# Try UTF-8 encoding first
|
|
with open(logFilePath, "r", encoding="utf-8") as file:
|
|
currentLine = 0
|
|
for line in file:
|
|
if currentLine > CheckedLine:
|
|
CheckedLine = currentLine
|
|
# 处理时间格式，移除毫秒部分
|
|
if line.startswith("[20"):
|
|
try:
|
|
time_end = line.index("]")
|
|
if time_end > 0:
|
|
time_str = line[1:time_end]
|
|
if "." in time_str:
|
|
time_str = time_str.split(".")[0]
|
|
line = "[" + time_str + "]" + line[time_end+1:]
|
|
except:
|
|
pass
|
|
resultText += line + "\n"
|
|
currentLine += 1
|
|
except UnicodeDecodeError:
|
|
# If UTF-8 fails, try with error handling
|
|
with open(logFilePath, "r", encoding="utf-8", errors="replace") as file:
|
|
currentLine = 0
|
|
for line in file:
|
|
if currentLine > CheckedLine:
|
|
CheckedLine = currentLine
|
|
# 处理时间格式，移除毫秒部分
|
|
if line.startswith("[20"):
|
|
try:
|
|
time_end = line.index("]")
|
|
if time_end > 0:
|
|
time_str = line[1:time_end]
|
|
if "." in time_str:
|
|
time_str = time_str.split(".")[0]
|
|
line = "[" + time_str + "]" + line[time_end+1:]
|
|
except:
|
|
pass
|
|
resultText += line + "\n"
|
|
currentLine += 1
|
|
PromptServer.instance.send_sync(
|
|
"/hotreload.terminal.log", {"text": resultText, "clear": requareClear}
|
|
)
|
|
finally:
|
|
# loop
|
|
time.sleep(1)
|
|
|
|
|
|
# get current largest stamp
|
|
currentMaxStamp = 0
|
|
for thread in threading.enumerate():
|
|
if thread.name == CHECKER_THREAD_NAME:
|
|
stamp = thread.__dict__.get("stamp")
|
|
currentMaxStamp = max(currentMaxStamp, stamp)
|
|
|
|
# start new checker
|
|
timestamp = max(int(datetime.datetime.now().timestamp()), currentMaxStamp + 1)
|
|
newThread = threading.Thread(target=LogFileChecker, name=CHECKER_THREAD_NAME, args=(timestamp,))
|
|
newThread.__dict__["stamp"] = timestamp
|
|
newThread.start()
|