35 lines
1.1 KiB
Python
35 lines
1.1 KiB
Python
import logging
|
||
from logging.handlers import RotatingFileHandler
|
||
import os
|
||
|
||
LOG_DIR = "logs"
|
||
LOG_FILE = "app.log"
|
||
LOG_PATH = os.path.join(LOG_DIR, LOG_FILE)
|
||
MAX_BYTES = 5 * 1024 * 1024 # 5MB
|
||
BACKUP_COUNT = 5
|
||
|
||
# 全局共用的 handler，只初始化一次
|
||
_handler = None
|
||
|
||
def setup_logger(name: str, level=logging.INFO) -> logging.Logger:
|
||
global _handler
|
||
|
||
if not os.path.exists(LOG_DIR):
|
||
os.makedirs(LOG_DIR)
|
||
|
||
logger = logging.getLogger(name)
|
||
logger.setLevel(level) # 设置这个 logger 的最低级别
|
||
|
||
# 避免重复添加 handler（只添加一次）
|
||
if not _handler:
|
||
_handler = RotatingFileHandler(LOG_PATH, maxBytes=MAX_BYTES, backupCount=BACKUP_COUNT, encoding='utf-8')
|
||
formatter = logging.Formatter('[%(asctime)s] %(levelname)s [%(name)s] %(message)s')
|
||
_handler.setFormatter(formatter)
|
||
_handler.setLevel(logging.DEBUG) # Handler 设置最低等级，捕获所有内容
|
||
|
||
if not logger.handlers:
|
||
logger.addHandler(_handler)
|
||
logger.propagate = False # 避免日志冒泡打印两遍
|
||
|
||
return logger
|