29 lines
906 B
Python
29 lines
906 B
Python
"""Provenance of shared answer content, independent of Gemini corrections."""
|
|
from hashlib import sha256
|
|
from pathlib import Path
|
|
|
|
from .json_io import atomic_update_json, read_json
|
|
|
|
|
|
def _path(root):
|
|
return Path(root) / ".copienator" / "answer_origins.json"
|
|
|
|
|
|
def origins_for(root, copy_id, label, pdf):
|
|
entry = read_json(_path(root), default={}).get(str(copy_id), {}).get(label, {})
|
|
if not entry or not Path(pdf).is_file():
|
|
return []
|
|
if entry.get("sha256") != sha256(Path(pdf).read_bytes()).hexdigest():
|
|
return []
|
|
return entry.get("labels", [])
|
|
|
|
|
|
def save_origins(root, copy_id, label, pdf, origins):
|
|
entry = {"labels": list(dict.fromkeys(origins)),
|
|
"sha256": sha256(Path(pdf).read_bytes()).hexdigest()}
|
|
|
|
def update(data):
|
|
data.setdefault(str(copy_id), {})[label] = entry
|
|
|
|
atomic_update_json(_path(root), update, default_factory=dict)
|