Skip to content

Commit 851910d

Browse files
author
Yousef Haggy
committed
fix model_generic python templates to have valid imports for polymorphism
1 parent dc175c5 commit 851910d

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

modules/openapi-generator/src/main/resources/python-pydantic-v1/model_generic.mustache

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ from typing import TYPE_CHECKING
1717
from importlib import import_module
1818
if TYPE_CHECKING:
1919
{{#mappedModels}}
20-
from {{packageName}}.models.{{model.classVarName}} import {{modelName}}
20+
from {{packageName}}.models.{{model.classFilename}} import {{modelName}}
2121
{{/mappedModels}}
2222

2323
{{/discriminator}}
@@ -238,7 +238,7 @@ class {{classname}}({{#parent}}{{{.}}}{{/parent}}{{^parent}}BaseModel{{/parent}}
238238
object_type = cls.get_discriminator_value(obj)
239239
{{#mappedModels}}
240240
if object_type == '{{{modelName}}}':
241-
return import_module("{{packageName}}.models.{{model.classVarName}}").{{modelName}}.from_dict(obj)
241+
return import_module("{{packageName}}.models.{{model.classFilename}}").{{modelName}}.from_dict(obj)
242242
{{/mappedModels}}
243243
raise ValueError("{{{classname}}} failed to lookup discriminator value from " +
244244
json.dumps(obj) + ". Discriminator property name: " + cls.__discriminator_property_name +

modules/openapi-generator/src/main/resources/python/model_generic.mustache

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ from typing_extensions import Self
1818
from typing import TYPE_CHECKING
1919
if TYPE_CHECKING:
2020
{{#mappedModels}}
21-
from {{packageName}}.models.{{model.classVarName}} import {{modelName}}
21+
from {{packageName}}.models.{{model.classFilename}} import {{modelName}}
2222
{{/mappedModels}}
2323

2424
{{/discriminator}}
@@ -261,7 +261,7 @@ class {{classname}}({{#parent}}{{{.}}}{{/parent}}{{^parent}}BaseModel{{/parent}}
261261
object_type = cls.get_discriminator_value(obj)
262262
{{#mappedModels}}
263263
if object_type == '{{{modelName}}}':
264-
return import_module("{{packageName}}.models.{{model.classVarName}}").{{modelName}}.from_dict(obj)
264+
return import_module("{{packageName}}.models.{{model.classFilename}}").{{modelName}}.from_dict(obj)
265265
{{/mappedModels}}
266266

267267
raise ValueError("{{{classname}}} failed to lookup discriminator value from " +

0 commit comments

Comments
 (0)