mindnet/app/core/note_payload.py

165 lines
5.7 KiB
Python
Raw Normal View History

2025-09-09 11:05:32 +02:00
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Modul: app/core/note_payload.py
Version: 1.3.0
Datum: 2025-09-09
Kurzbeschreibung
----------------
Erzeugt den Qdrant-Payload für **Notes**. Neben stabiler Hash-Bildung zur
Idempotenz wird der vollständige Body unter ``fulltext`` persistiert und
der Dateipfad relativ zum Vault gespeichert. Das erlaubt eine verlustfreie
Rekonstruktion im Export (erst ``fulltext``, sonst Chunks).
Wesentliche Features
--------------------
- Hash-Strategie via ENV ``MINDNET_HASH_MODE``:
* ``body`` (Default)
* ``frontmatter``
* ``body+frontmatter``
- Persistenter Volltext im Note-Payload: ``fulltext``
- Pfad-Relativierung (``path``) gegen ``vault_root``
- Optionale Note-Level-Wikilinks (Fallback-Refs)
Beispiele (CLI – Sichtprüfung)
------------------------------
python3 -m app.core.note_payload --from-file ./vault/demo.md --vault-root ./vault --print
"""
2025-09-08 12:22:05 +02:00
2025-09-02 19:51:47 +02:00
from __future__ import annotations
2025-09-09 11:05:32 +02:00
import argparse
2025-09-02 19:51:47 +02:00
import hashlib
2025-09-08 12:22:05 +02:00
import json
import os
2025-09-08 12:42:51 +02:00
from typing import Any, Dict, Optional
2025-09-08 12:22:05 +02:00
2025-09-09 11:05:32 +02:00
try:
from app.core.parser import parse_markdown, extract_wikilinks
except Exception: # pragma: no cover
from .parser import parse_markdown, extract_wikilinks # type: ignore
2025-09-08 12:22:05 +02:00
2025-09-09 11:05:32 +02:00
# ---------------------------------------------------------------------------
# Hashing
# ---------------------------------------------------------------------------
2025-09-08 12:22:05 +02:00
2025-09-09 11:05:32 +02:00
def _canon_frontmatter(fm: Dict[str, Any]) -> str:
"""Kanonische, stabile JSON-Serialisierung der Frontmatter für Hashbildung."""
# Keys sortieren, ASCII nicht erzwingen, um Umlaute stabil zu halten
return json.dumps(fm or {}, ensure_ascii=False, separators=(",", ":"), sort_keys=True)
2025-09-08 12:22:05 +02:00
2025-09-09 11:05:32 +02:00
def compute_hash(*, body: Optional[str], frontmatter: Optional[Dict[str, Any]], mode: Optional[str] = None) -> str:
2025-09-08 12:22:05 +02:00
"""
2025-09-09 11:05:32 +02:00
Berechnet einen Hex-Hash gemäß ``mode``.
2025-09-08 12:22:05 +02:00
2025-09-09 11:05:32 +02:00
mode:
- "body" (Default)
- "frontmatter"
- "body+frontmatter"
2025-09-08 12:22:05 +02:00
"""
2025-09-09 11:05:32 +02:00
if mode is None:
mode = os.environ.get("MINDNET_HASH_MODE", "body").strip().lower()
body = body or ""
fm_s = _canon_frontmatter(frontmatter or {})
h = hashlib.sha256()
if mode == "frontmatter":
h.update(fm_s.encode("utf-8"))
elif mode == "body+frontmatter":
h.update(body.encode("utf-8"))
h.update(b"\n--FM--\n")
h.update(fm_s.encode("utf-8"))
else: # body
h.update(body.encode("utf-8"))
return h.hexdigest()
# ---------------------------------------------------------------------------
# Kernfunktion
# ---------------------------------------------------------------------------
2025-09-08 12:22:05 +02:00
2025-09-08 12:42:51 +02:00
def make_note_payload(parsed: Any, vault_root: Optional[str] = None) -> Dict[str, Any]:
2025-09-08 12:22:05 +02:00
"""
2025-09-09 11:05:32 +02:00
Erzeugt den Payload für eine geparste Note.
Parameters
----------
parsed : Any
Objekt mit Attributen/Keys ``frontmatter``, ``body``, ``path``.
vault_root : Optional[str]
Vault-Wurzel (für Pfad-Relativierung). Wenn ``None``, wird ``path`` unverändert übernommen.
Returns
-------
Dict[str, Any]
Qdrant-Payload für die Notes-Collection.
2025-09-08 12:22:05 +02:00
"""
2025-09-09 11:05:32 +02:00
# "Duck typing": dict oder Objekt akzeptieren
fm = getattr(parsed, "frontmatter", None) or getattr(parsed, "fm", None) or getattr(parsed, "front_matter", None) or parsed.get("frontmatter", {})
body = getattr(parsed, "body", None) or parsed.get("body", "")
path = getattr(parsed, "path", None) or parsed.get("path", "")
2025-09-08 12:22:05 +02:00
2025-09-09 11:05:32 +02:00
# Hash gem. Modus bilden (Default: body)
2025-09-08 12:22:05 +02:00
hash_fulltext = compute_hash(body=body, frontmatter=fm, mode=None)
2025-09-09 11:05:32 +02:00
# Pfad relativieren
rel_path = path
try:
if vault_root:
rel = os.path.relpath(path, vault_root)
rel = rel.replace("\\", "/").lstrip("/") # normalisieren
rel_path = rel
except Exception:
# fail-safe, Pfad ist nicht kritisch für Hash/ID
pass
# Optionale Note-Level-Wikilinks (Fallback, wenn Chunks nicht geliefert werden)
note_level_refs = list(dict.fromkeys(extract_wikilinks(body))) if body else []
2025-09-08 12:22:05 +02:00
payload: Dict[str, Any] = {
2025-09-09 11:05:32 +02:00
"note_id": (fm or {}).get("id") or (fm or {}).get("note_id"),
"title": (fm or {}).get("title"),
"type": (fm or {}).get("type"),
"status": (fm or {}).get("status"),
"created": (fm or {}).get("created"),
"updated": (fm or {}).get("updated"),
"path": rel_path or (fm or {}).get("path"),
"tags": (fm or {}).get("tags"),
2025-09-08 12:22:05 +02:00
"hash_fulltext": hash_fulltext,
2025-09-09 11:05:32 +02:00
# --- WICHTIG: Volltext persistieren ---
"fulltext": body,
# --- Optionaler Fallback für Edge-Ableitung ---
"references": note_level_refs,
2025-09-08 12:22:05 +02:00
}
2025-09-09 11:05:32 +02:00
# Bekannte optionale Frontmatter-Felder durchreichen
for k in ("area", "project", "source", "lang", "slug"):
if k in (fm or {}):
2025-09-08 12:22:05 +02:00
payload[k] = fm[k]
return payload
2025-09-09 11:05:32 +02:00
# ---------------------------------------------------------------------------
# CLI – Sichtprüfung
# ---------------------------------------------------------------------------
def _cli() -> None:
ap = argparse.ArgumentParser(description="Note-Payload aus Markdown erzeugen und anzeigen")
ap.add_argument("--from-file", dest="src", required=True, help="Pfad zur Markdown-Datei")
ap.add_argument("--vault-root", dest="vault_root", default=None, help="Vault-Wurzel zur Pfad-Relativierung")
ap.add_argument("--print", dest="do_print", action="store_true", help="Payload auf stdout ausgeben")
args = ap.parse_args()
parsed = parse_markdown(args.src)
payload = make_note_payload(parsed, vault_root=args.vault_root)
if args.do_print:
print(json.dumps(payload, ensure_ascii=False, indent=2))
2025-09-08 12:42:51 +02:00
2025-09-09 11:05:32 +02:00
if __name__ == "__main__": # pragma: no cover
_cli()