DumiFlex
bd2cfe1635
Wire walk_transitive_refs into _build_ctx_with_catalog so the SPA Test Runner populates __wp_catalog__ from the live DB before resolution. Also fixes _resolve_ref to support SnapshotEntry payload.options shape alongside the legacy flat options shape, and updates db_session to accept an injected connection from request.app["wp_db"] for test isolation.
71 lines
2.2 KiB
Python
71 lines
2.2 KiB
Python
"""Shared helpers for wp_api route handlers."""
|
|
from __future__ import annotations
|
|
|
|
from contextlib import contextmanager
|
|
from typing import Any
|
|
|
|
from aiohttp import web
|
|
|
|
from engine.db.connection import get_connection
|
|
from engine.syntax.tokenize import tokenize_text
|
|
from engine.syntax.types import TokenKind
|
|
|
|
|
|
@contextmanager
|
|
def db_session(request: web.Request):
|
|
"""Yield a SQLite connection that is auto-closed on exit.
|
|
|
|
If the aiohttp app has a ``wp_db`` key (injected by tests or the
|
|
application startup), that connection is yielded directly and is NOT
|
|
closed on exit (lifetime owned by the injector). Otherwise, a fresh
|
|
file-based connection is opened via ``get_connection()`` and closed
|
|
when the context manager exits.
|
|
|
|
SQLite WAL mode + per-request connections avoid cross-task locking.
|
|
"""
|
|
injected = request.app.get("wp_db") if hasattr(request, "app") else None
|
|
if injected is not None:
|
|
yield injected
|
|
return
|
|
conn = get_connection()
|
|
try:
|
|
yield conn
|
|
finally:
|
|
conn.close()
|
|
|
|
|
|
def json_ok(data: Any, *, status: int = 200) -> web.Response:
|
|
return web.json_response(data, status=status)
|
|
|
|
|
|
def json_error(message: str, *, status: int) -> web.Response:
|
|
return web.json_response({"error": message}, status=status)
|
|
|
|
|
|
def extract_referenced_uuids(payload: Any) -> set[str]:
|
|
"""Recursively scan strings inside `payload` for `@{8hex}` refs and
|
|
return the set of unique uuids referenced.
|
|
|
|
Used by the SPA Test Runner to determine which wildcards must be
|
|
loaded from the library before resolution can proceed (lazy catalog,
|
|
spec §2.10). Reuses the engine tokenizer so the regex shape stays
|
|
locked in one place."""
|
|
refs: set[str] = set()
|
|
|
|
def _scan(value: Any) -> None:
|
|
if isinstance(value, str):
|
|
for tok in tokenize_text(value):
|
|
if tok.kind == TokenKind.REF:
|
|
uuid = (tok.meta or {}).get("uuid")
|
|
if isinstance(uuid, str):
|
|
refs.add(uuid)
|
|
elif isinstance(value, dict):
|
|
for v in value.values():
|
|
_scan(v)
|
|
elif isinstance(value, list):
|
|
for item in value:
|
|
_scan(item)
|
|
|
|
_scan(payload)
|
|
return refs
|