✨ New logging system
This commit is contained in:
25
src/logging/color_log_formatter.py
Normal file
25
src/logging/color_log_formatter.py
Normal file
@@ -0,0 +1,25 @@
|
||||
from logging import Formatter, LogRecord
|
||||
|
||||
|
||||
class ColorLogFormatter(Formatter):
|
||||
"""Formatter that outputs logs in a colored format"""
|
||||
|
||||
RESET = "\033[0m"
|
||||
DIM = "\033[2m"
|
||||
BOLD = "\033[1m"
|
||||
RED = "\033[31m"
|
||||
YELLOW = "\033[33m"
|
||||
|
||||
def format(self, record: LogRecord):
|
||||
super_format = super().format(record)
|
||||
match record.levelname:
|
||||
case "CRITICAL":
|
||||
return self.BOLD + self.RED + super_format + self.RESET
|
||||
case "ERROR":
|
||||
return self.RED + super_format + self.RESET
|
||||
case "WARNING":
|
||||
return self.YELLOW + super_format + self.RESET
|
||||
case "DEBUG":
|
||||
return self.DIM + super_format + self.RESET
|
||||
case _other:
|
||||
return super_format
|
||||
58
src/logging/setup.py
Normal file
58
src/logging/setup.py
Normal file
@@ -0,0 +1,58 @@
|
||||
import logging.config as logging_dot_config
|
||||
import os
|
||||
from datetime import datetime
|
||||
|
||||
from src import shared
|
||||
|
||||
|
||||
def setup_logging():
|
||||
"""Intitate the app's logging"""
|
||||
|
||||
# Prepare log file
|
||||
log_dir = shared.data_dir / "cartridges" / "logs"
|
||||
log_dir.mkdir(exist_ok=True)
|
||||
log_file = log_dir / f'{datetime.now().isoformat(timespec="seconds")}.log'
|
||||
|
||||
# Define log levels
|
||||
profile_main_log_level = "DEBUG" if shared.PROFILE == "development" else "WARNING"
|
||||
profile_lib_log_level = "INFO" if shared.PROFILE == "development" else "WARNING"
|
||||
main_log_level = os.environ.get("LOGLEVEL", profile_main_log_level).upper()
|
||||
lib_log_level = os.environ.get("LIBLOGLEVEL", profile_lib_log_level).upper()
|
||||
|
||||
# Load config
|
||||
config = {
|
||||
"version": 1,
|
||||
"formatters": {
|
||||
"console_formatter": {
|
||||
"class": "src.logging.color_log_formatter.ColorLogFormatter",
|
||||
"format": "%(name)s %(levelname)s - %(message)s",
|
||||
},
|
||||
"file_formatter": {
|
||||
"format": "%(asctime)s | %(name)s | %(levelname)s | %(message)s"
|
||||
},
|
||||
},
|
||||
"handlers": {
|
||||
"main_console_handler": {
|
||||
"class": "logging.StreamHandler",
|
||||
"formatter": "console_formatter",
|
||||
"level": main_log_level,
|
||||
},
|
||||
"lib_console_handler": {
|
||||
"class": "logging.StreamHandler",
|
||||
"formatter": "console_formatter",
|
||||
"level": lib_log_level,
|
||||
},
|
||||
"file_handler": {
|
||||
"class": "logging.FileHandler",
|
||||
"level": "DEBUG",
|
||||
"filename": str(log_file),
|
||||
"formatter": "file_formatter",
|
||||
},
|
||||
},
|
||||
"loggers": {
|
||||
"PIL": {"handlers": ["lib_console_handler", "file_handler"]},
|
||||
"urllib3": {"handlers": ["lib_console_handler", "file_handler"]},
|
||||
"root": {"handlers": ["main_console_handler", "file_handler"]},
|
||||
},
|
||||
}
|
||||
logging_dot_config.dictConfig(config)
|
||||
Reference in New Issue
Block a user