Add --jsonschema-version and --openapi-version CLI options

Author: koxudaxi

src/datamodel_code_generator/__init__.py

@@ -43,9 +43,11 @@
     GraphQLScope,
     InputFileType,
     InputModelRefStrategy,
+    JsonSchemaVersion,
     ModuleSplitMode,
     NamingStrategy,
     OpenAPIScope,
+    OpenAPIVersion,
     ReadOnlyWriteOnlyModelType,
     ReuseScope,
     TargetPydanticVersion,
@@ -288,6 +290,42 @@ def is_schema(data: dict) -> bool:
     return isinstance(data.get("properties"), dict)
 
 
+def detect_jsonschema_version(data: dict[str, Any]) -> JsonSchemaVersion:
+    """Detect JSON Schema version from $schema field.
+
+    Returns Auto if version cannot be detected, allowing all features.
+    """
+    schema = data.get("$schema", "")
+    if not isinstance(schema, str):
+        return JsonSchemaVersion.Auto
+    if "draft-04" in schema:
+        return JsonSchemaVersion.Draft04
+    if "draft-07" in schema:
+        return JsonSchemaVersion.Draft07
+    if "2019-09" in schema:
+        return JsonSchemaVersion.Draft201909
+    if "2020-12" in schema:
+        return JsonSchemaVersion.Draft202012
+    return JsonSchemaVersion.Auto
+
+
+def detect_openapi_version(data: dict[str, Any]) -> OpenAPIVersion:
+    """Detect OpenAPI version from openapi/swagger field.
+
+    Returns Auto if version cannot be detected, allowing all features.
+    """
+    if "swagger" in data:
+        return OpenAPIVersion.V20
+    openapi = data.get("openapi", "")
+    if not isinstance(openapi, str):
+        return OpenAPIVersion.Auto
+    if openapi.startswith("3.1"):
+        return OpenAPIVersion.V31
+    if openapi.startswith("3.0"):
+        return OpenAPIVersion.V30
+    return OpenAPIVersion.Auto
+
+
 RAW_DATA_TYPES: list[InputFileType] = [
     InputFileType.Json,
     InputFileType.Yaml,
@@ -338,6 +376,35 @@ def __init__(
         super().__init__(message=message)
 
 
+class SchemaVersionError(Error):
+    """Base exception for schema version-related errors."""
+
+
+class UnsupportedVersionError(SchemaVersionError):
+    """Raised when an unsupported schema version is encountered."""
+
+    def __init__(self, version: str, supported: list[str]) -> None:
+        """Initialize with version and list of supported versions."""
+        self.version = version
+        self.supported = supported
+        message = f"Unsupported schema version: {version}. Supported versions: {', '.join(supported)}"
+        super().__init__(message=message)
+
+
+class SchemaValidationError(SchemaVersionError):
+    """Raised when strict validation fails for a schema construct."""
+
+    def __init__(self, message: str, version: str, feature: str) -> None:
+        """Initialize with message, version, and feature name."""
+        self.version = version
+        self.feature = feature
+        super().__init__(message=f"[{version}] {message}")
+
+
+class VersionMismatchWarning(UserWarning):
+    """Warning for version-specific feature usage in wrong version."""
+
+
 class SchemaParseError(Error):
     """Raised when an error occurs during schema parsing with path context."""
 
@@ -966,17 +1033,25 @@ def __getattr__(name: str) -> Any:
     "InputModelRefStrategy",
     "InvalidClassNameError",
     "InvalidFileFormatError",
+    "JsonSchemaVersion",
     "LiteralType",
     "ModuleSplitMode",
     "NamingStrategy",
     "OpenAPIScope",
+    "OpenAPIVersion",
     "PythonVersion",
     "PythonVersionMin",
     "ReadOnlyWriteOnlyModelType",
     "ReuseScope",
     "SchemaParseError",
+    "SchemaValidationError",
+    "SchemaVersionError",
     "TargetPydanticVersion",
+    "UnsupportedVersionError",
+    "VersionMismatchWarning",
     "clear_dynamic_models_cache",  # noqa: F822
+    "detect_jsonschema_version",
+    "detect_openapi_version",
     "generate",
     "generate_dynamic_models",  # noqa: F822
 ]

src/datamodel_code_generator/__main__.py

@@ -63,9 +63,11 @@
     InputFileType,
     InputModelRefStrategy,
     InvalidClassNameError,
+    JsonSchemaVersion,
     ModuleSplitMode,
     NamingStrategy,
     OpenAPIScope,
+    OpenAPIVersion,
     ReadOnlyWriteOnlyModelType,
     ReuseScope,
     TargetPydanticVersion,
@@ -491,6 +493,8 @@ def validate_class_name_affix_scope(cls, v: str | ClassNameAffixScope | None) ->
     input_model: Optional[list[str]] = None  # noqa: UP045
     input_model_ref_strategy: Optional[InputModelRefStrategy] = None  # noqa: UP045
     input_file_type: InputFileType = InputFileType.Auto
+    jsonschema_version: JsonSchemaVersion = JsonSchemaVersion.Auto
+    openapi_version: OpenAPIVersion = OpenAPIVersion.Auto
     output_model_type: DataModelType = DataModelType.PydanticBaseModel
     output: Optional[Path] = None  # noqa: UP045
     check: bool = False
@@ -938,6 +942,8 @@ def run_generate_from_config(  # noqa: PLR0913, PLR0917
     result = generate(
         input_=input_,
         input_file_type=config.input_file_type,
+        jsonschema_version=config.jsonschema_version,
+        openapi_version=config.openapi_version,
         output=output,
         output_model_type=config.output_model_type,
         target_python_version=config.target_python_version,

src/datamodel_code_generator/_types/generate_config_dict.py

@@ -24,9 +24,11 @@
         FieldTypeCollisionStrategy,
         GraphQLScope,
         InputFileType,
+        JsonSchemaVersion,
         ModuleSplitMode,
         NamingStrategy,
         OpenAPIScope,
+        OpenAPIVersion,
         ReadOnlyWriteOnlyModelType,
         ReuseScope,
         StrictTypes,
@@ -41,6 +43,8 @@
 class GenerateConfigDict(TypedDict):
     input_filename: NotRequired[str | None]
     input_file_type: NotRequired[InputFileType]
+    jsonschema_version: NotRequired[JsonSchemaVersion]
+    openapi_version: NotRequired[OpenAPIVersion]
     output: NotRequired[Path | None]
     output_model_type: NotRequired[DataModelType]
     target_python_version: NotRequired[PythonVersion]

src/datamodel_code_generator/_types/parser_config_dicts.py

@@ -19,8 +19,10 @@
         CollapseRootModelsNameStrategy,
         DataclassArguments,
         FieldTypeCollisionStrategy,
+        JsonSchemaVersion,
         NamingStrategy,
         OpenAPIScope,
+        OpenAPIVersion,
         ReadOnlyWriteOnlyModelType,
         ReuseScope,
         StrictTypes,
@@ -45,6 +47,8 @@ class ParserConfigDict(TypedDict):
     data_model_root_type: NotRequired[type[DataModel]]
     data_type_manager_type: NotRequired[type[DataTypeManager]]
     data_model_field_type: NotRequired[type[DataModelFieldBase]]
+    jsonschema_version: NotRequired[JsonSchemaVersion]
+    openapi_version: NotRequired[OpenAPIVersion]
     base_class: NotRequired[str | None]
     base_class_map: NotRequired[dict[str, str] | None]
     additional_imports: NotRequired[list[str] | None]

src/datamodel_code_generator/arguments.py

@@ -26,9 +26,11 @@
     FieldTypeCollisionStrategy,
     InputFileType,
     InputModelRefStrategy,
+    JsonSchemaVersion,
     ModuleSplitMode,
     NamingStrategy,
     OpenAPIScope,
+    OpenAPIVersion,
     ReadOnlyWriteOnlyModelType,
     ReuseScope,
     StrictTypes,
@@ -146,6 +148,24 @@ def start_section(self, heading: str | None) -> None:
     ),
     choices=[i.value for i in InputFileType],
 )
+base_options.add_argument(
+    "--jsonschema-version",
+    help=(
+        "JSON Schema version (default: auto). "
+        "When 'auto', version is detected from $schema field. "
+        "Specify explicitly to override detection or when $schema is missing."
+    ),
+    choices=[v.value for v in JsonSchemaVersion],
+)
+base_options.add_argument(
+    "--openapi-version",
+    help=(
+        "OpenAPI version (default: auto). "
+        "When 'auto', version is detected from openapi/swagger field. "
+        "Specify explicitly to override detection."
+    ),
+    choices=[v.value for v in OpenAPIVersion],
+)
 base_options.add_argument(
     "--output",
     help="Output file (default: stdout)",

src/datamodel_code_generator/cli_options.py

@@ -52,6 +52,9 @@ class CLIOptionMeta:
     "--profile",
     "--no-color",
     "--generate-prompt",
+    # Schema version options - placeholders for future version-specific behavior
+    "--jsonschema-version",
+    "--openapi-version",
 })
 
 # Backward compatibility alias

src/datamodel_code_generator/config.py

@@ -22,9 +22,11 @@
     FieldTypeCollisionStrategy,
     GraphQLScope,
     InputFileType,
+    JsonSchemaVersion,
     ModuleSplitMode,
     NamingStrategy,
     OpenAPIScope,
+    OpenAPIVersion,
     ReadOnlyWriteOnlyModelType,
     ReuseScope,
     TargetPydanticVersion,
@@ -78,6 +80,8 @@ class Config:
 
     input_filename: str | None = None
     input_file_type: InputFileType = InputFileType.Auto
+    jsonschema_version: JsonSchemaVersion = JsonSchemaVersion.Auto
+    openapi_version: OpenAPIVersion = OpenAPIVersion.Auto
     output: Path | None = None
     output_model_type: DataModelType = DataModelType.PydanticBaseModel
     target_python_version: PythonVersion = PythonVersionMin
@@ -224,6 +228,8 @@ class Config:
     data_model_root_type: type[DataModel] = pydantic_model.CustomRootType
     data_type_manager_type: type[DataTypeManager] = pydantic_model.DataTypeManager
     data_model_field_type: type[DataModelFieldBase] = pydantic_model.DataModelField
+    jsonschema_version: JsonSchemaVersion = JsonSchemaVersion.Auto
+    openapi_version: OpenAPIVersion = OpenAPIVersion.Auto
     base_class: str | None = None
     base_class_map: dict[str, str] | None = None
     additional_imports: list[str] | None = None

src/datamodel_code_generator/enums.py

@@ -240,6 +240,33 @@ class StrictTypes(Enum):
     bool = "bool"
 
 
+class JsonSchemaVersion(Enum):
+    """JSON Schema draft versions.
+
+    Used to specify which JSON Schema draft to use for parsing and validation.
+    Different drafts have different features and semantics.
+    """
+
+    Auto = "auto"
+    Draft04 = "draft-04"
+    Draft07 = "draft-07"
+    Draft201909 = "2019-09"
+    Draft202012 = "2020-12"
+
+
+class OpenAPIVersion(Enum):
+    """OpenAPI specification versions.
+
+    Used to specify which OpenAPI version to use for parsing.
+    Different versions have different schema semantics (e.g., nullable handling).
+    """
+
+    Auto = "auto"
+    V20 = "2.0"
+    V30 = "3.0"
+    V31 = "3.1"
+
+
 __all__ = [
     "DEFAULT_SHARED_MODULE_NAME",
     "MAX_VERSION",
@@ -256,9 +283,11 @@ class StrictTypes(Enum):
     "GraphQLScope",
     "InputFileType",
     "InputModelRefStrategy",
+    "JsonSchemaVersion",
     "ModuleSplitMode",
     "NamingStrategy",
     "OpenAPIScope",
+    "OpenAPIVersion",
     "ReadOnlyWriteOnlyModelType",
     "ReuseScope",
     "StrictTypes",

tests/data/expected/main/input_model/config_class.py

@@ -77,6 +77,11 @@ class DataclassArguments(TypedDict):
 ]
 
 
+JsonSchemaVersion: TypeAlias = Literal[
+    'auto', 'draft-04', 'draft-07', '2019-09', '2020-12'
+]
+
+
 LiteralType: TypeAlias = Literal['all', 'one', 'none']
 
 
@@ -93,6 +98,9 @@ class DataclassArguments(TypedDict):
 ]
 
 
+OpenAPIVersion: TypeAlias = Literal['auto', '2.0', '3.0', '3.1']
+
+
 PythonVersion: TypeAlias = Literal['3.10', '3.11', '3.12', '3.13', '3.14']
 
 
@@ -117,6 +125,8 @@ class DataclassArguments(TypedDict):
 class GenerateConfig(TypedDict):
     input_filename: NotRequired[str | None]
     input_file_type: NotRequired[InputFileType]
+    jsonschema_version: NotRequired[JsonSchemaVersion]
+    openapi_version: NotRequired[OpenAPIVersion]
     output: NotRequired[str | None]
     output_model_type: NotRequired[DataModelType]
     target_python_version: NotRequired[PythonVersion]