戒酒的李白
57 lines
1.7 KiB
Python
57 lines
1.7 KiB
Python
import os
|
|
import logging
|
|
from logging.handlers import RotatingFileHandler
|
|
|
|
def setup_logger(name, log_file=None, level=logging.INFO):
|
|
"""
|
|
设置统一的日志记录器
|
|
|
|
Args:
|
|
name: 日志记录器名称
|
|
log_file: 日志文件路径，如果为None则只输出到控制台
|
|
level: 日志级别
|
|
|
|
Returns:
|
|
logger: 配置好的日志记录器
|
|
"""
|
|
# 创建日志记录器
|
|
logger = logging.getLogger(name)
|
|
logger.setLevel(level)
|
|
|
|
# 统一的日志格式
|
|
formatter = logging.Formatter(
|
|
'%(asctime)s [%(name)s] [%(levelname)s] %(message)s',
|
|
datefmt='%Y-%m-%d %H:%M:%S'
|
|
)
|
|
|
|
# 添加控制台处理器
|
|
console_handler = logging.StreamHandler()
|
|
console_handler.setFormatter(formatter)
|
|
logger.addHandler(console_handler)
|
|
|
|
# 如果指定了日志文件，添加文件处理器
|
|
if log_file:
|
|
# 确保日志目录存在
|
|
log_dir = os.path.dirname(log_file)
|
|
if log_dir and not os.path.exists(log_dir):
|
|
os.makedirs(log_dir)
|
|
|
|
# 使用 RotatingFileHandler 进行日志轮转
|
|
file_handler = RotatingFileHandler(
|
|
log_file,
|
|
maxBytes=10*1024*1024, # 10MB
|
|
backupCount=5,
|
|
encoding='utf-8'
|
|
)
|
|
file_handler.setFormatter(formatter)
|
|
logger.addHandler(file_handler)
|
|
|
|
return logger
|
|
|
|
# 创建默认的应用日志记录器
|
|
app_logger = setup_logger('weibo_analysis', 'logs/app.log')
|
|
spider_logger = setup_logger('spider', 'logs/spider.log')
|
|
model_logger = setup_logger('model', 'logs/model.log')
|
|
|
|
# 导出日志记录器
|
|
__all__ = ['setup_logger', 'app_logger', 'spider_logger', 'model_logger'] |