generate_model.py

from glob import glob
from pathlib import Path
from tempfile import NamedTemporaryFile

from datamodel_code_generator import InputFileType, generate, DataModelType
from datamodel_code_generator.model import PythonVersion

SCHEMA_DIR = Path("format/schemas")
OUTPUT_DIR = Path("src/ethdebug/format")

# Ensure output directory exists
OUTPUT_DIR.mkdir(parents=True, exist_ok=True)

# Temporarily rename LICENSE file so it doesn't cause parsing issues
license_path = SCHEMA_DIR / "LICENSE"

with NamedTemporaryFile(delete=False) as tmp_file:
    if license_path.exists():
        license_path.rename(tmp_file.name)
        temp_license_path = Path(tmp_file.name)
    try:
        generate(
            input_=SCHEMA_DIR,
            input_file_type=InputFileType.JsonSchema,
            output=OUTPUT_DIR,
            output_model_type=DataModelType.PydanticV2BaseModel,
            target_python_version=PythonVersion.PY_312,
            allow_extra_fields=False,
            disable_timestamp=True,
            reuse_model=True,
            use_annotated=True,
            field_constraints=True,
            custom_class_name_generator=lambda x: x.title().replace("Ethdebug/Format/", "").replace("/", "_"),
            use_exact_imports=True
        )
    finally:
        # Restore LICENSE file
        if temp_license_path.exists():
            temp_license_path.rename(license_path)