xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-09-10 03:17:06 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update logger.py

Browse Source
This commit is contained in:
John D. Pope 2025-04-07 21:52:52 +10:00 committed by GitHub
parent 70d7242e57
commit 25664388b3
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 31 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
app/logger.py
View File

@ -4,11 +4,36 @@ import io
import logging import logging
import sys import sys
import threading import threading
from rich.traceback import install
from rich.console import Console
from rich.theme import Theme
from rich.logging import RichHandler
logs = None logs = None
stdout_interceptor = None stdout_interceptor = None
stderr_interceptor = None stderr_interceptor = None
from rich.traceback import install
# Enable rich tracebacks globally
install()
# Configure rich console
console = Console(force_terminal=True)
# Set up handler
rich_handler = RichHandler(console=console, rich_tracebacks=True, markup=True)
# file_handler = logging.FileHandler("project.log") # Log to a file
# file_handler.setLevel(log_level)
from rich.traceback import install
# Enable rich tracebacks globally
install()
class LogInterceptor(io.TextIOWrapper): class LogInterceptor(io.TextIOWrapper):
def __init__(self, stream, *args, **kwargs): def __init__(self, stream, *args, **kwargs):
@ -56,6 +81,12 @@ def setup_logger(log_level: str = 'INFO', capacity: int = 300, use_stdout: bool
if logs: if logs:
return return
logging.basicConfig(
level=log_level,
format="%(message)s",
datefmt="[%X]",
handlers=[rich_handler] #file_handler
)
# Override output streams and log to buffer # Override output streams and log to buffer
logs = deque(maxlen=capacity) logs = deque(maxlen=capacity)