forked from loafle/openapi-generator-original
feat: remove unused imports from python clients; fix discriminator
This commit is contained in:
@@ -6,10 +6,6 @@ import pprint
|
||||
import re # noqa: F401
|
||||
|
||||
import six
|
||||
{{#imports}}{{#-first}}
|
||||
{{/-first}}
|
||||
{{import}} # noqa: F401,E501
|
||||
{{/imports}}
|
||||
|
||||
|
||||
{{#models}}
|
||||
@@ -170,7 +166,7 @@ class {{classname}}(object):
|
||||
{{#discriminator}}
|
||||
def get_real_child_model(self, data):
|
||||
"""Returns the real base class specified by the discriminator"""
|
||||
discriminator_value = data[self.discriminator].lower()
|
||||
discriminator_value = data[self.discriminator]
|
||||
return self.discriminator_value_class_map.get(discriminator_value)
|
||||
|
||||
{{/discriminator}}
|
||||
|
||||
@@ -102,7 +102,7 @@ class Animal(object):
|
||||
|
||||
def get_real_child_model(self, data):
|
||||
"""Returns the real base class specified by the discriminator"""
|
||||
discriminator_value = data[self.discriminator].lower()
|
||||
discriminator_value = data[self.discriminator]
|
||||
return self.discriminator_value_class_map.get(discriminator_value)
|
||||
|
||||
def to_dict(self):
|
||||
|
||||
@@ -15,8 +15,6 @@ import re # noqa: F401
|
||||
|
||||
import six
|
||||
|
||||
from petstore_api.models.animal import Animal # noqa: F401,E501
|
||||
|
||||
|
||||
class AnimalFarm(object):
|
||||
"""NOTE: This class is auto generated by OpenAPI Generator.
|
||||
|
||||
@@ -15,8 +15,6 @@ import re # noqa: F401
|
||||
|
||||
import six
|
||||
|
||||
from petstore_api.models.read_only_first import ReadOnlyFirst # noqa: F401,E501
|
||||
|
||||
|
||||
class ArrayTest(object):
|
||||
"""NOTE: This class is auto generated by OpenAPI Generator.
|
||||
|
||||
@@ -15,8 +15,6 @@ import re # noqa: F401
|
||||
|
||||
import six
|
||||
|
||||
from petstore_api.models.animal import Animal # noqa: F401,E501
|
||||
|
||||
|
||||
class Cat(object):
|
||||
"""NOTE: This class is auto generated by OpenAPI Generator.
|
||||
|
||||
@@ -15,8 +15,6 @@ import re # noqa: F401
|
||||
|
||||
import six
|
||||
|
||||
from petstore_api.models.animal import Animal # noqa: F401,E501
|
||||
|
||||
|
||||
class Dog(object):
|
||||
"""NOTE: This class is auto generated by OpenAPI Generator.
|
||||
|
||||
@@ -15,8 +15,6 @@ import re # noqa: F401
|
||||
|
||||
import six
|
||||
|
||||
from petstore_api.models.outer_enum import OuterEnum # noqa: F401,E501
|
||||
|
||||
|
||||
class EnumTest(object):
|
||||
"""NOTE: This class is auto generated by OpenAPI Generator.
|
||||
|
||||
@@ -15,8 +15,6 @@ import re # noqa: F401
|
||||
|
||||
import six
|
||||
|
||||
from petstore_api.models.animal import Animal # noqa: F401,E501
|
||||
|
||||
|
||||
class MixedPropertiesAndAdditionalPropertiesClass(object):
|
||||
"""NOTE: This class is auto generated by OpenAPI Generator.
|
||||
|
||||
@@ -15,9 +15,6 @@ import re # noqa: F401
|
||||
|
||||
import six
|
||||
|
||||
from petstore_api.models.category import Category # noqa: F401,E501
|
||||
from petstore_api.models.tag import Tag # noqa: F401,E501
|
||||
|
||||
|
||||
class Pet(object):
|
||||
"""NOTE: This class is auto generated by OpenAPI Generator.
|
||||
|
||||
@@ -102,7 +102,7 @@ class Animal(object):
|
||||
|
||||
def get_real_child_model(self, data):
|
||||
"""Returns the real base class specified by the discriminator"""
|
||||
discriminator_value = data[self.discriminator].lower()
|
||||
discriminator_value = data[self.discriminator]
|
||||
return self.discriminator_value_class_map.get(discriminator_value)
|
||||
|
||||
def to_dict(self):
|
||||
|
||||
@@ -15,8 +15,6 @@ import re # noqa: F401
|
||||
|
||||
import six
|
||||
|
||||
from petstore_api.models.animal import Animal # noqa: F401,E501
|
||||
|
||||
|
||||
class AnimalFarm(object):
|
||||
"""NOTE: This class is auto generated by OpenAPI Generator.
|
||||
|
||||
@@ -15,8 +15,6 @@ import re # noqa: F401
|
||||
|
||||
import six
|
||||
|
||||
from petstore_api.models.read_only_first import ReadOnlyFirst # noqa: F401,E501
|
||||
|
||||
|
||||
class ArrayTest(object):
|
||||
"""NOTE: This class is auto generated by OpenAPI Generator.
|
||||
|
||||
@@ -15,8 +15,6 @@ import re # noqa: F401
|
||||
|
||||
import six
|
||||
|
||||
from petstore_api.models.animal import Animal # noqa: F401,E501
|
||||
|
||||
|
||||
class Cat(object):
|
||||
"""NOTE: This class is auto generated by OpenAPI Generator.
|
||||
|
||||
@@ -15,8 +15,6 @@ import re # noqa: F401
|
||||
|
||||
import six
|
||||
|
||||
from petstore_api.models.animal import Animal # noqa: F401,E501
|
||||
|
||||
|
||||
class Dog(object):
|
||||
"""NOTE: This class is auto generated by OpenAPI Generator.
|
||||
|
||||
@@ -15,8 +15,6 @@ import re # noqa: F401
|
||||
|
||||
import six
|
||||
|
||||
from petstore_api.models.outer_enum import OuterEnum # noqa: F401,E501
|
||||
|
||||
|
||||
class EnumTest(object):
|
||||
"""NOTE: This class is auto generated by OpenAPI Generator.
|
||||
|
||||
@@ -15,8 +15,6 @@ import re # noqa: F401
|
||||
|
||||
import six
|
||||
|
||||
from petstore_api.models.animal import Animal # noqa: F401,E501
|
||||
|
||||
|
||||
class MixedPropertiesAndAdditionalPropertiesClass(object):
|
||||
"""NOTE: This class is auto generated by OpenAPI Generator.
|
||||
|
||||
@@ -15,9 +15,6 @@ import re # noqa: F401
|
||||
|
||||
import six
|
||||
|
||||
from petstore_api.models.category import Category # noqa: F401,E501
|
||||
from petstore_api.models.tag import Tag # noqa: F401,E501
|
||||
|
||||
|
||||
class Pet(object):
|
||||
"""NOTE: This class is auto generated by OpenAPI Generator.
|
||||
|
||||
@@ -102,7 +102,7 @@ class Animal(object):
|
||||
|
||||
def get_real_child_model(self, data):
|
||||
"""Returns the real base class specified by the discriminator"""
|
||||
discriminator_value = data[self.discriminator].lower()
|
||||
discriminator_value = data[self.discriminator]
|
||||
return self.discriminator_value_class_map.get(discriminator_value)
|
||||
|
||||
def to_dict(self):
|
||||
|
||||
@@ -15,8 +15,6 @@ import re # noqa: F401
|
||||
|
||||
import six
|
||||
|
||||
from petstore_api.models.animal import Animal # noqa: F401,E501
|
||||
|
||||
|
||||
class AnimalFarm(object):
|
||||
"""NOTE: This class is auto generated by OpenAPI Generator.
|
||||
|
||||
@@ -15,8 +15,6 @@ import re # noqa: F401
|
||||
|
||||
import six
|
||||
|
||||
from petstore_api.models.read_only_first import ReadOnlyFirst # noqa: F401,E501
|
||||
|
||||
|
||||
class ArrayTest(object):
|
||||
"""NOTE: This class is auto generated by OpenAPI Generator.
|
||||
|
||||
@@ -15,8 +15,6 @@ import re # noqa: F401
|
||||
|
||||
import six
|
||||
|
||||
from petstore_api.models.animal import Animal # noqa: F401,E501
|
||||
|
||||
|
||||
class Cat(object):
|
||||
"""NOTE: This class is auto generated by OpenAPI Generator.
|
||||
|
||||
@@ -15,8 +15,6 @@ import re # noqa: F401
|
||||
|
||||
import six
|
||||
|
||||
from petstore_api.models.animal import Animal # noqa: F401,E501
|
||||
|
||||
|
||||
class Dog(object):
|
||||
"""NOTE: This class is auto generated by OpenAPI Generator.
|
||||
|
||||
@@ -15,8 +15,6 @@ import re # noqa: F401
|
||||
|
||||
import six
|
||||
|
||||
from petstore_api.models.outer_enum import OuterEnum # noqa: F401,E501
|
||||
|
||||
|
||||
class EnumTest(object):
|
||||
"""NOTE: This class is auto generated by OpenAPI Generator.
|
||||
|
||||
@@ -15,8 +15,6 @@ import re # noqa: F401
|
||||
|
||||
import six
|
||||
|
||||
from petstore_api.models.animal import Animal # noqa: F401,E501
|
||||
|
||||
|
||||
class MixedPropertiesAndAdditionalPropertiesClass(object):
|
||||
"""NOTE: This class is auto generated by OpenAPI Generator.
|
||||
|
||||
@@ -15,9 +15,6 @@ import re # noqa: F401
|
||||
|
||||
import six
|
||||
|
||||
from petstore_api.models.category import Category # noqa: F401,E501
|
||||
from petstore_api.models.tag import Tag # noqa: F401,E501
|
||||
|
||||
|
||||
class Pet(object):
|
||||
"""NOTE: This class is auto generated by OpenAPI Generator.
|
||||
|
||||
@@ -71,6 +71,21 @@ class DeserializationTests(unittest.TestCase):
|
||||
self.assertTrue(isinstance(deserialized, dict))
|
||||
self.assertTrue(isinstance(deserialized['pet'], petstore_api.Pet))
|
||||
|
||||
def test_deserialize_dict_str_dog(self):
|
||||
""" deserialize dict(str, Dog), use discriminator"""
|
||||
data = {
|
||||
'dog': {
|
||||
"id": 0,
|
||||
"className": "Dog",
|
||||
"color": "white",
|
||||
"bread": "Jack Russel Terrier"
|
||||
}
|
||||
}
|
||||
|
||||
deserialized = self.deserialize(data, 'dict(str, Animal)')
|
||||
self.assertTrue(isinstance(deserialized, dict))
|
||||
self.assertTrue(isinstance(deserialized['dog'], petstore_api.Dog))
|
||||
|
||||
def test_deserialize_dict_str_int(self):
|
||||
""" deserialize dict(str, int) """
|
||||
data = {
|
||||
|
||||
Reference in New Issue
Block a user