67 lines
2.1 KiB
Python
67 lines
2.1 KiB
Python
import json
|
|
from pathlib import Path
|
|
|
|
|
|
class IOSAIStorage:
|
|
|
|
@staticmethod
|
|
def _get_iosai_dir() -> Path:
|
|
"""获取 C:/Users/<用户名>/IOSAI/ 目录"""
|
|
user_dir = Path.home()
|
|
iosai_dir = user_dir / "IOSAI"
|
|
iosai_dir.mkdir(parents=True, exist_ok=True)
|
|
return iosai_dir
|
|
|
|
@classmethod
|
|
def save(cls, data: dict | list, filename: str = "data.json") -> Path:
|
|
"""
|
|
存储数据到 C:/Users/<用户名>/IOSAI/filename
|
|
"""
|
|
file_path = cls._get_iosai_dir() / filename
|
|
try:
|
|
with open(file_path, "w", encoding="utf-8") as f:
|
|
json.dump(data, f, ensure_ascii=False, indent=2)
|
|
print(f"[IOSAIStorage] 已保存到: {file_path}")
|
|
except Exception as e:
|
|
print(f"[IOSAIStorage] 写入失败: {e}")
|
|
return file_path
|
|
|
|
@classmethod
|
|
def load(cls, filename: str = "data.json") -> dict | list | None:
|
|
"""
|
|
从 C:/Users/<用户名>/IOSAI/filename 读取数据
|
|
"""
|
|
file_path = cls._get_iosai_dir() / filename
|
|
if not file_path.exists():
|
|
print(f"[IOSAIStorage] 文件不存在: {file_path}")
|
|
return {}
|
|
|
|
try:
|
|
with open(file_path, "r", encoding="utf-8") as f:
|
|
return json.load(f)
|
|
except Exception as e:
|
|
print(f"[IOSAIStorage] 读取失败: {e}")
|
|
return {}
|
|
|
|
|
|
@classmethod
|
|
def overwrite(cls, data: dict | list, filename: str = "data.json") -> Path:
|
|
"""
|
|
强制覆盖写入数据到 C:/Users/<用户名>/IOSAI/filename
|
|
（无论是否存在，都会写入）
|
|
"""
|
|
file_path = cls._get_iosai_dir() / filename
|
|
try:
|
|
# "w" 模式本身就是覆盖，但这里单独做一个方法
|
|
with open(file_path, "w", encoding="utf-8") as f:
|
|
json.dump(data, f, ensure_ascii=False, indent=2)
|
|
print(f"[IOSAIStorage] 已覆盖写入: {file_path}")
|
|
except Exception as e:
|
|
print(f"[IOSAIStorage] 覆盖失败: {e}")
|
|
return file_path
|
|
|
|
|
|
|
|
|
|
|