openapi-generator/samples/server/petstore/flaskConnexion/swagger_server/encoder.py

18 lines
616 B
Python

from six import iteritems
from swagger_server.models.base_model_ import Model
from connexion.apps.flask_app import FlaskJSONEncoder
class JSONEncoder(FlaskJSONEncoder):
include_nulls = False
def default(self, o):
if isinstance(o, Model):
dikt = {}
for attr, _ in iteritems(o.swagger_types):
value = getattr(o, attr)
if value is None and not self.include_nulls:
continue
attr = o.attribute_map[attr]
dikt[attr] = value
return dikt
return FlaskJSONEncoder.default(self, o)