89 lines
2.4 KiB
Python
89 lines
2.4 KiB
Python
# https://stackoverflow.com/questions/4842424/list-of-ansi-color-escape-sequences
|
|
# https://en.wikipedia.org/wiki/ANSI_escape_code#3-bit_and_4-bit
|
|
COLORS_FG = {
|
|
'BLACK': '\33[30m',
|
|
'RED': '\33[31m',
|
|
'GREEN': '\33[32m',
|
|
'YELLOW': '\33[33m',
|
|
'BLUE': '\33[34m',
|
|
'MAGENTA': '\33[35m',
|
|
'CYAN': '\33[36m',
|
|
'WHITE': '\33[37m',
|
|
'GREY': '\33[90m',
|
|
'BRIGHT_RED': '\33[91m',
|
|
'BRIGHT_GREEN': '\33[92m',
|
|
'BRIGHT_YELLOW': '\33[93m',
|
|
'BRIGHT_BLUE': '\33[94m',
|
|
'BRIGHT_MAGENTA': '\33[95m',
|
|
'BRIGHT_CYAN': '\33[96m',
|
|
'BRIGHT_WHITE': '\33[97m',
|
|
}
|
|
COLORS_STYLE = {
|
|
'RESET': '\33[0m',
|
|
'BOLD': '\33[1m',
|
|
'NORMAL': '\33[22m',
|
|
'ITALIC': '\33[3m',
|
|
'UNDERLINE': '\33[4m',
|
|
'BLINK': '\33[5m',
|
|
'BLINK2': '\33[6m',
|
|
'SELECTED': '\33[7m',
|
|
}
|
|
COLORS_BG = {
|
|
'BLACK': '\33[40m',
|
|
'RED': '\33[41m',
|
|
'GREEN': '\33[42m',
|
|
'YELLOW': '\33[43m',
|
|
'BLUE': '\33[44m',
|
|
'MAGENTA': '\33[45m',
|
|
'CYAN': '\33[46m',
|
|
'WHITE': '\33[47m',
|
|
'GREY': '\33[100m',
|
|
'BRIGHT_RED': '\33[101m',
|
|
'BRIGHT_GREEN': '\33[102m',
|
|
'BRIGHT_YELLOW': '\33[103m',
|
|
'BRIGHT_BLUE': '\33[104m',
|
|
'BRIGHT_MAGENTA': '\33[105m',
|
|
'BRIGHT_CYAN': '\33[106m',
|
|
'BRIGHT_WHITE': '\33[107m',
|
|
}
|
|
|
|
|
|
def log_welcome(num_nodes=None):
|
|
"""Logs the welcome message."""
|
|
msg = f"{COLORS_FG['GREEN']}{COLORS_STYLE['BOLD']}rgthree\'s comfy nodes:"
|
|
msg += f"{COLORS_STYLE['RESET']}{COLORS_STYLE['BOLD']} Loaded"
|
|
if num_nodes:
|
|
print(f"{msg} {num_nodes} exciting nodes.{COLORS_STYLE['RESET']}")
|
|
else:
|
|
print(f"{msg}.{COLORS_STYLE['RESET']}")
|
|
|
|
|
|
def log_node_success(node_name, message):
|
|
"""Logs a success message."""
|
|
_log_node(COLORS_FG["GREEN"], node_name, message, prefix='✓ ')
|
|
|
|
|
|
def log_node_info(node_name, message):
|
|
"""Logs an info message."""
|
|
_log_node(COLORS_FG["CYAN"], node_name, message, prefix='🛈 ')
|
|
|
|
|
|
def log_node_warn(node_name, message):
|
|
"""Logs an warn message."""
|
|
_log_node(COLORS_FG["YELLOW"], node_name, message, prefix='⚠ ')
|
|
|
|
|
|
def log_node(node_name, message):
|
|
"""Logs a message."""
|
|
_log_node(COLORS_FG["CYAN"], node_name, message, prefix=' ')
|
|
|
|
|
|
def _log_node(color, node_name, message, prefix=''):
|
|
print(_get_log_msg(color, node_name, message, prefix=prefix))
|
|
|
|
|
|
def _get_log_msg(color, node_name, message, prefix=''):
|
|
msg = f'{COLORS_STYLE["BOLD"]}{color}{prefix}rgthree {node_name.replace(" (rgthree)", "")}'
|
|
msg += f':{COLORS_STYLE["RESET"]} {message}'
|
|
return msg
|