diff --git a/modules/openapi-generator/src/main/resources/python/model.mustache b/modules/openapi-generator/src/main/resources/python/model.mustache index 16c8dadbf620..b84e4efd42df 100644 --- a/modules/openapi-generator/src/main/resources/python/model.mustache +++ b/modules/openapi-generator/src/main/resources/python/model.mustache @@ -48,6 +48,18 @@ class {{classname}}(object): discriminator_value_class_map = { {{#children}}'{{^vendorExtensions.x-discriminator-value}}{{name}}{{/vendorExtensions.x-discriminator-value}}{{#vendorExtensions.x-discriminator-value}}{{{vendorExtensions.x-discriminator-value}}}{{/vendorExtensions.x-discriminator-value}}': '{{{classname}}}'{{^-last}}, {{/-last}}{{/children}} + {{^children}} + {{#mapping}} + {{#mappedModels}} + '{{mappingName}}': '{{modelName}}'{{^-last}},{{/-last}} + {{/mappedModels}} + {{/mapping}} + {{^mapping}} + {{#interfaces}} + '{{classname}}': '{{classname}}'{{^-last}},{{/-last}} + {{/interfaces}} + {{/mapping}} + {{/children}} } {{/discriminator}}