mirror of
https://github.com/PaiGramTeam/PamGram.git
synced 2024-11-25 07:36:27 +00:00
233e7ab58d
Co-authored-by: luoshuijs <luoshuijs@outlook.com> Co-authored-by: Karako <karakohear@gmail.com> Co-authored-by: xtaodada <xtao@xtaolink.cn>
46 lines
1.3 KiB
Python
46 lines
1.3 KiB
Python
import re
|
|
from functools import lru_cache
|
|
from typing import TYPE_CHECKING
|
|
|
|
from core.config import config
|
|
from utils.log._config import LoggerConfig
|
|
from utils.log._logger import LogFilter, Logger
|
|
|
|
if TYPE_CHECKING:
|
|
from logging import LogRecord
|
|
|
|
__all__ = ("logger",)
|
|
|
|
logger = Logger(
|
|
LoggerConfig(
|
|
name=config.logger.name,
|
|
width=config.logger.width,
|
|
time_format=config.logger.time_format,
|
|
traceback_max_frames=config.logger.traceback_max_frames,
|
|
log_path=config.logger.path,
|
|
keywords=config.logger.render_keywords,
|
|
traceback_locals_max_depth=config.logger.locals_max_depth,
|
|
traceback_locals_max_length=config.logger.locals_max_length,
|
|
traceback_locals_max_string=config.logger.locals_max_string,
|
|
)
|
|
)
|
|
|
|
|
|
@lru_cache
|
|
def _whitelist_name_filter(record_name: str) -> bool:
|
|
"""白名单过滤器"""
|
|
return any(re.match(rf"^{name}.*?$", record_name) for name in config.logger.filtered_names + [config.logger.name])
|
|
|
|
|
|
def name_filter(record: "LogRecord") -> bool:
|
|
"""默认的过滤器. 白名单
|
|
|
|
根据当前的 record 的 name 判断是否需要打印。如果应该打印，则返回 True;否则返回 False。
|
|
"""
|
|
return _whitelist_name_filter(record.name)
|
|
|
|
|
|
log_filter = LogFilter()
|
|
log_filter.add_filter(name_filter)
|
|
logger.addFilter(log_filter)
|