34 lines
1.2 KiB
Python
34 lines
1.2 KiB
Python
# 第 2-9 课完整示例：任务数据读写模块
|
|
#
|
|
# 本模块只负责把任务列表保存到文件，或从文件读取任务列表。
|
|
|
|
import json
|
|
from pathlib import Path
|
|
|
|
|
|
def load_tasks(data_file: Path) -> list[dict[str, object]]:
|
|
"""从 JSON 文件读取任务列表；文件不存在或内容有误时返回空列表。"""
|
|
if not data_file.exists():
|
|
return []
|
|
|
|
try:
|
|
content = data_file.read_text(encoding="utf-8")
|
|
tasks = json.loads(content)
|
|
except (OSError, json.JSONDecodeError):
|
|
# 读取失败或 JSON 格式不正确时，给出安全的空列表，避免程序崩溃。
|
|
return []
|
|
|
|
# JSON 可以保存多种数据。这里只接受列表，避免后续遍历时出现意外错误。
|
|
if not isinstance(tasks, list):
|
|
return []
|
|
|
|
return tasks
|
|
|
|
|
|
def save_tasks(data_file: Path, tasks: list[dict[str, object]]) -> None:
|
|
"""把任务列表保存为 UTF-8 编码的 JSON 文件。"""
|
|
# 父目录不存在时先创建；exist_ok=True 允许程序重复运行。
|
|
data_file.parent.mkdir(parents=True, exist_ok=True)
|
|
content = json.dumps(tasks, ensure_ascii=False, indent=2)
|
|
data_file.write_text(content, encoding="utf-8")
|