Skip to content

Commit aee89fc

Browse files
committed
Tuples and unions are only used if there are truly more then one item there
1 parent dc5b91c commit aee89fc

File tree

1 file changed

+13
-14
lines changed

1 file changed

+13
-14
lines changed

src/openapi_python_generator/language_converters/python/model_generator.py

Lines changed: 13 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -56,13 +56,12 @@ def type_converter(schema: Schema, required: bool = False) -> TypeConversion:
5656
original_type = (
5757
"tuple<" + ",".join([i.original_type for i in conversions]) + ">"
5858
)
59-
converted_type = (
60-
pre_type
61-
+ "Tuple["
62-
+ ",".join([i.converted_type for i in conversions])
63-
+ "]"
64-
+ post_type
65-
)
59+
if len(conversions) == 1:
60+
converted_type = conversions[0].converted_type
61+
else:
62+
converted_type = ",".join([i.converted_type for i in conversions])
63+
64+
converted_type = pre_type + "Tuple[" + converted_type + "]" + post_type
6665
import_types = [
6766
i.import_types[0] for i in conversions if i.import_types is not None
6867
]
@@ -87,13 +86,13 @@ def type_converter(schema: Schema, required: bool = False) -> TypeConversion:
8786
original_type = (
8887
"union<" + ",".join([i.original_type for i in conversions]) + ">"
8988
)
90-
converted_type = (
91-
pre_type
92-
+ "Union["
93-
+ ",".join([i.converted_type for i in conversions])
94-
+ "]"
95-
+ post_type
96-
)
89+
90+
if len(conversions) == 1:
91+
converted_type = conversions[0].converted_type
92+
else:
93+
converted_type = ",".join([i.converted_type for i in conversions])
94+
95+
converted_type = pre_type + "Union[" + converted_type + "]" + post_type
9796
import_types = list(
9897
itertools.chain(
9998
*[i.import_types for i in conversions if i.import_types is not None]

0 commit comments

Comments
 (0)