datamodel-code-generator 0.30.2__tar.gz → 0.31.0__tar.gz

{datamodel_code_generator-0.30.2 → datamodel_code_generator-0.31.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: datamodel-code-generator
-Version: 0.30.2
+Version: 0.31.0
 Summary: Datamodel Code Generator
 Project-URL: Homepage, https://github.com/koxudaxi/datamodel-code-generator
 Project-URL: Source, https://github.com/koxudaxi/datamodel-code-generator
@@ -489,8 +489,9 @@ Field customization:
                         Use schema description to populate field docstring
 
 Model customization:
-  --allow-extra-fields  Allow passing extra fields, if this flag is not passed, extra fields
-                        are forbidden.
+  --allow-extra-fields  Deprecated: Allow passing extra fields, if this flag is not passed,
+                        extra fields are forbidden. This flag is deprecated. Use `--extra-
+                        fields=allow` instead.
   --allow-population-by-field-name
                         Allow population by field name
   --class-name CLASS_NAME
@@ -505,6 +506,8 @@ Model customization:
                         Enable faux immutability
   --enable-version-header
                         Enable package version on file headers
+  --extra-fields {allow,ignore,forbid}
+                        Set the generated models to allow, forbid, or ignore extra fields.
   --frozen-dataclasses  Generate frozen dataclasses (dataclass(frozen=True)). Only applies
                         to dataclass output.
   --keep-model-order    Keep generated models'' order

{datamodel_code_generator-0.30.2 → datamodel_code_generator-0.31.0}/README.md

@@ -439,8 +439,9 @@ Field customization:
                         Use schema description to populate field docstring
 
 Model customization:
-  --allow-extra-fields  Allow passing extra fields, if this flag is not passed, extra fields
-                        are forbidden.
+  --allow-extra-fields  Deprecated: Allow passing extra fields, if this flag is not passed,
+                        extra fields are forbidden. This flag is deprecated. Use `--extra-
+                        fields=allow` instead.
   --allow-population-by-field-name
                         Allow population by field name
   --class-name CLASS_NAME
@@ -455,6 +456,8 @@ Model customization:
                         Enable faux immutability
   --enable-version-header
                         Enable package version on file headers
+  --extra-fields {allow,ignore,forbid}
+                        Set the generated models to allow, forbid, or ignore extra fields.
   --frozen-dataclasses  Generate frozen dataclasses (dataclass(frozen=True)). Only applies
                         to dataclass output.
   --keep-model-order    Keep generated models'' order

{datamodel_code_generator-0.30.2 → datamodel_code_generator-0.31.0}/src/datamodel_code_generator/__init__.py

@@ -233,6 +233,7 @@ def generate(  # noqa: PLR0912, PLR0913, PLR0914, PLR0915
     enable_version_header: bool = False,
     allow_population_by_field_name: bool = False,
     allow_extra_fields: bool = False,
+    extra_fields: str | None = None,
     apply_default_values_for_required_fields: bool = False,
     force_optional_for_required_fields: bool = False,
     class_name: str | None = None,
@@ -428,6 +429,7 @@ def generate(  # noqa: PLR0912, PLR0913, PLR0914, PLR0915
         aliases=aliases,
         allow_population_by_field_name=allow_population_by_field_name,
         allow_extra_fields=allow_extra_fields,
+        extra_fields=extra_fields,
         apply_default_values_for_required_fields=apply_default_values_for_required_fields,
         force_optional_for_required_fields=force_optional_for_required_fields,
         class_name=class_name,

{datamodel_code_generator-0.30.2 → datamodel_code_generator-0.31.0}/src/datamodel_code_generator/__main__.py

@@ -261,6 +261,7 @@ class Config(BaseModel):
     enable_version_header: bool = False
     allow_population_by_field_name: bool = False
     allow_extra_fields: bool = False
+    extra_fields: Optional[str] = None  # noqa: UP045
     use_default: bool = False
     force_optional: bool = False
     class_name: Optional[str] = None  # noqa: UP045
@@ -475,6 +476,7 @@ def main(args: Sequence[str] | None = None) -> Exit:  # noqa: PLR0911, PLR0912,
             enable_version_header=config.enable_version_header,
             allow_population_by_field_name=config.allow_population_by_field_name,
             allow_extra_fields=config.allow_extra_fields,
+            extra_fields=config.extra_fields,
             apply_default_values_for_required_fields=config.use_default,
             force_optional_for_required_fields=config.force_optional,
             class_name=config.class_name,

{datamodel_code_generator-0.30.2 → datamodel_code_generator-0.31.0}/src/datamodel_code_generator/arguments.py

@@ -43,6 +43,7 @@ base_options = arg_parser.add_argument_group("Options")
 typing_options = arg_parser.add_argument_group("Typing customization")
 field_options = arg_parser.add_argument_group("Field customization")
 model_options = arg_parser.add_argument_group("Model customization")
+extra_fields_model_options = model_options.add_mutually_exclusive_group()
 template_options = arg_parser.add_argument_group("Template customization")
 openapi_options = arg_parser.add_argument_group("OpenAPI-only options")
 general_options = arg_parser.add_argument_group("General options")
@@ -94,9 +95,10 @@ base_options.add_argument(
 # ======================================================================================
 # Customization options for generated models
 # ======================================================================================
-model_options.add_argument(
+extra_fields_model_options.add_argument(
     "--allow-extra-fields",
-    help="Allow passing extra fields, if this flag is not passed, extra fields are forbidden.",
+    help="Deprecated: Allow passing extra fields, if this flag is not passed, extra fields are forbidden. "
+    "This flag is deprecated. Use `--extra-fields=allow` instead.",
     action="store_true",
     default=None,
 )
@@ -141,6 +143,12 @@ model_options.add_argument(
     action="store_true",
     default=None,
 )
+extra_fields_model_options.add_argument(
+    "--extra-fields",
+    help="Set the generated models to allow, forbid, or ignore extra fields.",
+    choices=["allow", "ignore", "forbid"],
+    default=None,
+)
 model_options.add_argument(
     "--keep-model-order",
     help="Keep generated models' order",

{datamodel_code_generator-0.30.2 → datamodel_code_generator-0.31.0}/src/datamodel_code_generator/model/pydantic/__init__.py

@@ -22,6 +22,7 @@ class Config(_BaseModel):
     title: Optional[str] = None  # noqa: UP045
     allow_population_by_field_name: Optional[bool] = None  # noqa: UP045
     allow_extra_fields: Optional[bool] = None  # noqa: UP045
+    extra_fields: Optional[str] = None  # noqa: UP045
     allow_mutation: Optional[bool] = None  # noqa: UP045
     arbitrary_types_allowed: Optional[bool] = None  # noqa: UP045
     orm_mode: Optional[bool] = None  # noqa: UP045

{datamodel_code_generator-0.30.2 → datamodel_code_generator-0.31.0}/src/datamodel_code_generator/model/pydantic/base_model.py

@@ -286,12 +286,20 @@ class BaseModel(BaseModelBase):
 
         additional_properties = self.extra_template_data.get("additionalProperties")
         allow_extra_fields = self.extra_template_data.get("allow_extra_fields")
-        if additional_properties is not None or allow_extra_fields:
-            config_parameters["extra"] = (
-                "Extra.allow" if additional_properties or allow_extra_fields else "Extra.forbid"
-            )
+        extra_fields = self.extra_template_data.get("extra_fields")
+
+        if allow_extra_fields or extra_fields or additional_properties is not None:
             self._additional_imports.append(IMPORT_EXTRA)
 
+        if allow_extra_fields:
+            config_parameters["extra"] = "Extra.allow"
+        elif extra_fields:
+            config_parameters["extra"] = f"Extra.{extra_fields}"
+        elif additional_properties is True:
+            config_parameters["extra"] = "Extra.allow"
+        elif additional_properties is False:
+            config_parameters["extra"] = "Extra.forbid"
+
         for config_attribute in "allow_population_by_field_name", "allow_mutation":
             if config_attribute in self.extra_template_data:
                 config_parameters[config_attribute] = self.extra_template_data[config_attribute]

{datamodel_code_generator-0.30.2 → datamodel_code_generator-0.31.0}/src/datamodel_code_generator/model/pydantic_v2/__init__.py

@@ -21,6 +21,7 @@ class ConfigDict(_BaseModel):
     title: Optional[str] = None  # noqa: UP045
     populate_by_name: Optional[bool] = None  # noqa: UP045
     allow_extra_fields: Optional[bool] = None  # noqa: UP045
+    extra_fields: Optional[str] = None  # noqa: UP045
     from_attributes: Optional[bool] = None  # noqa: UP045
     frozen: Optional[bool] = None  # noqa: UP045
     arbitrary_types_allowed: Optional[bool] = None  # noqa: UP045

{datamodel_code_generator-0.30.2 → datamodel_code_generator-0.31.0}/src/datamodel_code_generator/parser/base.py

@@ -326,6 +326,7 @@ class Parser(ABC):
         allow_population_by_field_name: bool = False,
         apply_default_values_for_required_fields: bool = False,
         allow_extra_fields: bool = False,
+        extra_fields: str | None = None,
         force_optional_for_required_fields: bool = False,
         class_name: str | None = None,
         use_standard_collections: bool = False,
@@ -452,6 +453,9 @@ class Parser(ABC):
         if allow_extra_fields:
             self.extra_template_data[ALL_MODEL]["allow_extra_fields"] = True
 
+        if extra_fields:
+            self.extra_template_data[ALL_MODEL]["extra_fields"] = extra_fields
+
         if enable_faux_immutability:
             self.extra_template_data[ALL_MODEL]["allow_mutation"] = False
 
@@ -1135,7 +1139,7 @@ class Parser(ABC):
         if self.data_model_type != pydantic_model_v2.BaseModel:
             return
         for model in models:
-            if model.base_class == "Enum":
+            if "Enum" in model.base_class:
                 continue
 
             for field in model.fields:

{datamodel_code_generator-0.30.2 → datamodel_code_generator-0.31.0}/src/datamodel_code_generator/parser/graphql.py

@@ -102,6 +102,7 @@ class GraphQLParser(Parser):
         allow_population_by_field_name: bool = False,
         apply_default_values_for_required_fields: bool = False,
         allow_extra_fields: bool = False,
+        extra_fields: str | None = None,
         force_optional_for_required_fields: bool = False,
         class_name: str | None = None,
         use_standard_collections: bool = False,
@@ -177,6 +178,7 @@ class GraphQLParser(Parser):
             aliases=aliases,
             allow_population_by_field_name=allow_population_by_field_name,
             allow_extra_fields=allow_extra_fields,
+            extra_fields=extra_fields,
             apply_default_values_for_required_fields=apply_default_values_for_required_fields,
             force_optional_for_required_fields=force_optional_for_required_fields,
             class_name=class_name,

{datamodel_code_generator-0.30.2 → datamodel_code_generator-0.31.0}/src/datamodel_code_generator/parser/jsonschema.py

@@ -381,6 +381,7 @@ class JsonSchemaParser(Parser):
         allow_population_by_field_name: bool = False,
         apply_default_values_for_required_fields: bool = False,
         allow_extra_fields: bool = False,
+        extra_fields: str | None = None,
         force_optional_for_required_fields: bool = False,
         class_name: str | None = None,
         use_standard_collections: bool = False,
@@ -456,6 +457,7 @@ class JsonSchemaParser(Parser):
             aliases=aliases,
             allow_population_by_field_name=allow_population_by_field_name,
             allow_extra_fields=allow_extra_fields,
+            extra_fields=extra_fields,
             apply_default_values_for_required_fields=apply_default_values_for_required_fields,
             force_optional_for_required_fields=force_optional_for_required_fields,
             class_name=class_name,

{datamodel_code_generator-0.30.2 → datamodel_code_generator-0.31.0}/src/datamodel_code_generator/parser/openapi.py

@@ -162,6 +162,7 @@ class OpenAPIParser(JsonSchemaParser):
         aliases: Mapping[str, str] | None = None,
         allow_population_by_field_name: bool = False,
         allow_extra_fields: bool = False,
+        extra_fields: str | None = None,
         apply_default_values_for_required_fields: bool = False,
         force_optional_for_required_fields: bool = False,
         class_name: str | None = None,
@@ -239,6 +240,7 @@ class OpenAPIParser(JsonSchemaParser):
             aliases=aliases,
             allow_population_by_field_name=allow_population_by_field_name,
             allow_extra_fields=allow_extra_fields,
+            extra_fields=extra_fields,
             apply_default_values_for_required_fields=apply_default_values_for_required_fields,
             force_optional_for_required_fields=force_optional_for_required_fields,
             class_name=class_name,

datamodel_code_generator-0.31.0/tests/data/expected/main/graphql/simple_star_wars_extra_fields_allow.py

@@ -0,0 +1,172 @@
+# generated by datamodel-codegen:
+#   filename:  simple-star-wars.graphql
+#   timestamp: 2019-07-26T00:00:00+00:00
+
+from __future__ import annotations
+
+from typing import List, Literal, Optional, TypeAlias
+
+from pydantic import BaseModel, Extra, Field
+
+Boolean: TypeAlias = bool
+"""
+The `Boolean` scalar type represents `true` or `false`.
+"""
+
+
+ID: TypeAlias = str
+"""
+The `ID` scalar type represents a unique identifier, often used to refetch an object or as key for a cache. The ID type appears in a JSON response as a String; however, it is not intended to be human-readable. When expected as an input type, any string (such as `"4"`) or integer (such as `4`) input value will be accepted as an ID.
+"""
+
+
+Int: TypeAlias = int
+"""
+The `Int` scalar type represents non-fractional signed whole numeric values. Int can represent values between -(2^31) and 2^31 - 1.
+"""
+
+
+String: TypeAlias = str
+"""
+The `String` scalar type represents textual data, represented as UTF-8 character sequences. The String type is most often used by GraphQL to represent free-form human-readable text.
+"""
+
+
+class Film(BaseModel):
+    class Config:
+        extra = Extra.allow
+
+    characters: List[Person]
+    characters_ids: List[ID]
+    director: String
+    episode_id: Int
+    id: ID
+    opening_crawl: String
+    planets: List[Planet]
+    planets_ids: List[ID]
+    producer: Optional[String] = None
+    release_date: String
+    species: List[Species]
+    species_ids: List[ID]
+    starships: List[Starship]
+    starships_ids: List[ID]
+    title: String
+    vehicles: List[Vehicle]
+    vehicles_ids: List[ID]
+    typename__: Optional[Literal['Film']] = Field('Film', alias='__typename')
+
+
+class Person(BaseModel):
+    class Config:
+        extra = Extra.allow
+
+    birth_year: Optional[String] = None
+    eye_color: Optional[String] = None
+    films: List[Film]
+    films_ids: List[ID]
+    gender: Optional[String] = None
+    hair_color: Optional[String] = None
+    height: Optional[Int] = None
+    homeworld: Optional[Planet] = None
+    homeworld_id: Optional[ID] = None
+    id: ID
+    mass: Optional[Int] = None
+    name: String
+    skin_color: Optional[String] = None
+    species: List[Species]
+    species_ids: List[ID]
+    starships: List[Starship]
+    starships_ids: List[ID]
+    vehicles: List[Vehicle]
+    vehicles_ids: List[ID]
+    typename__: Optional[Literal['Person']] = Field('Person', alias='__typename')
+
+
+class Planet(BaseModel):
+    class Config:
+        extra = Extra.allow
+
+    climate: Optional[String] = None
+    diameter: Optional[String] = None
+    films: List[Film]
+    films_ids: List[ID]
+    gravity: Optional[String] = None
+    id: ID
+    name: String
+    orbital_period: Optional[String] = None
+    population: Optional[String] = None
+    residents: List[Person]
+    residents_ids: List[ID]
+    rotation_period: Optional[String] = None
+    surface_water: Optional[String] = None
+    terrain: Optional[String] = None
+    typename__: Optional[Literal['Planet']] = Field('Planet', alias='__typename')
+
+
+class Species(BaseModel):
+    class Config:
+        extra = Extra.allow
+
+    average_height: Optional[String] = None
+    average_lifespan: Optional[String] = None
+    classification: Optional[String] = None
+    designation: Optional[String] = None
+    eye_colors: Optional[String] = None
+    films: List[Film]
+    films_ids: List[ID]
+    hair_colors: Optional[String] = None
+    id: ID
+    language: Optional[String] = None
+    name: String
+    people: List[Person]
+    people_ids: List[ID]
+    skin_colors: Optional[String] = None
+    typename__: Optional[Literal['Species']] = Field('Species', alias='__typename')
+
+
+class Starship(BaseModel):
+    class Config:
+        extra = Extra.allow
+
+    MGLT: Optional[String] = None
+    cargo_capacity: Optional[String] = None
+    consumables: Optional[String] = None
+    cost_in_credits: Optional[String] = None
+    crew: Optional[String] = None
+    films: List[Film]
+    films_ids: List[ID]
+    hyperdrive_rating: Optional[String] = None
+    id: ID
+    length: Optional[String] = None
+    manufacturer: Optional[String] = None
+    max_atmosphering_speed: Optional[String] = None
+    model: Optional[String] = None
+    name: String
+    passengers: Optional[String] = None
+    pilots: List[Person]
+    pilots_ids: List[ID]
+    starship_class: Optional[String] = None
+    typename__: Optional[Literal['Starship']] = Field('Starship', alias='__typename')
+
+
+class Vehicle(BaseModel):
+    class Config:
+        extra = Extra.allow
+
+    cargo_capacity: Optional[String] = None
+    consumables: Optional[String] = None
+    cost_in_credits: Optional[String] = None
+    crew: Optional[String] = None
+    films: List[Film]
+    films_ids: List[ID]
+    id: ID
+    length: Optional[String] = None
+    manufacturer: Optional[String] = None
+    max_atmosphering_speed: Optional[String] = None
+    model: Optional[String] = None
+    name: String
+    passengers: Optional[String] = None
+    pilots: List[Person]
+    pilots_ids: List[ID]
+    vehicle_class: Optional[String] = None
+    typename__: Optional[Literal['Vehicle']] = Field('Vehicle', alias='__typename')

datamodel_code_generator-0.31.0/tests/data/expected/main/jsonschema/extra_fields_ignore.py

@@ -0,0 +1,39 @@
+# generated by datamodel-codegen:
+#   filename:  extra_fields_ignore.json
+#   timestamp: 2019-07-26T00:00:00+00:00
+
+from __future__ import annotations
+
+from typing import Optional
+
+from pydantic import BaseModel, Extra
+
+
+class Foo(BaseModel):
+    class Config:
+        extra = Extra.ignore
+
+    x: Optional[int] = None
+
+
+class Bar(BaseModel):
+    class Config:
+        extra = Extra.ignore
+
+    y: Optional[int] = None
+
+
+class Baz(BaseModel):
+    class Config:
+        extra = Extra.ignore
+
+    z: Optional[int] = None
+
+
+class Test(BaseModel):
+    class Config:
+        extra = Extra.ignore
+
+    foo: Foo
+    bar: Optional[Bar] = None
+    baz: Optional[Baz] = None

datamodel_code_generator-0.31.0/tests/data/expected/main/openapi/additional_properties.py

@@ -0,0 +1,110 @@
+# generated by datamodel-codegen:
+#   filename:  additional_properties.yaml
+#   timestamp: 2019-07-26T00:00:00+00:00
+
+from __future__ import annotations
+
+from typing import Dict, List, Optional
+
+from pydantic import BaseModel, Extra, Field
+
+
+class Pet(BaseModel):
+    class Config:
+        extra = Extra.forbid
+
+    id: int
+    name: str
+    tag: Optional[str] = None
+
+
+class Pets(BaseModel):
+    class Config:
+        extra = Extra.forbid
+
+    __root__: List[Pet]
+
+
+class User(BaseModel):
+    class Config:
+        extra = Extra.forbid
+
+    id: int
+    name: str
+    tag: Optional[str] = None
+
+
+class Users(BaseModel):
+    class Config:
+        extra = Extra.forbid
+
+    __root__: List[User]
+
+
+class Id(BaseModel):
+    class Config:
+        extra = Extra.forbid
+
+    __root__: str
+
+
+class Rules(BaseModel):
+    class Config:
+        extra = Extra.forbid
+
+    __root__: List[str]
+
+
+class Error(BaseModel):
+    class Config:
+        extra = Extra.forbid
+
+    code: int
+    message: str
+
+
+class Event(BaseModel):
+    class Config:
+        extra = Extra.forbid
+
+    name: Optional[str] = None
+
+
+class Result(BaseModel):
+    class Config:
+        extra = Extra.forbid
+
+    event: Optional[Event] = None
+
+
+class Broken(BaseModel):
+    class Config:
+        extra = Extra.forbid
+
+    foo: Optional[str] = None
+    bar: Optional[int] = None
+
+
+class BrokenArray(BaseModel):
+    class Config:
+        extra = Extra.forbid
+
+    broken: Optional[Dict[str, List[Broken]]] = None
+
+
+class FileSetUpload(BaseModel):
+    class Config:
+        extra = Extra.forbid
+
+    task_id: Optional[str] = Field(None, title='task id')
+    tags: Dict[str, List[str]] = Field(
+        ..., title='Dict of tags, each containing a list of file names'
+    )
+
+
+class Test(BaseModel):
+    class Config:
+        extra = Extra.forbid
+
+    broken: Optional[Dict[str, Broken]] = None
+    failing: Optional[Dict[str, str]] = {}

datamodel_code_generator-0.31.0/tests/data/jsonschema/extra_fields_ignore.json

@@ -0,0 +1,36 @@
+{
+  "title": "Test",
+  "type": "object",
+  "required": [
+    "foo"
+  ],
+  "properties": {
+    "foo": {
+      "type": "object",
+      "properties": {
+        "x": {
+          "type": "integer"
+        }
+      },
+      "additionalProperties": true
+    },
+    "bar": {
+      "type": "object",
+      "properties": {
+        "y": {
+          "type": "integer"
+        }
+      },
+      "additionalProperties": false
+    },
+    "baz": {
+      "type": "object",
+      "properties": {
+        "z": {
+          "type": "integer"
+        }
+      }
+    }
+  },
+  "additionalProperties": false
+}

{datamodel_code_generator-0.30.2 → datamodel_code_generator-0.31.0}/tests/main/graphql/test_main_graphql.py

@@ -276,3 +276,21 @@ def test_main_graphql_use_union_operator() -> None:
         ])
         assert return_code == Exit.OK
         assert output_file.read_text() == (EXPECTED_GRAPHQL_PATH / "use_union_operator.py").read_text()
+
+
+@freeze_time("2019-07-26")
+def test_main_graphql_extra_fields_allow() -> None:
+    with TemporaryDirectory() as output_dir:
+        output_file: Path = Path(output_dir) / "output.py"
+        return_code: Exit = main([
+            "--input",
+            str(GRAPHQL_DATA_PATH / "simple-star-wars.graphql"),
+            "--output",
+            str(output_file),
+            "--input-file-type",
+            "graphql",
+            "--extra-fields",
+            "allow",
+        ])
+        assert return_code == Exit.OK
+        assert output_file.read_text() == (EXPECTED_GRAPHQL_PATH / "simple_star_wars_extra_fields_allow.py").read_text()

{datamodel_code_generator-0.30.2 → datamodel_code_generator-0.31.0}/tests/main/jsonschema/test_main_jsonschema.py

@@ -3247,3 +3247,21 @@ def test_main_json_pointer_percent_encoded_segments() -> None:
         result = output_file.read_text()
         # Normalize whitespace for comparison
         assert "".join(result.split()) == "".join(expected.split())
+
+
+@freeze_time("2019-07-26")
+def test_main_extra_fields_ignore() -> None:
+    with TemporaryDirectory() as output_dir:
+        output_file: Path = Path(output_dir) / "output.py"
+        return_code: Exit = main([
+            "--input",
+            str(JSON_SCHEMA_DATA_PATH / "extra_fields_ignore.json"),
+            "--output",
+            str(output_file),
+            "--input-file-type",
+            "jsonschema",
+            "--extra-fields",
+            "ignore",
+        ])
+        assert return_code == Exit.OK
+        assert output_file.read_text() == (EXPECTED_JSON_SCHEMA_PATH / "extra_fields_ignore.py").read_text()

{datamodel_code_generator-0.30.2 → datamodel_code_generator-0.31.0}/tests/main/openapi/test_main_openapi.py

@@ -2608,3 +2608,21 @@ def test_main_openapi_shadowed_imports() -> None:
         ])
         assert return_code == Exit.OK
         assert output_file.read_text() == (EXPECTED_OPENAPI_PATH / "shadowed_imports.py").read_text()
+
+
+@freeze_time("2019-07-26")
+def test_main_openapi_extra_fields_forbid() -> None:
+    with TemporaryDirectory() as output_dir:
+        output_file: Path = Path(output_dir) / "output.py"
+        return_code: Exit = main([
+            "--input",
+            str(OPEN_API_DATA_PATH / "additional_properties.yaml"),
+            "--output",
+            str(output_file),
+            "--input-file-type",
+            "openapi",
+            "--extra-fields",
+            "forbid",
+        ])
+        assert return_code == Exit.OK
+        assert output_file.read_text() == (EXPECTED_OPENAPI_PATH / "additional_properties.py").read_text()