101 lines
3.3 KiB
Python
101 lines
3.3 KiB
Python
import os
|
||
import json
|
||
from pathlib import Path
|
||
|
||
|
||
class JsonUtils:
|
||
@staticmethod
|
||
def _normalize_filename(filename: str) -> str:
|
||
"""
|
||
确保文件名以 .json 结尾
|
||
"""
|
||
if not filename.endswith(".json"):
|
||
filename = f"{filename}.json"
|
||
return filename
|
||
|
||
@staticmethod
|
||
def _get_data_path(filename: str) -> str:
|
||
"""
|
||
根据文件名生成 data 目录下的完整路径
|
||
"""
|
||
filename = JsonUtils._normalize_filename(filename)
|
||
base_dir = os.path.abspath(os.path.dirname(os.path.dirname(__file__))) # 当前项目根目录
|
||
data_dir = os.path.join(base_dir, "data")
|
||
Path(data_dir).mkdir(parents=True, exist_ok=True) # 确保 data 目录存在
|
||
return os.path.join(data_dir, filename)
|
||
|
||
@staticmethod
|
||
def read_json(filename: str) -> dict:
|
||
"""
|
||
读取 JSON 文件，返回字典
|
||
如果文件不存在，返回空字典
|
||
"""
|
||
file_path = JsonUtils._get_data_path(filename)
|
||
try:
|
||
if not os.path.exists(file_path):
|
||
return {}
|
||
with open(file_path, "r", encoding="utf-8") as f:
|
||
data = json.load(f)
|
||
return data if isinstance(data, dict) else {}
|
||
except Exception as e:
|
||
print(f"读取 JSON 文件失败: {e}")
|
||
return {}
|
||
|
||
@staticmethod
|
||
def write_json(filename: str, data: dict, overwrite: bool = True) -> bool:
|
||
"""
|
||
将字典写入 JSON 文件
|
||
:param filename: 文件名（不用写后缀，自动补 .json）
|
||
:param data: 要写入的字典
|
||
:param overwrite: True=覆盖写，False=合并更新
|
||
"""
|
||
file_path = JsonUtils._get_data_path(filename)
|
||
try:
|
||
if not overwrite and os.path.exists(file_path):
|
||
with open(file_path, "r", encoding="utf-8") as f:
|
||
old_data = json.load(f)
|
||
if not isinstance(old_data, dict):
|
||
old_data = {}
|
||
old_data.update(data)
|
||
data = old_data
|
||
|
||
with open(file_path, "w", encoding="utf-8") as f:
|
||
json.dump(data, f, ensure_ascii=False, indent=4)
|
||
return True
|
||
except Exception as e:
|
||
print(f"写入 JSON 文件失败: {e}")
|
||
return False
|
||
|
||
@staticmethod
|
||
def update_json(filename: str, new_data: dict) -> bool:
|
||
"""
|
||
修改 JSON 文件：
|
||
- 如果 key 已存在，则修改其值
|
||
- 如果 key 不存在，则新增
|
||
"""
|
||
try:
|
||
data = JsonUtils.read_json(filename)
|
||
if not isinstance(data, dict):
|
||
data = {}
|
||
data.update(new_data)
|
||
return JsonUtils.write_json(filename, data)
|
||
except Exception as e:
|
||
print(f"更新 JSON 文件失败: {e}")
|
||
return False
|
||
|
||
@staticmethod
|
||
def delete_json_key(filename: str, key: str) -> bool:
|
||
"""
|
||
删除 JSON 文件中的某个 key
|
||
"""
|
||
try:
|
||
data = JsonUtils.read_json(filename)
|
||
if not isinstance(data, dict):
|
||
data = {}
|
||
if key in data:
|
||
del data[key]
|
||
return JsonUtils.write_json(filename, data)
|
||
except Exception as e:
|
||
print(f"删除 JSON key 失败: {e}")
|
||
return False
|