92 lines
2.4 KiB
Python
92 lines
2.4 KiB
Python
"""MinerU File转Markdown转换服务的配置工具。"""
|
||
|
||
import os
|
||
import logging
|
||
from pathlib import Path
|
||
from dotenv import load_dotenv
|
||
|
||
# 从 .env 文件加载环境变量
|
||
load_dotenv()
|
||
|
||
# API 配置
|
||
MINERU_API_BASE = os.getenv("MINERU_API_BASE", "https://mineru.net")
|
||
MINERU_API_KEY = os.getenv("MINERU_API_KEY", "")
|
||
|
||
# 本地API配置
|
||
USE_LOCAL_API = os.getenv("USE_LOCAL_API", "").lower() in ["true", "1", "yes"]
|
||
LOCAL_MINERU_API_BASE = os.getenv("LOCAL_MINERU_API_BASE", "http://localhost:8080")
|
||
|
||
# 转换后文件的默认输出目录
|
||
DEFAULT_OUTPUT_DIR = os.getenv("OUTPUT_DIR", "./downloads")
|
||
|
||
|
||
# 设置日志系统
|
||
def setup_logging():
|
||
"""
|
||
设置日志系统，根据环境变量配置日志级别。
|
||
|
||
Returns:
|
||
logging.Logger: 配置好的日志记录器。
|
||
"""
|
||
# 获取环境变量中的日志级别设置
|
||
log_level = os.getenv("MINERU_LOG_LEVEL", "INFO").upper()
|
||
debug_mode = os.getenv("MINERU_DEBUG", "").lower() in ["true", "1", "yes"]
|
||
|
||
# 如果设置了debug_mode，则覆盖log_level
|
||
if debug_mode:
|
||
log_level = "DEBUG"
|
||
|
||
# 确保log_level是有效的
|
||
valid_levels = ["DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL"]
|
||
if log_level not in valid_levels:
|
||
log_level = "INFO"
|
||
|
||
# 设置日志格式
|
||
log_format = "%(asctime)s - %(name)s - %(levelname)s - %(message)s"
|
||
|
||
# 配置日志
|
||
logging.basicConfig(level=getattr(logging, log_level), format=log_format)
|
||
|
||
logger = logging.getLogger("mineru")
|
||
logger.setLevel(getattr(logging, log_level))
|
||
|
||
# 输出日志级别信息
|
||
logger.info(f"日志级别设置为: {log_level}")
|
||
|
||
return logger
|
||
|
||
|
||
# 创建默认的日志记录器
|
||
logger = setup_logging()
|
||
|
||
|
||
# 如果输出目录不存在，则创建它
|
||
def ensure_output_dir(output_dir=None):
|
||
"""
|
||
确保输出目录存在。
|
||
|
||
Args:
|
||
output_dir: 输出目录的可选路径。如果为 None，则使用 DEFAULT_OUTPUT_DIR。
|
||
|
||
Returns:
|
||
表示输出目录的 Path 对象。
|
||
"""
|
||
output_path = Path(output_dir or DEFAULT_OUTPUT_DIR)
|
||
output_path.mkdir(parents=True, exist_ok=True)
|
||
return output_path
|
||
|
||
|
||
# 验证 API 配置
|
||
def validate_api_config():
|
||
"""
|
||
验证是否已设置所需的 API 配置。
|
||
|
||
Returns:
|
||
dict: 配置状态。
|
||
"""
|
||
return {
|
||
"api_base": MINERU_API_BASE,
|
||
"api_key_set": bool(MINERU_API_KEY),
|
||
"output_dir": DEFAULT_OUTPUT_DIR,
|
||
}
|