|
5 | 5 | from typing import Any, Dict, List, Optional |
6 | 6 |
|
7 | 7 | import typer |
8 | | -from datamodel_code_generator import LiteralType, PythonVersion, chdir, DataModelType |
9 | | -from datamodel_code_generator.model import get_data_model_types |
| 8 | +from datamodel_code_generator import DataModelType, LiteralType, PythonVersion, chdir |
10 | 9 | from datamodel_code_generator.format import CodeFormatter |
11 | 10 | from datamodel_code_generator.imports import Import, Imports |
| 11 | +from datamodel_code_generator.model import get_data_model_types |
12 | 12 | from datamodel_code_generator.reference import Reference |
13 | 13 | from datamodel_code_generator.types import DataType |
14 | 14 | from jinja2 import Environment, FileSystemLoader |
@@ -59,9 +59,11 @@ def main( |
59 | 59 | ), |
60 | 60 | disable_timestamp: bool = typer.Option(False, "--disable-timestamp"), |
61 | 61 | output_model_type: DataModelType = typer.Option( |
62 | | - DataModelType.PydanticBaseModel, "--data-model-type", "-d"), |
| 62 | + DataModelType.PydanticBaseModel, "--data-model-type", "-d" |
| 63 | + ), |
63 | 64 | python_version: PythonVersion = typer.Option( |
64 | | - PythonVersion.PY_38, "--python-version", "-p"), |
| 65 | + PythonVersion.PY_38, "--python-version", "-p" |
| 66 | + ), |
65 | 67 | ) -> None: |
66 | 68 | input_name: str = input_file |
67 | 69 | input_text: str |
@@ -129,14 +131,15 @@ def generate_code( |
129 | 131 |
|
130 | 132 | data_model_types = get_data_model_types(output_model_type, python_version) |
131 | 133 |
|
132 | | - parser = OpenAPIParser(input_text, |
| 134 | + parser = OpenAPIParser( |
| 135 | + input_text, |
133 | 136 | enum_field_as_literal=enum_field_as_literal, |
134 | 137 | data_model_type=data_model_types.data_model, |
135 | 138 | data_model_root_type=data_model_types.root_model, |
136 | 139 | data_model_field_type=data_model_types.field_model, |
137 | 140 | data_type_manager_type=data_model_types.data_type_manager, |
138 | 141 | dump_resolve_reference_action=data_model_types.dump_resolve_reference_action, |
139 | | - ) |
| 142 | + ) |
140 | 143 |
|
141 | 144 | with chdir(output_dir): |
142 | 145 | models = parser.parse() |
|
0 commit comments