39 lines
1.2 KiB
Python
39 lines
1.2 KiB
Python
|
import os
|
|||
|
|
import sys
|
|||
|
|
from loguru import logger
|
|||
|
|
|
|||
|
|
# 移除默认的 handler
|
|||
|
|
logger.remove()
|
|||
|
|
|
|||
|
|
# 获取日志级别（从环境变量或默认为 ERROR，正式使用时只显示错误）
|
|||
|
|
log_level = os.getenv("AUTOTERMINAL_LOG_LEVEL", "ERROR")
|
|||
|
|
|
|||
|
|
# 添加控制台输出（stderr）- 默认只显示错误
|
|||
|
|
logger.add(
|
|||
|
|
sys.stderr,
|
|||
|
|
format="<level>{level}: {message}</level>",
|
|||
|
|
level=log_level,
|
|||
|
|
colorize=True
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 添加文件输出（可选，存储在 ~/.autoterminal/ 目录）
|
|||
|
|
enable_file_logging = os.getenv("AUTOTERMINAL_FILE_LOG", "true").lower() != "false"
|
|||
|
|
if enable_file_logging:
|
|||
|
|
home_dir = os.path.expanduser("~")
|
|||
|
|
log_dir = os.path.join(home_dir, ".autoterminal")
|
|||
|
|
os.makedirs(log_dir, exist_ok=True)
|
|||
|
|
log_file = os.path.join(log_dir, "autoterminal.log")
|
|||
|
|
|
|||
|
|
logger.add(
|
|||
|
|
log_file,
|
|||
|
|
format="{time:YYYY-MM-DD HH:mm:ss} | {level: <8} | {name}:{function}:{line} - {message}",
|
|||
|
|
level="DEBUG", # 文件记录所有级别的日志
|
|||
|
|
rotation="10 MB", # 日志文件达到 10MB 时轮转
|
|||
|
|
retention="7 days", # 保留最近 7 天的日志
|
|||
|
|
compression="zip", # 压缩旧日志
|
|||
|
|
encoding="utf-8"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 导出 logger 供其他模块使用
|
|||
|
|
__all__ = ["logger"]
|