Added a logging utility class and supplemented, standardized the logging output for all modules.
This commit is contained in:
@@ -0,0 +1,57 @@
|
||||
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']
|
||||
Reference in New Issue
Block a user