重建仓库,重新提交。
This commit is contained in:
88
Utils/IOSAIStorage.py
Normal file
88
Utils/IOSAIStorage.py
Normal file
@@ -0,0 +1,88 @@
|
||||
import json
|
||||
import os
|
||||
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", mode: str = "overwrite") -> Path:
|
||||
file_path = cls._get_iosai_dir() / filename
|
||||
file_path.parent.mkdir(parents=True, exist_ok=True)
|
||||
|
||||
def _load_json():
|
||||
try:
|
||||
return json.loads(file_path.read_text("utf-8"))
|
||||
except Exception:
|
||||
return {} if isinstance(data, dict) else []
|
||||
|
||||
if mode == "merge" and isinstance(data, dict):
|
||||
old = _load_json()
|
||||
if not isinstance(old, dict):
|
||||
old = {}
|
||||
old.update(data)
|
||||
to_write = old
|
||||
elif mode == "append" and isinstance(data, list):
|
||||
old = _load_json()
|
||||
if not isinstance(old, list):
|
||||
old = []
|
||||
old.extend(data)
|
||||
to_write = old
|
||||
else:
|
||||
to_write = data # 覆盖
|
||||
|
||||
# 原子写入
|
||||
tmp = file_path.with_suffix(file_path.suffix + ".tmp")
|
||||
with open(tmp, "w", encoding="utf-8") as f:
|
||||
json.dump(to_write, f, ensure_ascii=False, indent=2)
|
||||
os.replace(tmp, file_path)
|
||||
print(f"[IOSAIStorage] 已写入: {file_path}")
|
||||
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
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user