Source-checkout quickstart

from pathlib import Path

from core.spec_extractor.openapi.extractor import OpenAPIReader
from core.state.canonical.schema import CanonicalStateSchema
from core.state.factory.state_factory import StateFactory
from core.state.interpretive.schema import Inference, InterpretiveStateSchema


schema_path = Path("tests/data/schemas/registeration.yaml")
schema = OpenAPIReader(root_entity="Registration").read(schema_path)

factory = StateFactory()
canonical = factory.create_canonical_state(schema)

canonical.set_field("registrant.name", "Ada Lovelace")
canonical.set_field("registrant.email", "ada@example.com")
canonical.set_field("status", "draft")

interpretive = factory.create_interpretive_state(canonical)
interpretive.add_inference(
    "registrant.name",
    Inference(
        content="Loaded from canonical registration data",
        mutator_id="quickstart",
        message_id="example-1",
    ),
)

canonical_payload = CanonicalStateSchema(root=canonical.get_all_fields())
interpretive_payload = InterpretiveStateSchema(
    root={
        path: field_state
        for path, field_state in interpretive.get_all_fields().items()
        if field_state is not None
    }
)

assert canonical.get_field("registrant.name") == "Ada Lovelace"
assert interpretive.get_best_value("registrant.name").value == "Ada Lovelace"

print(canonical_payload.model_dump_json(indent=2))
print(interpretive_payload.model_dump_json(indent=2))

poetry run python quickstart.py