From 6c443d39bedeabbec03c25a67cb57c4937e5f2be Mon Sep 17 00:00:00 2001 From: Ben Strutt Date: Thu, 4 Jul 2019 16:32:42 +0100 Subject: [PATCH] Add checks for OrderedModel in swagger.py Add check for OrderedModel in Swagger.register_model(self, model) Add check for OrderedModel in ref(model) --- flask_restplus/swagger.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/flask_restplus/swagger.py b/flask_restplus/swagger.py index 681b08d2..3f71050a 100644 --- a/flask_restplus/swagger.py +++ b/flask_restplus/swagger.py @@ -16,7 +16,7 @@ from werkzeug.routing import parse_rule from . import fields -from .model import Model, ModelBase +from .model import Model, OrderedModel, ModelBase from .reqparse import RequestParser from .utils import merge, not_none, not_none_sorted from ._http import HTTPStatus @@ -50,7 +50,7 @@ def ref(model): '''Return a reference to model in definitions''' - name = model.name if isinstance(model, ModelBase) else model + name = model.name if isinstance(model, (ModelBase, OrderedModel)) else model return {'$ref': '#/definitions/{0}'.format(name)} @@ -578,7 +578,7 @@ def register_model(self, model): if isinstance(specs, ModelBase): for parent in specs.__parents__: self.register_model(parent) - if isinstance(specs, Model): + if isinstance(specs, (Model, OrderedModel)): for field in itervalues(specs): self.register_field(field) return ref(model)