87 lines
3.3 KiB
Python
87 lines
3.3 KiB
Python
# ============================================================
|
||
# 数据包（moduleId: state-packs, 可重生 ✅）
|
||
# 把「一套完整世界数据」封装为可重放的 JSON 包：
|
||
# - 演示厂 = server/data/packs/demo.json（由脚本从种子生成）
|
||
# - 行业种子 / 客户现场数据都可做成包，data.reset 即「重放数据包」
|
||
# 日期重基：包内记录 baseDate（生成日），加载时把所有 YYYY-MM-DD
|
||
# 字面日期整体平移到「今天」，保证任意运行日行为一致（黄金测试稳定）。
|
||
# ============================================================
|
||
from __future__ import annotations
|
||
|
||
import json
|
||
import os
|
||
import re
|
||
from datetime import datetime, timedelta
|
||
from typing import Any
|
||
|
||
PACKS_DIR = os.path.join("server", "data", "packs")
|
||
_DATE_RE = re.compile(r"^(\d{4}-\d{2}-\d{2})")
|
||
|
||
|
||
def export_pack(world: dict[str, Any], path: str, *, name: str = "", description: str = "") -> None:
|
||
"""把世界数据导出为数据包（含日期基准，供加载时重基）。"""
|
||
pack = {
|
||
"packVersion": 1,
|
||
"name": name or os.path.splitext(os.path.basename(path))[0],
|
||
"description": description,
|
||
"baseDate": datetime.now().strftime("%Y-%m-%d"),
|
||
"world": world,
|
||
}
|
||
os.makedirs(os.path.dirname(path) or ".", exist_ok=True)
|
||
with open(path, "w", encoding="utf-8") as f:
|
||
json.dump(pack, f, ensure_ascii=False, indent=1)
|
||
|
||
|
||
def _shift_dates(obj: Any, delta_days: int) -> Any:
|
||
"""递归平移所有 YYYY-MM-DD 开头的字符串日期。"""
|
||
if delta_days == 0:
|
||
return obj
|
||
if isinstance(obj, dict):
|
||
return {k: _shift_dates(v, delta_days) for k, v in obj.items()}
|
||
if isinstance(obj, list):
|
||
return [_shift_dates(v, delta_days) for v in obj]
|
||
if isinstance(obj, str):
|
||
m = _DATE_RE.match(obj)
|
||
if m:
|
||
try:
|
||
d = datetime.strptime(m.group(1), "%Y-%m-%d") + timedelta(days=delta_days)
|
||
return d.strftime("%Y-%m-%d") + obj[10:]
|
||
except ValueError:
|
||
return obj
|
||
return obj
|
||
|
||
|
||
def load_pack(path: str) -> dict[str, Any]:
|
||
"""读取数据包并把日期重基到今天，返回 world dict。"""
|
||
with open(path, "r", encoding="utf-8") as f:
|
||
pack = json.load(f)
|
||
world = pack.get("world") or {}
|
||
base = pack.get("baseDate")
|
||
if base:
|
||
try:
|
||
delta = (datetime.now().date() - datetime.strptime(base, "%Y-%m-%d").date()).days
|
||
world = _shift_dates(world, delta)
|
||
except ValueError:
|
||
pass
|
||
return world
|
||
|
||
|
||
def list_packs() -> list[dict[str, Any]]:
|
||
"""列出可用数据包（名称/描述/路径）。"""
|
||
out: list[dict[str, Any]] = []
|
||
if not os.path.isdir(PACKS_DIR):
|
||
return out
|
||
for fn in sorted(os.listdir(PACKS_DIR)):
|
||
if not fn.endswith(".json"):
|
||
continue
|
||
path = os.path.join(PACKS_DIR, fn)
|
||
try:
|
||
with open(path, "r", encoding="utf-8") as f:
|
||
head = json.load(f)
|
||
out.append({"file": fn, "name": head.get("name") or fn,
|
||
"description": head.get("description") or "",
|
||
"baseDate": head.get("baseDate") or "", "path": path})
|
||
except (json.JSONDecodeError, OSError):
|
||
continue
|
||
return out
|