wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-05-28 01:47:32 +08:00
Code Issues Actions 28 Packages Projects Releases Wiki Activity

Add support to coloredFormatter to accept extra={color: red}, color Prompt executed in ... green
Some checks are pending
Python Linting / Run Ruff (push) Waiting to run
Details
Python Linting / Run Pylint (push) Waiting to run
Details

Browse Source
This commit is contained in:
Talmaj Marinc 2026-05-23 23:52:14 +02:00
parent 536dc26c53
commit 3a5afc101e
2 changed files with 25 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
app/logger.py
View File

@ -5,13 +5,25 @@ import logging
import sys import sys
import threading import threading
ANSI_LEVEL_COLORS = { ANSI_NAMED_COLORS = {
'DEBUG': '\033[36m', # cyan 'black': '\033[30m',
'INFO': '\033[32m', # green 'red': '\033[31m',
'WARNING': '\033[33m', # yellow 'green': '\033[32m',
'ERROR': '\033[31m', # red 'yellow': '\033[33m',
'CRITICAL': '\033[35m', # magenta 'blue': '\033[34m',
'magenta': '\033[35m',
'cyan': '\033[36m',
'white': '\033[37m',
} }
ANSI_LEVEL_COLORS = {
'DEBUG': ANSI_NAMED_COLORS['cyan'],
'INFO': ANSI_NAMED_COLORS['green'],
'WARNING': ANSI_NAMED_COLORS['yellow'],
'ERROR': ANSI_NAMED_COLORS['red'],
'CRITICAL': ANSI_NAMED_COLORS['magenta'],
}
ANSI_RESET = '\033[0m' ANSI_RESET = '\033[0m'
ANSI_BOLD = '\033[1m' ANSI_BOLD = '\033[1m'
@ -21,7 +33,11 @@ class ColoredFormatter(logging.Formatter):
color = ANSI_LEVEL_COLORS.get(record.levelname, '') color = ANSI_LEVEL_COLORS.get(record.levelname, '')
bold = ANSI_BOLD if record.levelno >= logging.WARNING else '' bold = ANSI_BOLD if record.levelno >= logging.WARNING else ''
level_tag = f"{bold}{color}[{record.levelname}]{ANSI_RESET} " level_tag = f"{bold}{color}[{record.levelname}]{ANSI_RESET} "
return level_tag + super().format(record) message = super().format(record)
line_color = ANSI_NAMED_COLORS.get(getattr(record, 'color', ''), '')
if line_color:
return f"{level_tag}{line_color}{message}{ANSI_RESET}"
return level_tag + message
logs = None logs = None
stdout_interceptor = None stdout_interceptor = None

4
main.py
View File

@ -344,9 +344,9 @@ def prompt_worker(q, server_instance):
# Log Time in a more readable way after 10 minutes # Log Time in a more readable way after 10 minutes
if execution_time > 600: if execution_time > 600:
execution_time = time.strftime("%H:%M:%S", time.gmtime(execution_time)) execution_time = time.strftime("%H:%M:%S", time.gmtime(execution_time))
logging.info(f"Prompt executed in {execution_time}") logging.info(f"Prompt executed in {execution_time}", extra={'color': 'green'})
else: else:
logging.info("Prompt executed in {:.2f} seconds".format(execution_time)) logging.info("Prompt executed in {:.2f} seconds".format(execution_time), extra={'color': 'green'})
if not asset_seeder.is_disabled(): if not asset_seeder.is_disabled():
paths = _collect_output_absolute_paths(e.history_result) paths = _collect_output_absolute_paths(e.history_result)