refactor(codegen): separate v1 schema semantics (#140)

Author: PsiACE

scripts/_schema_semantics.py

@@ -0,0 +1,46 @@
+from __future__ import annotations
+
+from dataclasses import dataclass
+from importlib import import_module
+from pathlib import Path
+from typing import Protocol, cast
+
+ROOT = Path(__file__).resolve().parents[1]
+
+DEFAULT_SEMANTICS_MODULE = "scripts.gen_schema_v1"
+
+
+@dataclass(frozen=True, slots=True)
+class SchemaSemantics:
+    schema_json: Path
+    version_file: Path
+    schema_out: Path
+    model_name_map: dict[str, str]
+    compatibility_aliases: str = ""
+
+
+class _SemanticsModule(Protocol):
+    SEMANTICS: SchemaSemantics
+
+
+def get_default_schema_semantics() -> SchemaSemantics:
+    module = cast(_SemanticsModule, import_module(DEFAULT_SEMANTICS_MODULE))
+    return module.SEMANTICS
+
+
+def inline_model_ref(definition: str, *steps: tuple[str, int | None]) -> str:
+    ref = f"#/$defs/{definition}"
+    for keyword, index in steps:
+        ref += f"#-datamodel-code-generator-#-{keyword}-#-special-#"
+        if index is not None:
+            ref += f"/{index}"
+    return ref
+
+
+def variant_model_map(
+    definition: str,
+    keyword: str,
+    branch: str,
+    names: tuple[str, ...],
+) -> dict[str, str]:
+    return {inline_model_ref(definition, (keyword, index), (branch, None)): name for index, name in enumerate(names)}