[PYTHON] Fix for failing to lookup discriminator value using AllOf and discriminator (#18498)

* fix issue 18495

* add tests and update samples

* update samples

Author: zishunwei

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

@@ -237,7 +237,7 @@ class {{classname}}({{#parent}}{{{.}}}{{/parent}}{{^parent}}BaseModel{{/parent}}
         # look up the object type based on discriminator mapping
         object_type = cls.get_discriminator_value(obj)
         {{#mappedModels}}
-        if object_type ==  '{{{mappingName}}}':
+        if object_type ==  '{{{modelName}}}':
             return import_module("{{packageName}}.models.{{model.classVarName}}").{{modelName}}.from_dict(obj)
         {{/mappedModels}}
         raise ValueError("{{{classname}}} failed to lookup discriminator value from " +

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

@@ -253,7 +253,7 @@ class {{classname}}({{#parent}}{{{.}}}{{/parent}}{{^parent}}BaseModel{{/parent}}
         # look up the object type based on discriminator mapping
         object_type = cls.get_discriminator_value(obj)
         {{#mappedModels}}
-        if object_type ==  '{{{mappingName}}}':
+        if object_type ==  '{{{modelName}}}':
             return import_module("{{packageName}}.models.{{model.classVarName}}").{{modelName}}.from_dict(obj)
         {{/mappedModels}}
 

modules/openapi-generator/src/test/resources/3_0/python/petstore-with-fake-endpoints-models-for-testing.yaml

@@ -2741,3 +2741,26 @@ components:
         _name:
           type: string
       type: object
+    Info:
+      allOf:
+        - $ref: '#/components/schemas/BaseDiscriminator'
+        - properties:
+            val:
+              $ref: '#/components/schemas/BaseDiscriminator'
+      type: object
+    BaseDiscriminator:
+      discriminator:
+        mapping:
+          string: '#/components/schemas/PrimitiveString'
+        propertyName: _typeName
+      properties:
+        _typeName:
+          type: string
+      type: object
+    PrimitiveString:
+      allOf:
+        - $ref: '#/components/schemas/BaseDiscriminator'
+        - properties:
+            _value:
+              type: string
+      type: object

samples/openapi3/client/petstore/python-aiohttp/.openapi-generator/FILES

@@ -17,6 +17,7 @@ docs/ArrayOfArrayOfModel.md
 docs/ArrayOfArrayOfNumberOnly.md
 docs/ArrayOfNumberOnly.md
 docs/ArrayTest.md
+docs/BaseDiscriminator.md
 docs/BasquePig.md
 docs/Bathing.md
 docs/Capitalization.md
@@ -53,6 +54,7 @@ docs/FormatTest.md
 docs/HasOnlyReadOnly.md
 docs/HealthCheckResult.md
 docs/ImportTestDatetimeApi.md
+docs/Info.md
 docs/InnerDictWithProperty.md
 docs/InputAllOf.md
 docs/IntOrString.md
@@ -84,6 +86,7 @@ docs/Pet.md
 docs/PetApi.md
 docs/Pig.md
 docs/PoopCleaning.md
+docs/PrimitiveString.md
 docs/PropertyMap.md
 docs/PropertyNameCollision.md
 docs/ReadOnlyFirst.md
@@ -137,6 +140,7 @@ petstore_api/models/array_of_array_of_model.py
 petstore_api/models/array_of_array_of_number_only.py
 petstore_api/models/array_of_number_only.py
 petstore_api/models/array_test.py
+petstore_api/models/base_discriminator.py
 petstore_api/models/basque_pig.py
 petstore_api/models/bathing.py
 petstore_api/models/capitalization.py
@@ -169,6 +173,7 @@ petstore_api/models/foo_get_default_response.py
 petstore_api/models/format_test.py
 petstore_api/models/has_only_read_only.py
 petstore_api/models/health_check_result.py
+petstore_api/models/info.py
 petstore_api/models/inner_dict_with_property.py
 petstore_api/models/input_all_of.py
 petstore_api/models/int_or_string.py
@@ -199,6 +204,7 @@ petstore_api/models/parent_with_optional_dict.py
 petstore_api/models/pet.py
 petstore_api/models/pig.py
 petstore_api/models/poop_cleaning.py
+petstore_api/models/primitive_string.py
 petstore_api/models/property_map.py
 petstore_api/models/property_name_collision.py
 petstore_api/models/read_only_first.py

samples/openapi3/client/petstore/python-aiohttp/README.md

@@ -163,6 +163,7 @@ Class | Method | HTTP request | Description
  - [ArrayOfArrayOfNumberOnly](docs/ArrayOfArrayOfNumberOnly.md)
  - [ArrayOfNumberOnly](docs/ArrayOfNumberOnly.md)
  - [ArrayTest](docs/ArrayTest.md)
+ - [BaseDiscriminator](docs/BaseDiscriminator.md)
  - [BasquePig](docs/BasquePig.md)
  - [Bathing](docs/Bathing.md)
  - [Capitalization](docs/Capitalization.md)
@@ -195,6 +196,7 @@ Class | Method | HTTP request | Description
  - [FormatTest](docs/FormatTest.md)
  - [HasOnlyReadOnly](docs/HasOnlyReadOnly.md)
  - [HealthCheckResult](docs/HealthCheckResult.md)
+ - [Info](docs/Info.md)
  - [InnerDictWithProperty](docs/InnerDictWithProperty.md)
  - [InputAllOf](docs/InputAllOf.md)
  - [IntOrString](docs/IntOrString.md)
@@ -225,6 +227,7 @@ Class | Method | HTTP request | Description
  - [Pet](docs/Pet.md)
  - [Pig](docs/Pig.md)
  - [PoopCleaning](docs/PoopCleaning.md)
+ - [PrimitiveString](docs/PrimitiveString.md)
  - [PropertyMap](docs/PropertyMap.md)
  - [PropertyNameCollision](docs/PropertyNameCollision.md)
  - [ReadOnlyFirst](docs/ReadOnlyFirst.md)

samples/openapi3/client/petstore/python-aiohttp/docs/BaseDiscriminator.md

@@ -0,0 +1,29 @@
+# BaseDiscriminator
+
+
+## Properties
+
+Name | Type | Description | Notes
+------------ | ------------- | ------------- | -------------
+**type_name** | **str** |  | [optional] 
+
+## Example
+
+```python
+from petstore_api.models.base_discriminator import BaseDiscriminator
+
+# TODO update the JSON string below
+json = "{}"
+# create an instance of BaseDiscriminator from a JSON string
+base_discriminator_instance = BaseDiscriminator.from_json(json)
+# print the JSON string representation of the object
+print(BaseDiscriminator.to_json())
+
+# convert the object into a dict
+base_discriminator_dict = base_discriminator_instance.to_dict()
+# create an instance of BaseDiscriminator from a dict
+base_discriminator_from_dict = BaseDiscriminator.from_dict(base_discriminator_dict)
+```
+[[Back to Model list]](../README.md#documentation-for-models) [[Back to API list]](../README.md#documentation-for-api-endpoints) [[Back to README]](../README.md)
+
+

samples/openapi3/client/petstore/python-aiohttp/docs/Info.md

@@ -0,0 +1,29 @@
+# Info
+
+
+## Properties
+
+Name | Type | Description | Notes
+------------ | ------------- | ------------- | -------------
+**val** | [**BaseDiscriminator**](BaseDiscriminator.md) |  | [optional] 
+
+## Example
+
+```python
+from petstore_api.models.info import Info
+
+# TODO update the JSON string below
+json = "{}"
+# create an instance of Info from a JSON string
+info_instance = Info.from_json(json)
+# print the JSON string representation of the object
+print(Info.to_json())
+
+# convert the object into a dict
+info_dict = info_instance.to_dict()
+# create an instance of Info from a dict
+info_from_dict = Info.from_dict(info_dict)
+```
+[[Back to Model list]](../README.md#documentation-for-models) [[Back to API list]](../README.md#documentation-for-api-endpoints) [[Back to README]](../README.md)
+
+

samples/openapi3/client/petstore/python-aiohttp/docs/PrimitiveString.md

@@ -0,0 +1,29 @@
+# PrimitiveString
+
+
+## Properties
+
+Name | Type | Description | Notes
+------------ | ------------- | ------------- | -------------
+**value** | **str** |  | [optional] 
+
+## Example
+
+```python
+from petstore_api.models.primitive_string import PrimitiveString
+
+# TODO update the JSON string below
+json = "{}"
+# create an instance of PrimitiveString from a JSON string
+primitive_string_instance = PrimitiveString.from_json(json)
+# print the JSON string representation of the object
+print(PrimitiveString.to_json())
+
+# convert the object into a dict
+primitive_string_dict = primitive_string_instance.to_dict()
+# create an instance of PrimitiveString from a dict
+primitive_string_from_dict = PrimitiveString.from_dict(primitive_string_dict)
+```
+[[Back to Model list]](../README.md#documentation-for-models) [[Back to API list]](../README.md#documentation-for-api-endpoints) [[Back to README]](../README.md)
+
+

samples/openapi3/client/petstore/python-aiohttp/petstore_api/__init__.py

@@ -52,6 +52,7 @@
 from petstore_api.models.array_of_array_of_number_only import ArrayOfArrayOfNumberOnly
 from petstore_api.models.array_of_number_only import ArrayOfNumberOnly
 from petstore_api.models.array_test import ArrayTest
+from petstore_api.models.base_discriminator import BaseDiscriminator
 from petstore_api.models.basque_pig import BasquePig
 from petstore_api.models.bathing import Bathing
 from petstore_api.models.capitalization import Capitalization
@@ -84,6 +85,7 @@
 from petstore_api.models.format_test import FormatTest
 from petstore_api.models.has_only_read_only import HasOnlyReadOnly
 from petstore_api.models.health_check_result import HealthCheckResult
+from petstore_api.models.info import Info
 from petstore_api.models.inner_dict_with_property import InnerDictWithProperty
 from petstore_api.models.input_all_of import InputAllOf
 from petstore_api.models.int_or_string import IntOrString
@@ -114,6 +116,7 @@
 from petstore_api.models.pet import Pet
 from petstore_api.models.pig import Pig
 from petstore_api.models.poop_cleaning import PoopCleaning
+from petstore_api.models.primitive_string import PrimitiveString
 from petstore_api.models.property_map import PropertyMap
 from petstore_api.models.property_name_collision import PropertyNameCollision
 from petstore_api.models.read_only_first import ReadOnlyFirst

samples/openapi3/client/petstore/python-aiohttp/petstore_api/models/__init__.py

@@ -27,6 +27,7 @@
 from petstore_api.models.array_of_array_of_number_only import ArrayOfArrayOfNumberOnly
 from petstore_api.models.array_of_number_only import ArrayOfNumberOnly
 from petstore_api.models.array_test import ArrayTest
+from petstore_api.models.base_discriminator import BaseDiscriminator
 from petstore_api.models.basque_pig import BasquePig
 from petstore_api.models.bathing import Bathing
 from petstore_api.models.capitalization import Capitalization
@@ -59,6 +60,7 @@
 from petstore_api.models.format_test import FormatTest
 from petstore_api.models.has_only_read_only import HasOnlyReadOnly
 from petstore_api.models.health_check_result import HealthCheckResult
+from petstore_api.models.info import Info
 from petstore_api.models.inner_dict_with_property import InnerDictWithProperty
 from petstore_api.models.input_all_of import InputAllOf
 from petstore_api.models.int_or_string import IntOrString
@@ -89,6 +91,7 @@
 from petstore_api.models.pet import Pet
 from petstore_api.models.pig import Pig
 from petstore_api.models.poop_cleaning import PoopCleaning
+from petstore_api.models.primitive_string import PrimitiveString
 from petstore_api.models.property_map import PropertyMap
 from petstore_api.models.property_name_collision import PropertyNameCollision
 from petstore_api.models.read_only_first import ReadOnlyFirst