Commit d3af9d0c authored by Serge S. Koval's avatar Serge S. Koval

Merge pull request #645 from LevonXXL/master

issue #636
parents 338eec36 7940022d
from mongoengine import ReferenceField
from mongoengine.base import BaseDocument, DocumentMetaclass
from mongoengine.base import BaseDocument, DocumentMetaclass, get_document
from wtforms import fields, validators
from flask.ext.mongoengine.wtf import orm, fields as mongo_fields
......@@ -205,6 +205,10 @@ def get_form(model, converter,
A converter to generate the fields based on the model properties. If
not set, ``ModelConverter`` is used.
"""
if isinstance(model, str):
model = get_document(model)
if not isinstance(model, (BaseDocument, DocumentMetaclass)):
raise TypeError('Model must be a mongoengine Document schema')
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment