mirror of
https://git.datalinker.icu/comfyanonymous/ComfyUI
synced 2026-09-13 11:17:08 +08:00
64 lines
1.7 KiB
Python
64 lines
1.7 KiB
Python
import logging
|
|
from logging.handlers import MemoryHandler
|
|
from collections import deque
|
|
from rich import pretty
|
|
from rich.theme import Theme
|
|
from rich.console import Console
|
|
from rich.logging import RichHandler
|
|
|
|
|
|
logs = None
|
|
LOGFMT = "%(asctime)s %(levelname)s [%(filename)s:%(lineno)s] %(message)s"
|
|
formatter = logging.Formatter(LOGFMT)
|
|
|
|
|
|
def get_logs():
|
|
return "\n".join([formatter.format(x) for x in logs])
|
|
|
|
|
|
def setup_logger(log_level: str = 'INFO', capacity: int = 300):
|
|
global logs
|
|
if logs:
|
|
return
|
|
|
|
# Setup default global logger
|
|
logger = logging.getLogger()
|
|
logger.setLevel(log_level)
|
|
|
|
stream_handler = get_rich_hander(log_level)
|
|
logger.addHandler(stream_handler)
|
|
|
|
# Create a memory handler with a deque as its buffer
|
|
logs = deque(maxlen=capacity)
|
|
memory_handler = MemoryHandler(capacity, flushLevel=logging.INFO)
|
|
memory_handler.buffer = logs
|
|
memory_handler.setFormatter(formatter)
|
|
logger.addHandler(memory_handler)
|
|
|
|
def get_rich_hander(log_level: str = "INFO") -> RichHandler:
|
|
console = Console(
|
|
log_time=True,
|
|
log_time_format="%Y-%m-%d %H:%M:%S",
|
|
theme=Theme({
|
|
"log.time": "green",
|
|
"inspect.value.border": "black",
|
|
"traceback.border": "black",
|
|
"traceback.border.syntax_error": "black",
|
|
}),
|
|
)
|
|
|
|
pretty.install(console=console)
|
|
handler = RichHandler(
|
|
console=console,
|
|
level=log_level,
|
|
markup=False,
|
|
show_time=True,
|
|
show_level=True,
|
|
show_path=True,
|
|
rich_tracebacks=True,
|
|
enable_link_path=False,
|
|
omit_repeated_times=False,
|
|
log_time_format="%Y-%m-%d %H:%M:%S",
|
|
)
|
|
return handler
|