148 lines
4.6 KiB
Python
148 lines
4.6 KiB
Python
import inspect
|
||
import logging
|
||
from logging.handlers import RotatingFileHandler
|
||
from pathlib import Path
|
||
from typing import Dict, Any
|
||
|
||
import click
|
||
|
||
from app.core.config import settings
|
||
|
||
# 日志级别颜色
|
||
level_name_colors = {
|
||
logging.DEBUG: lambda level_name: click.style(str(level_name), fg="cyan"),
|
||
logging.INFO: lambda level_name: click.style(str(level_name), fg="green"),
|
||
logging.WARNING: lambda level_name: click.style(str(level_name), fg="yellow"),
|
||
logging.ERROR: lambda level_name: click.style(str(level_name), fg="red"),
|
||
logging.CRITICAL: lambda level_name: click.style(
|
||
str(level_name), fg="bright_red"
|
||
),
|
||
}
|
||
|
||
|
||
class CustomFormatter(logging.Formatter):
|
||
"""
|
||
定义日志输出格式
|
||
"""
|
||
|
||
def format(self, record):
|
||
seperator = " " * (8 - len(record.levelname))
|
||
record.leveltext = level_name_colors[record.levelno](record.levelname + ":") + seperator
|
||
return super().format(record)
|
||
|
||
|
||
class LoggerManager:
|
||
"""
|
||
日志管理
|
||
"""
|
||
# 管理所有的Logger
|
||
_loggers: Dict[str, Any] = {}
|
||
# 默认日志文件
|
||
_default_log_file = "moviepilot.log"
|
||
|
||
@staticmethod
|
||
def __setup_logger(log_file: str):
|
||
"""
|
||
设置日志
|
||
log_file：日志文件相对路径
|
||
"""
|
||
log_file_path = settings.LOG_PATH / log_file
|
||
if not log_file_path.parent.exists():
|
||
log_file_path.parent.mkdir(parents=True, exist_ok=True)
|
||
|
||
# 创建新实例
|
||
_logger = logging.getLogger(log_file_path.stem)
|
||
|
||
# DEBUG
|
||
if settings.DEBUG:
|
||
_logger.setLevel(logging.DEBUG)
|
||
else:
|
||
_logger.setLevel(logging.INFO)
|
||
|
||
# 移除已有的 handler，避免重复添加
|
||
for handler in _logger.handlers:
|
||
_logger.removeHandler(handler)
|
||
|
||
# 终端日志
|
||
console_handler = logging.StreamHandler()
|
||
console_handler.setLevel(logging.DEBUG)
|
||
console_formatter = CustomFormatter("%(leveltext)s%(name)s - %(message)s")
|
||
console_handler.setFormatter(console_formatter)
|
||
_logger.addHandler(console_handler)
|
||
|
||
# 文件日志
|
||
file_handler = RotatingFileHandler(filename=log_file_path,
|
||
mode='w',
|
||
maxBytes=5 * 1024 * 1024,
|
||
backupCount=3,
|
||
encoding='utf-8')
|
||
file_handler.setLevel(logging.INFO)
|
||
file_formater = CustomFormatter("【%(levelname)s】%(asctime)s - %(name)s - %(message)s")
|
||
file_handler.setFormatter(file_formater)
|
||
_logger.addHandler(file_handler)
|
||
|
||
return _logger
|
||
|
||
def logger(self, path: str) -> logging.Logger:
|
||
"""
|
||
获取模块的logger
|
||
:param path: 当前运行程序路径
|
||
"""
|
||
filepath = Path(path)
|
||
|
||
# 区分插件日志
|
||
if "plugins" in filepath.parts:
|
||
# 使用插件日志文件
|
||
plugin_name = filepath.parts[filepath.parts.index("plugins") + 1]
|
||
logfile = Path("plugins") / f"{plugin_name}.log"
|
||
else:
|
||
# 使用默认日志文件
|
||
logfile = self._default_log_file
|
||
|
||
# 获取调用者的模块的logger
|
||
_logger = self._loggers.get(logfile)
|
||
if not _logger:
|
||
_logger = self.__setup_logger(logfile)
|
||
self._loggers[logfile] = _logger
|
||
return _logger
|
||
|
||
def info(self, msg, *args, **kwargs):
|
||
"""
|
||
重载info方法，按模块区分输出
|
||
"""
|
||
self.logger(inspect.stack()[1].filename).info(msg, *args, **kwargs)
|
||
|
||
def debug(self, msg, *args, **kwargs):
|
||
"""
|
||
重载debug方法，按模块区分输出
|
||
"""
|
||
self.logger(inspect.stack()[1].filename).debug(msg, *args, **kwargs)
|
||
|
||
def warning(self, msg, *args, **kwargs):
|
||
"""
|
||
重载warning方法，按模块区分输出
|
||
"""
|
||
self.logger(inspect.stack()[1].filename).warning(msg, *args, **kwargs)
|
||
|
||
def warn(self, msg, *args, **kwargs):
|
||
"""
|
||
重载warn方法，按模块区分输出
|
||
"""
|
||
self.warning(msg, *args, **kwargs)
|
||
|
||
def error(self, msg, *args, **kwargs):
|
||
"""
|
||
重载error方法，按模块区分输出
|
||
"""
|
||
self.logger(inspect.stack()[1].filename).error(msg, *args, **kwargs)
|
||
|
||
def critical(self, msg, *args, **kwargs):
|
||
"""
|
||
重载critical方法，按模块区分输出
|
||
"""
|
||
self.logger(inspect.stack()[1].filename).critical(msg, *args, **kwargs)
|
||
|
||
|
||
# 初始化公共日志
|
||
logger = LoggerManager()
|