53 lines
1.5 KiB
Python
53 lines
1.5 KiB
Python
|
import logging
|
||
|
|
import sys
|
||
|
|
|
||
|
|
from loguru import logger
|
||
|
|
|
||
|
|
|
||
|
|
class InterceptHandler(logging.Handler):
|
||
|
|
LEVELS_MAP = {
|
||
|
|
logging.CRITICAL: "CRITICAL",
|
||
|
|
logging.ERROR: "ERROR",
|
||
|
|
logging.WARNING: "WARNING",
|
||
|
|
logging.INFO: "INFO",
|
||
|
|
logging.DEBUG: "DEBUG",
|
||
|
|
}
|
||
|
|
|
||
|
|
def _get_level(self, record):
|
||
|
|
return self.LEVELS_MAP.get(record.levelno, record.levelno)
|
||
|
|
|
||
|
|
def emit(self, record):
|
||
|
|
logger_opt = logger.opt(depth=6, exception=record.exc_info)
|
||
|
|
logger_opt.log(self._get_level(record), record.getMessage())
|
||
|
|
|
||
|
|
|
||
|
|
def configure_logger(capture_exceptions: bool = False) -> None:
|
||
|
|
logger.remove()
|
||
|
|
level = "INFO"
|
||
|
|
logger.add(
|
||
|
|
"logs/log_{time:YYYY-MM-DD}.log",
|
||
|
|
rotation="12:00",
|
||
|
|
format="{time:YYYY-MM-DD HH:mm:ss} | {level} | {file}:{line} | {message}",
|
||
|
|
level="INFO",
|
||
|
|
encoding="utf-8",
|
||
|
|
compression="zip",
|
||
|
|
)
|
||
|
|
logger.add(
|
||
|
|
sys.stdout,
|
||
|
|
colorize=True,
|
||
|
|
format="<green>{time:YYYY-MM-DD at HH:mm:ss}</green> | <level>{level}</level> | {file}:{line} | "
|
||
|
|
"{message}",
|
||
|
|
level=level,
|
||
|
|
)
|
||
|
|
if capture_exceptions:
|
||
|
|
logger.add(
|
||
|
|
"logs/error_log_{time:YYYY-MM-DD}.log",
|
||
|
|
rotation="12:00",
|
||
|
|
format="{time:YYYY-MM-DD HH:mm:ss} | {level} | {file}:{line} | {message}",
|
||
|
|
level="ERROR",
|
||
|
|
encoding="utf-8",
|
||
|
|
compression="zip",
|
||
|
|
)
|
||
|
|
|
||
|
|
logging.basicConfig(handlers=[InterceptHandler()], level=logging.INFO)
|