Spaces:
Sleeping
Sleeping
File size: 2,004 Bytes
ee933ab | 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 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 | import logging
import sys
from pathlib import Path
from typing import Optional
# ANSI Color Codes
class Colors:
HEADER = '\033[95m'
OKBLUE = '\033[94m'
OKCYAN = '\033[96m'
OKGREEN = '\033[92m'
WARNING = '\033[93m'
FAIL = '\033[91m'
ENDC = '\033[0m'
BOLD = '\033[1m'
UNDERLINE = '\033[4m'
class ColorFormatter(logging.Formatter):
"""Custom Formatter to add colors to log levels."""
FORMATS = {
logging.DEBUG: Colors.OKBLUE + "%(levelname)s: %(message)s" + Colors.ENDC,
logging.INFO: Colors.OKGREEN + "%(levelname)s: %(message)s" + Colors.ENDC,
logging.WARNING: Colors.WARNING + "%(levelname)s: %(message)s" + Colors.ENDC,
logging.ERROR: Colors.FAIL + "%(levelname)s: %(message)s" + Colors.ENDC,
logging.CRITICAL: Colors.FAIL + Colors.BOLD + "%(levelname)s: %(message)s" + Colors.ENDC
}
def format(self, record):
log_fmt = self.FORMATS.get(record.levelno)
formatter = logging.Formatter(log_fmt)
return formatter.format(record)
def get_logger(name: str, level: int = logging.INFO, log_file: Optional[Path] = None) -> logging.Logger:
"""Returns a color-coded logger instance, optionally logging to a file."""
logger = logging.getLogger(name)
logger.setLevel(level)
if not logger.handlers:
# Console handler with colors
console_handler = logging.StreamHandler(sys.stdout)
console_handler.setFormatter(ColorFormatter())
logger.addHandler(console_handler)
# File handler without colors (cleaner for logs)
if log_file:
from pathlib import Path
file_handler = logging.FileHandler(str(log_file), encoding='utf-8')
file_formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
file_handler.setFormatter(file_formatter)
logger.addHandler(file_handler)
return logger
|