254 lines
7.4 KiB
Python
254 lines
7.4 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
"""
|
|
Update State - 状态更新脚本
|
|
|
|
用途：
|
|
- 更新 state.json 中的各种状态信息
|
|
- 支持多种更新操作：添加审查报告、更新实体、修改进度等
|
|
- 提供 CLI 接口供 Skills 调用
|
|
|
|
典型用法：
|
|
python update_state.py --project-root <root> --add-review "1-10" "审查报告/第 1-10 章审查报告.md"
|
|
python update_state.py --project-root <root> --update-entity 角色 萧炎 '{"境界": "斗皇"}'
|
|
python update_state.py --project-root <root> --set-chapter-meta 100 '{"word_count": 2300}'
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import argparse
|
|
import json
|
|
import os
|
|
import sys
|
|
from pathlib import Path
|
|
from datetime import datetime
|
|
|
|
# 添加 scripts 目录到 sys.path
|
|
scripts_dir = Path(__file__).resolve().parent
|
|
sys.path.insert(0, str(scripts_dir))
|
|
|
|
from runtime_compat import enable_windows_utf8_stdio, normalize_windows_path
|
|
|
|
|
|
def load_state(state_file: Path) -> dict:
|
|
"""加载 state.json"""
|
|
if not state_file.exists():
|
|
raise FileNotFoundError(f"state.json not found: {state_file}")
|
|
|
|
with open(state_file, 'r', encoding='utf-8') as f:
|
|
return json.load(f)
|
|
|
|
|
|
def save_state(state_file: Path, state: dict) -> None:
|
|
"""保存 state.json（带文件锁保护）"""
|
|
try:
|
|
from filelock import FileLock
|
|
except ImportError:
|
|
FileLock = None
|
|
|
|
lock_file = state_file.with_suffix('.lock')
|
|
|
|
if FileLock:
|
|
with FileLock(str(lock_file)):
|
|
_write_state(state_file, state)
|
|
else:
|
|
_write_state(state_file, state)
|
|
|
|
|
|
def _write_state(state_file: Path, state: dict) -> None:
|
|
"""实际写入文件"""
|
|
import tempfile
|
|
import shutil
|
|
|
|
# 原子写入：先写临时文件，再替换
|
|
dir_path = state_file.parent
|
|
dir_path.mkdir(parents=True, exist_ok=True)
|
|
|
|
fd, tmp_path = tempfile.mkstemp(dir=str(dir_path), suffix='.tmp')
|
|
try:
|
|
with os.fdopen(fd, 'w', encoding='utf-8') as f:
|
|
json.dump(state, f, ensure_ascii=False, indent=2)
|
|
f.write('\n')
|
|
|
|
shutil.move(str(tmp_path), str(state_file))
|
|
except Exception:
|
|
if os.path.exists(tmp_path):
|
|
os.unlink(tmp_path)
|
|
raise
|
|
|
|
|
|
def cmd_add_review(args: argparse.Namespace) -> int:
|
|
"""
|
|
添加审查报告记录
|
|
|
|
用法：
|
|
--add-review "章节范围" "报告路径"
|
|
"""
|
|
state_file = args.project_root / ".noma" / "novel_data" / "state.json"
|
|
state = load_state(state_file)
|
|
|
|
chapter_range = args.chapter_range
|
|
report_path = args.report_path
|
|
|
|
# 初始化 review_checkpoints
|
|
state.setdefault("review_checkpoints", [])
|
|
|
|
# 添加新记录
|
|
new_record = {
|
|
"chapter_range": chapter_range,
|
|
"report_path": report_path,
|
|
"timestamp": datetime.now().isoformat(),
|
|
"status": "completed"
|
|
}
|
|
|
|
state["review_checkpoints"].append(new_record)
|
|
|
|
# 保存
|
|
save_state(state_file, state)
|
|
|
|
print(f"✅ Added review checkpoint: {chapter_range} -> {report_path}")
|
|
return 0
|
|
|
|
|
|
def cmd_update_entity(args: argparse.Namespace) -> int:
|
|
"""
|
|
更新实体属性
|
|
|
|
用法：
|
|
--update-entity <类型> <ID> <JSON 数据>
|
|
"""
|
|
state_file = args.project_root / ".noma" / "novel_data" / "state.json"
|
|
state = load_state(state_file)
|
|
|
|
entity_type = args.entity_type # 角色/地点/物品/势力
|
|
entity_id = args.entity_id
|
|
updates = json.loads(args.data)
|
|
|
|
# 确保 entities_v3 存在
|
|
state.setdefault("entities_v3", {})
|
|
state["entities_v3"].setdefault(entity_type, {})
|
|
|
|
# 获取或创建实体
|
|
if entity_id not in state["entities_v3"][entity_type]:
|
|
state["entities_v3"][entity_type][entity_id] = {
|
|
"id": entity_id,
|
|
"name": entity_id,
|
|
"type": entity_type,
|
|
"tier": "次要"
|
|
}
|
|
|
|
# 更新属性
|
|
entity = state["entities_v3"][entity_type][entity_id]
|
|
entity.update(updates)
|
|
|
|
# 保存
|
|
save_state(state_file, state)
|
|
|
|
print(f"✅ Updated entity: {entity_type}/{entity_id}")
|
|
print(f" Fields updated: {list(updates.keys())}")
|
|
return 0
|
|
|
|
|
|
def cmd_set_chapter_meta(args: argparse.Namespace) -> int:
|
|
"""
|
|
设置章节元数据
|
|
|
|
用法：
|
|
--set-chapter-meta <章节号> <JSON 数据>
|
|
"""
|
|
state_file = args.project_root / ".noma" / "novel_data" / "state.json"
|
|
state = load_state(state_file)
|
|
|
|
chapter_num = int(args.chapter)
|
|
meta = json.loads(args.data)
|
|
|
|
# 确保 chapter_meta 存在
|
|
state.setdefault("chapter_meta", {})
|
|
|
|
# 转换为字符串键
|
|
chapter_key = str(chapter_num)
|
|
|
|
# 合并现有数据
|
|
if chapter_key not in state["chapter_meta"]:
|
|
state["chapter_meta"][chapter_key] = {}
|
|
|
|
state["chapter_meta"][chapter_key].update(meta)
|
|
|
|
# 保存
|
|
save_state(state_file, state)
|
|
|
|
print(f"✅ Set chapter meta for chapter {chapter_num}")
|
|
print(f" Keys: {list(meta.keys())}")
|
|
return 0
|
|
|
|
|
|
def cmd_set_progress(args: argparse.Namespace) -> int:
|
|
"""
|
|
设置项目进度
|
|
|
|
用法：
|
|
--set-progress --current-chapter <N>
|
|
"""
|
|
state_file = args.project_root / ".noma" / "novel_data" / "state.json"
|
|
state = load_state(state_file)
|
|
|
|
# 更新 project 信息
|
|
state.setdefault("project", {})
|
|
state["project"]["current_chapter"] = args.current_chapter
|
|
state["project"]["last_updated"] = datetime.now().isoformat()
|
|
|
|
# 保存
|
|
save_state(state_file, state)
|
|
|
|
print(f"✅ Updated progress: current_chapter = {args.current_chapter}")
|
|
return 0
|
|
|
|
|
|
def main() -> None:
|
|
parser = argparse.ArgumentParser(description="Update state.json")
|
|
parser.add_argument("--project-root", type=str, required=True, help="项目根目录")
|
|
|
|
subparsers = parser.add_subparsers(dest="command", help="命令")
|
|
|
|
# add-review 命令
|
|
p_review = subparsers.add_parser("add-review", help="添加审查报告记录")
|
|
p_review.add_argument("chapter_range", help="章节范围 (如 '1-10')")
|
|
p_review.add_argument("report_path", help="报告文件路径")
|
|
p_review.set_defaults(func=cmd_add_review)
|
|
|
|
# update-entity 命令
|
|
p_entity = subparsers.add_parser("update-entity", help="更新实体属性")
|
|
p_entity.add_argument("entity_type", help="实体类型 (角色/地点/物品/势力)")
|
|
p_entity.add_argument("entity_id", help="实体 ID")
|
|
p_entity.add_argument("data", help="JSON 格式的属性数据")
|
|
p_entity.set_defaults(func=cmd_update_entity)
|
|
|
|
# set-chapter-meta 命令
|
|
p_meta = subparsers.add_parser("set-chapter-meta", help="设置章节元数据")
|
|
p_meta.add_argument("chapter", help="章节号")
|
|
p_meta.add_argument("data", help="JSON 格式的元数据")
|
|
p_meta.set_defaults(func=cmd_set_chapter_meta)
|
|
|
|
# set-progress 命令
|
|
p_progress = subparsers.add_parser("set-progress", help="设置项目进度")
|
|
p_progress.add_argument("--current-chapter", type=int, required=True, help="当前章节号")
|
|
p_progress.set_defaults(func=cmd_set_progress)
|
|
|
|
# 解析参数
|
|
args = parser.parse_args()
|
|
|
|
if not args.command:
|
|
parser.print_help()
|
|
sys.exit(1)
|
|
|
|
# 规范化路径
|
|
args.project_root = normalize_windows_path(args.project_root).resolve()
|
|
|
|
# 执行命令
|
|
enable_windows_utf8_stdio(skip_in_pytest=True)
|
|
sys.exit(args.func(args))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|