Skip to content

Commit

Permalink
Add checks for OrderedModel in swagger.py
Browse files Browse the repository at this point in the history
Add check for OrderedModel in Swagger.register_model(self, model)
Add check for OrderedModel in ref(model)
  • Loading branch information
strutt committed Jul 4, 2019
1 parent 3f22236 commit 6c443d3
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions flask_restplus/swagger.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down Expand Up @@ -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)}


Expand Down Expand Up @@ -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)
Expand Down

0 comments on commit 6c443d3

Please sign in to comment.