#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
QueryNest 日志工具模块
"""
import logging
import logging.handlers
import os
from pathlib import Path
from typing import Optional
def setup_logging(log_level: str = "INFO", log_file: Optional[str] = None) -> None:
"""
设置日志配置
Args:
log_level: 日志级别 (DEBUG, INFO, WARNING, ERROR, CRITICAL)
log_file: 日志文件路径，如果为None则只输出到控制台
"""
# 创建日志目录
if log_file:
log_dir = Path(log_file).parent
log_dir.mkdir(parents=True, exist_ok=True)
# 设置日志级别
level = getattr(logging, log_level.upper(), logging.INFO)
# 创建格式化器
formatter = logging.Formatter(
'%(asctime)s - %(name)s - %(levelname)s - %(message)s',
datefmt='%Y-%m-%d %H:%M:%S'
)
# 获取根日志记录器
root_logger = logging.getLogger()
root_logger.setLevel(level)
# 清除现有处理器
for handler in root_logger.handlers[:]:
root_logger.removeHandler(handler)
# 添加控制台处理器
console_handler = logging.StreamHandler()
console_handler.setLevel(level)
console_handler.setFormatter(formatter)
root_logger.addHandler(console_handler)
# 添加文件处理器（如果指定了日志文件）
if log_file:
file_handler = logging.handlers.RotatingFileHandler(
log_file,
maxBytes=10*1024*1024, # 10MB
backupCount=5
)
file_handler.setLevel(level)
file_handler.setFormatter(formatter)
root_logger.addHandler(file_handler)
def get_logger(name: str) -> logging.Logger:
"""
获取指定名称的日志记录器
Args:
name: 日志记录器名称
Returns:
logging.Logger: 日志记录器实例
"""
return logging.getLogger(name)
def set_log_level(level: str) -> None:
"""
设置日志级别
Args:
level: 日志级别字符串
"""
log_level = getattr(logging, level.upper(), logging.INFO)
logging.getLogger().setLevel(log_level)
# 更新所有处理器的级别
for handler in logging.getLogger().handlers:
handler.setLevel(log_level)
