|
| 1 | +# todo: remove this file when the following PR is merged into datamodels-code-generator: |
| 2 | +# |
| 3 | +# https://github.com/koxudaxi/datamodel-code-generator/pull/2379 |
| 4 | + |
1 | 5 | import logging |
2 | 6 | from itertools import groupby |
3 | 7 | from pathlib import Path |
4 | | -from typing import NamedTuple |
| 8 | +from typing import NamedTuple, Optional, Union |
5 | 9 |
|
6 | 10 | from datamodel_code_generator.format import CodeFormatter |
7 | 11 | from datamodel_code_generator.imports import IMPORT_ANNOTATIONS, Import, Imports |
@@ -31,17 +35,17 @@ def __alias_shadowed_imports( |
31 | 35 |
|
32 | 36 | def _parse( # noqa: PLR0912, PLR0914, PLR0915 |
33 | 37 | self: base.Parser, |
34 | | - with_import: bool | None = True, # noqa: FBT001, FBT002 |
35 | | - format_: bool | None = True, # noqa: FBT001, FBT002 |
36 | | - settings_path: Path | None = None, |
37 | | - ) -> str | dict[tuple[str, ...], base.Result]: |
| 38 | + with_import: Optional[bool] = True, # noqa: FBT001, FBT002 |
| 39 | + format_: Optional[bool] = True, # noqa: FBT001, FBT002 |
| 40 | + settings_path: Optional[Path] = None, |
| 41 | + ) -> Union[str, dict[tuple[str, ...], base.Result]]: |
38 | 42 | self.parse_raw() |
39 | 43 |
|
40 | 44 | if with_import: |
41 | 45 | self.imports.append(IMPORT_ANNOTATIONS) |
42 | 46 |
|
43 | 47 | if format_: |
44 | | - code_formatter: CodeFormatter | None = CodeFormatter( |
| 48 | + code_formatter: Optional[CodeFormatter] = CodeFormatter( |
45 | 49 | self.target_python_version, |
46 | 50 | settings_path, |
47 | 51 | self.wrap_string_literal, |
|
0 commit comments