62 lines
2.4 KiB
Python
62 lines
2.4 KiB
Python
"""Dump ratified engine enum values for Workbench UI coverage tests.
|
|
|
|
Read-only helper: parses source with Python AST and writes JSON to stdout.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import ast
|
|
import json
|
|
from pathlib import Path
|
|
|
|
|
|
ROOT = Path(__file__).resolve().parents[1]
|
|
|
|
|
|
def enum_values(path: Path, class_name: str) -> list[str]:
|
|
module = ast.parse(path.read_text(encoding="utf-8"))
|
|
for node in module.body:
|
|
if isinstance(node, ast.ClassDef) and node.name == class_name:
|
|
values: list[str] = []
|
|
for statement in node.body:
|
|
if (
|
|
isinstance(statement, ast.Assign)
|
|
and isinstance(statement.value, ast.Constant)
|
|
and isinstance(statement.value.value, str)
|
|
):
|
|
values.append(statement.value.value)
|
|
return values
|
|
raise SystemExit(f"enum class not found: {class_name}")
|
|
|
|
|
|
def literal_values(path: Path, name: str) -> list[str]:
|
|
module = ast.parse(path.read_text(encoding="utf-8"))
|
|
for node in module.body:
|
|
if isinstance(node, ast.Assign):
|
|
if not any(isinstance(target, ast.Name) and target.id == name for target in node.targets):
|
|
continue
|
|
value = node.value
|
|
if (
|
|
isinstance(value, ast.Subscript)
|
|
and isinstance(value.value, ast.Name)
|
|
and value.value.id == "Literal"
|
|
):
|
|
items = value.slice.elts if isinstance(value.slice, ast.Tuple) else [value.slice]
|
|
return [
|
|
item.value
|
|
for item in items
|
|
if isinstance(item, ast.Constant) and isinstance(item.value, str)
|
|
]
|
|
raise SystemExit(f"literal alias not found: {name}")
|
|
|
|
|
|
snapshot = {
|
|
"EpistemicState": enum_values(ROOT / "core" / "epistemic_state.py", "EpistemicState"),
|
|
"GroundingSource": literal_values(ROOT / "core" / "epistemic_state.py", "GroundingSource"),
|
|
"NormativeClearance": enum_values(ROOT / "core" / "epistemic_state.py", "NormativeClearance"),
|
|
"ReviewState": literal_values(ROOT / "teaching" / "proposals.py", "ReviewState"),
|
|
"ReplayDivergenceSeverity": literal_values(ROOT / "workbench" / "schemas.py", "ReplayDivergenceSeverity"),
|
|
"ReplayStatus": literal_values(ROOT / "workbench" / "schemas.py", "ReplayStatus"),
|
|
}
|
|
|
|
print(json.dumps(snapshot, indent=2, sort_keys=True))
|