31 lines
950 B
Python
31 lines
950 B
Python
""" Formatter for the logging module, coloring terminal output according to error criticity. """
|
|
|
|
import enum
|
|
import logging
|
|
import sys
|
|
|
|
|
|
Colors = enum.Enum("Colors", ("RED", "GREEN", "YELLOW", "BLUE"))
|
|
|
|
LEVEL_COLOR_MAPPING = {logging.WARNING: Colors.YELLOW,
|
|
logging.ERROR: Colors.RED,
|
|
logging.CRITICAL: Colors.RED}
|
|
LEVEL_BOLD_MAPPING = {logging.WARNING: False,
|
|
logging.ERROR: False,
|
|
logging.CRITICAL: True}
|
|
|
|
|
|
class ColoredFormatter(logging.Formatter):
|
|
|
|
def format(self, record):
|
|
message = super().format(record)
|
|
if sys.stderr.isatty() and not sys.platform.startswith("win32"):
|
|
try:
|
|
color_code = LEVEL_COLOR_MAPPING[record.levelno].value
|
|
bold = LEVEL_BOLD_MAPPING[record.levelno]
|
|
except KeyError:
|
|
pass
|
|
else:
|
|
message = "\033[%u;%um%s\033[0m" % (int(bold), 30 + color_code, message)
|
|
return message
|