diff --git a/flask_mongoengine/wtf/models.py b/flask_mongoengine/wtf/models.py index abd473f5..857b40cf 100644 --- a/flask_mongoengine/wtf/models.py +++ b/flask_mongoengine/wtf/models.py @@ -1,6 +1,9 @@ -from flask_wtf import Form +# -*- coding:utf-8 -*- -class ModelForm(Form): +from flask_wtf import FlaskForm + + +class ModelForm(FlaskForm): """A WTForms mongoengine model form""" def __init__(self, formdata=None, obj=None, **kwargs): @@ -18,4 +21,4 @@ def save(self, commit=True, **kwargs): if commit: self.instance.save(**kwargs) - return self.instance \ No newline at end of file + return self.instance diff --git a/setup.py b/setup.py index 7e9a21d6..795e5a3c 100644 --- a/setup.py +++ b/setup.py @@ -51,8 +51,8 @@ def load_module(module_name, script_file): platforms='any', install_requires=[ 'Flask>=0.8', + 'Flask-WTF>=0.13', 'mongoengine>=0.8.0', - 'flask-wtf', ], packages=['flask_mongoengine', 'flask_mongoengine.wtf'],