69 lines
1.5 KiB
Python
69 lines
1.5 KiB
Python
"""
|
|
通用工具模块
|
|
包含 JSON 读写等通用功能
|
|
"""
|
|
|
|
import json
|
|
import logging
|
|
import os
|
|
from typing import Any
|
|
|
|
logger = logging.getLogger("SimplePrompt")
|
|
|
|
|
|
def load_json(path: str, default: Any = None) -> Any:
|
|
"""
|
|
加载 JSON 文件
|
|
|
|
Args:
|
|
path: 文件路径
|
|
default: 文件不存在或加载失败时返回的默认值
|
|
|
|
Returns:
|
|
解析后的 JSON 数据，或默认值
|
|
"""
|
|
if not os.path.exists(path):
|
|
return default
|
|
try:
|
|
with open(path, "r", encoding="utf-8") as f:
|
|
return json.load(f)
|
|
except Exception as e:
|
|
logger.error(f"Failed to load JSON {path}: {e}")
|
|
return default
|
|
|
|
|
|
def save_json(path: str, data: Any) -> None:
|
|
"""
|
|
保存数据到 JSON 文件
|
|
|
|
Args:
|
|
path: 文件路径
|
|
data: 要保存的数据
|
|
|
|
Raises:
|
|
Exception: 保存失败时抛出异常
|
|
"""
|
|
try:
|
|
# 确保目录存在
|
|
os.makedirs(os.path.dirname(path), exist_ok=True)
|
|
with open(path, "w", encoding="utf-8") as f:
|
|
json.dump(data, f, indent=4, ensure_ascii=False)
|
|
except Exception as e:
|
|
logger.error(f"Failed to save JSON {path}: {e}")
|
|
raise e
|
|
|
|
|
|
def get_sql_path(path: str) -> str | None:
|
|
"""
|
|
获取用于 SQL 查询的文件路径（将反斜杠转换为正斜杠）
|
|
|
|
Args:
|
|
path: 文件路径
|
|
|
|
Returns:
|
|
格式化后的路径，如果文件不存在则返回 None
|
|
"""
|
|
if os.path.exists(path):
|
|
return path.replace("\\", "/")
|
|
return None
|