mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2025-01-11 02:15:17 +00:00
e91662e784
* Add route for getting output logs * Include ComfyUI version * Move to own function * Changed to memory logger * Unify logger setup logic * Fix get version git fallback --------- Co-authored-by: pythongosssss <125205205+pythongosssss@users.noreply.github.com>
32 lines
929 B
Python
32 lines
929 B
Python
import logging
|
|
from logging.handlers import MemoryHandler
|
|
from collections import deque
|
|
|
|
logs = None
|
|
formatter = logging.Formatter("%(asctime)s - %(name)s - %(levelname)s - %(message)s")
|
|
|
|
|
|
def get_logs():
|
|
return "\n".join([formatter.format(x) for x in logs])
|
|
|
|
|
|
def setup_logger(verbose: bool = False, capacity: int = 300):
|
|
global logs
|
|
if logs:
|
|
return
|
|
|
|
# Setup default global logger
|
|
logger = logging.getLogger()
|
|
logger.setLevel(logging.DEBUG if verbose else logging.INFO)
|
|
|
|
stream_handler = logging.StreamHandler()
|
|
stream_handler.setFormatter(logging.Formatter("%(message)s"))
|
|
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)
|