Commit 62eab644 authored by Serge S. Koval's avatar Serge S. Koval

Fixed #252. Mongoengine form generation with choices now works

parent b23f46a6
from mongoengine import ReferenceField from mongoengine import ReferenceField
from wtforms import validators from wtforms import fields, validators
from flask.ext.mongoengine.wtf import orm, fields from flask.ext.mongoengine.wtf import orm, fields as mongo_fields
from flask.ext.admin import form from flask.ext.admin import form
from flask.ext.admin.model.fields import InlineFieldList from flask.ext.admin.model.fields import InlineFieldList
...@@ -82,7 +82,7 @@ class CustomModelConverter(orm.ModelConverter): ...@@ -82,7 +82,7 @@ class CustomModelConverter(orm.ModelConverter):
kwargs['widget'] = form.Select2Widget(multiple=True) kwargs['widget'] = form.Select2Widget(multiple=True)
doc_type = field.field.document_type doc_type = field.field.document_type
return fields.ModelSelectMultipleField(model=doc_type, **kwargs) return mongo_fields.ModelSelectMultipleField(model=doc_type, **kwargs)
if field.field.choices: if field.field.choices:
kwargs['multiple'] = True kwargs['multiple'] = True
return self.convert(model, field.field, kwargs) return self.convert(model, field.field, kwargs)
......
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