67 lines
2.4 KiB
Python
67 lines
2.4 KiB
Python
|
import logging
|
|||
|
|
import re
|
|||
|
|
from logging.handlers import TimedRotatingFileHandler
|
|||
|
|
|
|||
|
|
# 统一日志格式：含模块名(name)与源码行号，便于 grep 定位
|
|||
|
|
LOG_FORMAT = '%(asctime)s %(name)s:%(lineno)d [%(levelname)s] %(message)s'
|
|||
|
|
|
|||
|
|
|
|||
|
|
class SensitiveDataFilter(logging.Filter):
|
|||
|
|
"""Redact common credentials even when callers log an entire request mapping."""
|
|||
|
|
_assignments = re.compile(
|
|||
|
|
r"(?i)(['\"]?(?:api[_-]?key|password|passwd|secret|token|authorization)['\"]?\s*[:=]\s*)"
|
|||
|
|
r"(['\"][^'\"]*['\"]|[^,}\s]+)",
|
|||
|
|
)
|
|||
|
|
_url_userinfo = re.compile(r"(?i)\b((?:rtsp|rtmp|https?)://)[^/@\s]+@")
|
|||
|
|
|
|||
|
|
def filter(self, record):
|
|||
|
|
try:
|
|||
|
|
rendered = record.getMessage()
|
|||
|
|
rendered = self._assignments.sub(r"\1[REDACTED]", rendered)
|
|||
|
|
rendered = self._url_userinfo.sub(r"\1[REDACTED]@", rendered)
|
|||
|
|
record.msg = rendered
|
|||
|
|
record.args = ()
|
|||
|
|
except Exception:
|
|||
|
|
pass
|
|||
|
|
return True
|
|||
|
|
|
|||
|
|
|
|||
|
|
def CreateLogger(filepath, is_show_console=False,log_debug=False):
|
|||
|
|
LOGGER_WHEN = 'd'
|
|||
|
|
LOGFILE_BACKUPCOUNT = 3
|
|||
|
|
if log_debug:
|
|||
|
|
level = logging.DEBUG
|
|||
|
|
else:
|
|||
|
|
level = logging.INFO
|
|||
|
|
logger = logging.getLogger()
|
|||
|
|
logger.setLevel(level)
|
|||
|
|
formatter = logging.Formatter(LOG_FORMAT)
|
|||
|
|
|
|||
|
|
# 最基础
|
|||
|
|
# fileHandler = logging.FileHandler(filepath, encoding='utf-8') # 指定utf-8格式编码，避免输出的日志文本乱码
|
|||
|
|
# fileHandler.setLevel(level)
|
|||
|
|
# fileHandler.setFormatter(formatter)
|
|||
|
|
# logger.addHandler(fileHandler)
|
|||
|
|
|
|||
|
|
# 时间滚动切分
|
|||
|
|
# when:备份的时间单位，backupCount:备份保存的时间长度
|
|||
|
|
timedRotatingFileHandler = TimedRotatingFileHandler(filepath,
|
|||
|
|
when=LOGGER_WHEN,
|
|||
|
|
backupCount=LOGFILE_BACKUPCOUNT,
|
|||
|
|
encoding='utf-8')
|
|||
|
|
|
|||
|
|
timedRotatingFileHandler.setLevel(level)
|
|||
|
|
timedRotatingFileHandler.setFormatter(formatter)
|
|||
|
|
timedRotatingFileHandler.addFilter(SensitiveDataFilter())
|
|||
|
|
logger.addHandler(timedRotatingFileHandler)
|
|||
|
|
|
|||
|
|
# 控制台打印
|
|||
|
|
if is_show_console:
|
|||
|
|
streamHandler = logging.StreamHandler()
|
|||
|
|
streamHandler.setLevel(level)
|
|||
|
|
streamHandler.setFormatter(formatter)
|
|||
|
|
streamHandler.addFilter(SensitiveDataFilter())
|
|||
|
|
logger.addHandler(streamHandler)
|
|||
|
|
|
|||
|
|
return logger
|