File size: 802 Bytes
80abb6e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 | import logging
import sys
class ColorCode:
RESET = "\033[0m"
BLACK = "\033[0;90m"
CYAN = "\033[0;36m"
YELLOW = "\033[0;33m"
RED = "\033[0;31m"
MAP = {
"DEBUG": BLACK,
"INFO": CYAN,
"WARNING": YELLOW,
"ERROR": RED,
}
class ColoredFormatter(logging.Formatter):
def format(self, record):
levelname = record.levelname
record.levelname = f"{ColorCode.MAP[levelname]}{levelname}{ColorCode.RESET}"
return super().format(record)
logger = logging.getLogger("ChunkWeight")
logger.setLevel(logging.INFO)
logger.propagate = False
if not logger.handlers:
handler = logging.StreamHandler(sys.stdout)
handler.setFormatter(ColoredFormatter("[%(name)s] %(levelname)s - %(message)s"))
logger.addHandler(handler)
|