71 lines
2.2 KiB
Python
71 lines
2.2 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):
|
|
msg='{}{}rgthree\'s comfy nodes:{}{} Loaded'.format(COLORS_FG['GREEN'], COLORS_STYLE['BOLD'], COLORS_STYLE['RESET'], COLORS_STYLE['BOLD'])
|
|
if num_nodes:
|
|
print('{} {} exciting nodes.{}'.format(msg, num_nodes, COLORS_STYLE['RESET']))
|
|
else:
|
|
print('{}.{}'.format(msg, COLORS_STYLE['RESET']))
|
|
|
|
def log_node_success(node_name, message):
|
|
_log_node(COLORS_FG["GREEN"], node_name, message, prefix='✓ ')
|
|
|
|
def log_node_info(node_name, message):
|
|
_log_node(COLORS_FG["CYAN"], node_name, message, prefix='🛈 ')
|
|
|
|
def log_node(node_name, message):
|
|
_log_node(COLORS_FG["CYAN"], node_name, message, prefix=' ')
|
|
|
|
def log_node_warn(node_name, message):
|
|
_log_node(COLORS_FG["YELLOW"], node_name, message, prefix='⚠ ')
|
|
|
|
def _log_node(color, node_name, message, prefix=''):
|
|
print(f'{COLORS_STYLE["BOLD"]}{color}{prefix}rgthree {node_name.replace(" (rgthree)", "")}:{COLORS_STYLE["RESET"]} {message}')
|