98 lines
3.1 KiB
Python
98 lines
3.1 KiB
Python
import os
|
||
import sys
|
||
from pathlib import Path
|
||
from loguru import logger
|
||
import platform
|
||
from datetime import datetime
|
||
import zipfile
|
||
|
||
class CrossPlatformLog:
|
||
"""跨平台日志系统（支持Linux/Windows/Mac）"""
|
||
|
||
def __init__(self):
|
||
self.log_dir = self._get_log_dir()
|
||
self._setup_logger()
|
||
|
||
def _get_log_dir(self):
|
||
"""获取跨平台日志目录（相对路径）"""
|
||
base_dir = Path(__file__).parent.parent # 项目根目录
|
||
log_dir = base_dir / "logs"
|
||
|
||
# 自动创建日志目录
|
||
log_dir.mkdir(exist_ok=True)
|
||
|
||
# Windows特殊权限处理
|
||
if platform.system() == "Windows":
|
||
try:
|
||
os.chmod(log_dir, 0o777) # 确保写入权限
|
||
except:
|
||
pass
|
||
|
||
return log_dir
|
||
|
||
def _setup_logger(self):
|
||
"""配置跨平台日志处理器"""
|
||
logger.remove() # 清除默认配置
|
||
|
||
# 统一控制台输出格式
|
||
logger.add(
|
||
sys.stdout,
|
||
level="INFO",
|
||
format="<green>{time:YYYY-MM-DD HH:mm:ss}</green> | <level>{level: <8}</level> | <cyan>{module}</cyan> - <level>{message}</level>",
|
||
filter=lambda record: record["level"].no >= 20 # INFO及以上级别
|
||
)
|
||
|
||
# 主日志文件（兼容所有平台路径）
|
||
self._add_main_log()
|
||
|
||
# 错误日志单独存储
|
||
self._add_error_log()
|
||
|
||
def _add_main_log(self):
|
||
"""主日志文件配置"""
|
||
main_log = self.log_dir / "application.log"
|
||
logger.add(
|
||
str(main_log),
|
||
rotation="20 MB",
|
||
compression=self._compress_log,
|
||
encoding="utf-8",
|
||
level="DEBUG",
|
||
format="{time:YYYY-MM-DD HH:mm:ss.SSS} | {level: <8} | {module}:{line} - {message}",
|
||
retention="30 days",
|
||
enqueue=True # 线程安全
|
||
)
|
||
|
||
def _add_error_log(self):
|
||
"""错误日志专用配置"""
|
||
error_log = self.log_dir / "errors.log"
|
||
logger.add(
|
||
str(error_log),
|
||
level="ERROR",
|
||
format="{time:YYYY-MM-DD HH:mm:ss.SSS} | ERROR | {module}:{line} - {message}\n{exception}",
|
||
rotation="10 MB",
|
||
retention="90 days"
|
||
)
|
||
|
||
@staticmethod
|
||
def _compress_log(log_path):
|
||
"""通用日志压缩方法（兼容所有平台）"""
|
||
if not os.path.exists(log_path):
|
||
return
|
||
|
||
try:
|
||
zip_path = f"{log_path}.{datetime.now().strftime('%Y%m%d')}.zip"
|
||
with zipfile.ZipFile(zip_path, 'w', zipfile.ZIP_DEFLATED) as zipf:
|
||
zipf.write(log_path, arcname=os.path.basename(log_path))
|
||
os.remove(log_path)
|
||
return zip_path
|
||
except Exception as e:
|
||
print(f"日志压缩失败: {str(e)}")
|
||
return log_path # 返回原文件路径继续使用
|
||
|
||
@classmethod
|
||
def get_logger(cls, module_name=None):
|
||
"""获取模块专属日志器"""
|
||
return logger.bind(module=module_name or "__main__")
|
||
|
||
# 初始化全局日志器
|
||
log = CrossPlatformLog().get_logger() |