106 lines
2.7 KiB
Python
106 lines
2.7 KiB
Python
|
#!/usr/bin/env python3
|
|||
|
|
# -*- coding: utf-8 -*-
|
|||
|
|
"""
|
|||
|
|
日志管理模块
|
|||
|
|
提供统一的日志输出功能，支持控制台和文件输出
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
import os
|
|||
|
|
from datetime import datetime
|
|||
|
|
from pathlib import Path
|
|||
|
|
|
|||
|
|
|
|||
|
|
class Logger:
|
|||
|
|
"""日志管理器"""
|
|||
|
|
|
|||
|
|
# 颜色定义
|
|||
|
|
RED = '\033[0;31m'
|
|||
|
|
GREEN = '\033[0;32m'
|
|||
|
|
YELLOW = '\033[1;33m'
|
|||
|
|
BLUE = '\033[0;34m'
|
|||
|
|
NC = '\033[0m' # No Color
|
|||
|
|
|
|||
|
|
# 日志配置
|
|||
|
|
_log_file = None
|
|||
|
|
_max_size = 10485760 # 10MB
|
|||
|
|
_initialized = False
|
|||
|
|
|
|||
|
|
@classmethod
|
|||
|
|
def init(cls, log_file=None, max_size=None):
|
|||
|
|
"""
|
|||
|
|
初始化日志配置
|
|||
|
|
|
|||
|
|
参数：
|
|||
|
|
log_file: 日志文件路径
|
|||
|
|
max_size: 日志文件最大大小（字节）
|
|||
|
|
"""
|
|||
|
|
if log_file:
|
|||
|
|
cls._log_file = Path(log_file)
|
|||
|
|
# 确保日志目录存在
|
|||
|
|
cls._log_file.parent.mkdir(parents=True, exist_ok=True)
|
|||
|
|
|
|||
|
|
if max_size:
|
|||
|
|
cls._max_size = max_size
|
|||
|
|
|
|||
|
|
cls._initialized = True
|
|||
|
|
|
|||
|
|
@classmethod
|
|||
|
|
def _rotate_log(cls):
|
|||
|
|
"""日志轮转：如果日志文件超过最大大小，进行轮转"""
|
|||
|
|
if not cls._log_file or not cls._log_file.exists():
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
if cls._log_file.stat().st_size >= cls._max_size:
|
|||
|
|
# 轮转日志文件
|
|||
|
|
backup_file = cls._log_file.with_suffix('.log.1')
|
|||
|
|
if backup_file.exists():
|
|||
|
|
backup_file.unlink()
|
|||
|
|
cls._log_file.rename(backup_file)
|
|||
|
|
|
|||
|
|
@classmethod
|
|||
|
|
def _write_log(cls, level, message):
|
|||
|
|
"""写入日志到控制台和文件"""
|
|||
|
|
timestamp = datetime.now().strftime('%Y-%m-%d %H:%M:%S')
|
|||
|
|
|
|||
|
|
# 控制台输出（带颜色）
|
|||
|
|
color_map = {
|
|||
|
|
'INFO': cls.GREEN,
|
|||
|
|
'ERROR': cls.RED,
|
|||
|
|
'WARN': cls.YELLOW,
|
|||
|
|
'DEBUG': cls.BLUE
|
|||
|
|
}
|
|||
|
|
color = color_map.get(level, cls.NC)
|
|||
|
|
print(f"{color}[{level}]{cls.NC} {timestamp} - {message}")
|
|||
|
|
|
|||
|
|
# 文件输出（不带颜色）
|
|||
|
|
if cls._log_file:
|
|||
|
|
cls._rotate_log()
|
|||
|
|
log_line = f"[{level}] {timestamp} - {message}\n"
|
|||
|
|
with open(cls._log_file, 'a', encoding='utf-8') as f:
|
|||
|
|
f.write(log_line)
|
|||
|
|
|
|||
|
|
@classmethod
|
|||
|
|
def info(cls, message):
|
|||
|
|
"""输出信息日志"""
|
|||
|
|
cls._write_log('INFO', message)
|
|||
|
|
|
|||
|
|
@classmethod
|
|||
|
|
def error(cls, message):
|
|||
|
|
"""输出错误日志"""
|
|||
|
|
cls._write_log('ERROR', message)
|
|||
|
|
|
|||
|
|
@classmethod
|
|||
|
|
def warn(cls, message):
|
|||
|
|
"""输出警告日志"""
|
|||
|
|
cls._write_log('WARN', message)
|
|||
|
|
|
|||
|
|
@classmethod
|
|||
|
|
def debug(cls, message):
|
|||
|
|
"""输出调试日志"""
|
|||
|
|
cls._write_log('DEBUG', message)
|
|||
|
|
|
|||
|
|
@staticmethod
|
|||
|
|
def separator():
|
|||
|
|
"""输出分隔线"""
|
|||
|
|
print("=" * 60)
|