feature:基本框架

utils/logger.py

@@ -0,0 +1,65 @@
+"""分级日志（info.log / error.log + console），自动携带 trace_id，金融敏感字段脱敏。"""
+import logging
+import re
+import sys
+from logging.handlers import TimedRotatingFileHandler
+from pathlib import Path
+
+from utils.request_id import get_request_id
+
+_TRACE_FMT = "%(asctime)s %(levelname)-7s [%(trace_id)s] %(name)s: %(message)s"
+
+# 手机号（11位）/ 身份证（18位，末位可 X）→ 日志脱敏
+_PHONE = re.compile(r"(?<!\d)1\d{10}(?!\d)")
+_IDCARD = re.compile(r"(?<!\d)\d{15}(\d\d[\dXx])(?!\d)|(?<!\d)\d{17}[\dXx](?!\d)")
+
+_logging_configured = False
+
+
+class TraceIdFilter(logging.Filter):
+    def filter(self, record: logging.LogRecord) -> bool:
+        record.trace_id = get_request_id() or "-"
+        return True
+
+
+class SensitiveMaskFilter(logging.Filter):
+    """身份证保留前4后4、手机号保留前3后4，其余用 * 掩码。"""
+
+    def filter(self, record: logging.LogRecord) -> bool:
+        msg = record.getMessage()
+        msg = _IDCARD.sub(lambda m: m.group()[:4] + "**********" + m.group()[-4:], msg)
+        msg = _PHONE.sub(lambda m: m.group()[:3] + "****" + m.group()[-4:], msg)
+        record.msg = msg
+        record.args = ()
+        return True
+
+
+def get_logger(name: str = "app") -> logging.Logger:
+    return logging.getLogger(name)
+
+
+def setup_logging(log_dir: str = "./logs") -> None:
+    """幂等初始化：console + 按天滚动的 info.log / error.log。"""
+    global _logging_configured
+    if _logging_configured:
+        return
+    _logging_configured = True
+
+    root = logging.getLogger()
+    root.setLevel(logging.INFO)
+    formatter = logging.Formatter(_TRACE_FMT)
+    Path(log_dir).mkdir(parents=True, exist_ok=True)
+
+    handlers = [
+        logging.StreamHandler(sys.stderr),
+        TimedRotatingFileHandler(str(Path(log_dir) / "info.log"), when="midnight", backupCount=30, encoding="utf-8"),
+        TimedRotatingFileHandler(str(Path(log_dir) / "error.log"), when="midnight", backupCount=30, encoding="utf-8"),
+    ]
+    handlers[0].setLevel(logging.INFO)
+    handlers[1].setLevel(logging.INFO)
+    handlers[2].setLevel(logging.ERROR)
+    for h in handlers:
+        h.setFormatter(formatter)
+        h.addFilter(TraceIdFilter())
+        h.addFilter(SensitiveMaskFilter())
+        root.addHandler(h)