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
|