62 lines
2.2 KiB
Python
62 lines
2.2 KiB
Python
# utils/logger.py
|
|
import logging
|
|
from datetime import datetime
|
|
from pathlib import Path
|
|
from typing import Optional
|
|
|
|
class SupersetLogger:
|
|
def __init__(
|
|
self,
|
|
name: str = "superset_tool",
|
|
log_dir: Optional[Path] = None,
|
|
level: int = logging.INFO,
|
|
console: bool = True
|
|
):
|
|
self.logger = logging.getLogger(name)
|
|
self.logger.setLevel(level)
|
|
|
|
formatter = logging.Formatter(
|
|
'%(asctime)s - %(levelname)s - %(message)s'
|
|
)
|
|
|
|
# Очищаем существующие обработчики
|
|
if self.logger.handlers:
|
|
for handler in self.logger.handlers[:]:
|
|
self.logger.removeHandler(handler)
|
|
|
|
# Файловый обработчик
|
|
if log_dir:
|
|
log_dir.mkdir(parents=True, exist_ok=True)
|
|
file_handler = logging.FileHandler(
|
|
log_dir / f"{name}_{self._get_timestamp()}.log"
|
|
)
|
|
file_handler.setFormatter(formatter)
|
|
self.logger.addHandler(file_handler)
|
|
|
|
# Консольный обработчик
|
|
if console:
|
|
console_handler = logging.StreamHandler()
|
|
console_handler.setFormatter(formatter)
|
|
self.logger.addHandler(console_handler)
|
|
|
|
def _get_timestamp(self) -> str:
|
|
return datetime.now().strftime("%Y%m%d")
|
|
|
|
def info(self, message: str, extra: Optional[dict] = None, exc_info: bool = False):
|
|
self.logger.info(message, extra=extra, exc_info=exc_info)
|
|
|
|
def error(self, message: str, extra: Optional[dict] = None, exc_info: bool = False):
|
|
self.logger.error(message, extra=extra, exc_info=exc_info)
|
|
|
|
def warning(self, message: str, extra: Optional[dict] = None, exc_info: bool = False):
|
|
self.logger.warning(message, extra=extra, exc_info=exc_info)
|
|
|
|
def critical(self, message: str, extra: Optional[dict] = None, exc_info: bool = False):
|
|
self.logger.critical(message, extra=extra, exc_info=exc_info)
|
|
|
|
def debug(self, message: str, extra: Optional[dict] = None, exc_info: bool = False):
|
|
self.logger.debug(message, extra=extra, exc_info=exc_info)
|
|
|
|
def exception(self, message: str):
|
|
self.logger.exception(message)
|