andras/great-ai
1
0
Fork 0
Code Issues Pull requests 1 Projects Releases 4 Packages Wiki Activity Actions

Move files

Browse source
This commit is contained in:
Andras Schmelczer 2022-07-04 13:55:44 +02:00
parent d2af1b8151
commit f160b07163
159 changed files with 104 additions and 88 deletions
Download patch file Download diff file Expand all files Collapse all files

21
src/great_ai/utilities/logger/custom_formatter.py
View file

@ -1,21 +0,0 @@
import logging
from .colors import BLUE, BOLD_RED, GREY, RED, RESET, YELLOW
class CustomFormatter(logging.Formatter):
def __init__(self, fmt: str):
super().__init__()
self._fmt = fmt
self._color_mapping = {
logging.DEBUG: GREY + self._fmt + RESET,
logging.INFO: BLUE + self._fmt + RESET,
logging.WARNING: YELLOW + self._fmt + RESET,
logging.ERROR: RED + self._fmt + RESET,
logging.CRITICAL: BOLD_RED + self._fmt + RESET,
}
def format(self, record: logging.LogRecord) -> str:
log_fmt = self._color_mapping.get(record.levelno)
formatter = logging.Formatter(log_fmt)
return formatter.format(record)