85 lines
2.6 KiB
Python
85 lines
2.6 KiB
Python
import logging
|
|
from datetime import datetime
|
|
|
|
|
|
class ColoredConsoleFormatter(logging.Formatter):
|
|
TIME_COLOR = '\033[38;2;89;89;89m'
|
|
RESET = '\033[0m'
|
|
BOLD = '\033[1m'
|
|
DARK_CYAN = '\033[36m'
|
|
DARK_YELLOW = '\033[33m'
|
|
RED = '\033[91m'
|
|
|
|
LEVEL_COLORS = {
|
|
logging.DEBUG: '\033[36m',
|
|
logging.INFO: '\033[32m',
|
|
logging.WARNING: '\033[33m',
|
|
logging.ERROR: RED,
|
|
logging.CRITICAL: '\033[35m',
|
|
}
|
|
|
|
LEVEL_NAMES = {
|
|
logging.DEBUG: 'DBG',
|
|
logging.INFO: 'INF',
|
|
logging.WARNING: 'WRN',
|
|
logging.ERROR: 'ERR',
|
|
logging.CRITICAL: 'CRT',
|
|
}
|
|
|
|
def format(self, record: logging.LogRecord) -> str:
|
|
timestamp = f'{self.TIME_COLOR}{datetime.fromtimestamp(record.created).strftime("%H:%M:%S")}{self.RESET}'
|
|
|
|
level_name = self.LEVEL_NAMES.get(record.levelno, record.levelname[:3])
|
|
level_color = self.LEVEL_COLORS.get(record.levelno, '')
|
|
level = f'{level_color}{level_name}{self.RESET}'
|
|
|
|
message_color = self.RED if record.levelno >= logging.ERROR else ''
|
|
message = f'{message_color}{self.BOLD}{record.getMessage()}{self.RESET}'
|
|
|
|
# Add extra fields
|
|
extra_parts = []
|
|
for key, value in record.__dict__.items():
|
|
if key not in [
|
|
'name',
|
|
'msg',
|
|
'args',
|
|
'created',
|
|
'filename',
|
|
'funcName',
|
|
'levelname',
|
|
'levelno',
|
|
'lineno',
|
|
'module',
|
|
'msecs',
|
|
'message',
|
|
'pathname',
|
|
'process',
|
|
'processName',
|
|
'relativeCreated',
|
|
'thread',
|
|
'threadName',
|
|
'exc_info',
|
|
'exc_text',
|
|
'stack_info',
|
|
'app_name',
|
|
'app_version',
|
|
'taskName',
|
|
'color_message',
|
|
]:
|
|
value_color = self.RED if key == 'error' else self.DARK_YELLOW
|
|
extra_parts.append(f'{self.DARK_CYAN}{key}{self.RESET}={value_color}{value}{self.RESET}')
|
|
|
|
if extra_parts:
|
|
message += f' {" ".join(extra_parts)}'
|
|
|
|
if record.levelno >= logging.WARNING:
|
|
location = f'{self.TIME_COLOR}{record.module}:{record.lineno}{self.RESET}'
|
|
result = f'{timestamp} {level} {location} {message}'
|
|
else:
|
|
result = f'{timestamp} {level} {message}'
|
|
|
|
if record.exc_info:
|
|
result += '\n' + self.formatException(record.exc_info)
|
|
|
|
return result
|