Added customisation hooks for FieldList sub fields (MongoEngine#19)

Author: rozza

docs/changelog.rst

@@ -4,6 +4,7 @@ Changelog
 
 Changes in 0.7
 ==============
+- Added customisation hooks for FieldList sub fields (#19)
 - Handle non ascii chars in the MongoDebugPanel (#22)
 - Fixed toolbar stacktrace if a html directory is in the path (#31)
 - ModelForms no longer patch Document.update (#32)

flask_mongoengine/wtf/models.py

@@ -3,6 +3,7 @@
 
 class ModelForm(Form):
     """A WTForms mongoengine model form"""
+
     def __init__(self, formdata=None, obj=None, prefix='', **kwargs):
         self.instance = (kwargs.pop('instance', None) or kwargs.get('obj', None))
         if self.instance and not formdata:

flask_mongoengine/wtf/orm.py

@@ -141,8 +141,9 @@ def conv_List(self, model, field, kwargs):
             return ModelSelectMultipleField(model=field.field.document_type, **kwargs)
         if field.field.choices:
             kwargs['multiple'] = True
-            return self.convert(model, field.field, kwargs)
-        unbound_field = self.convert(model, field.field, {})
+            return self.convert(model, field.field, **kwargs)
+        sub_field_args = kwargs.pop("sub_field_args", {})
+        unbound_field = self.convert(model, field.field, sub_field_args)
         unacceptable = {
             'validators': [],
             'filters': [],

tests.py

@@ -284,5 +284,26 @@ class Item (db.Document):
 
             self.assertEqual(1, Item.objects.count())
 
+    def test_sub_field_args(self):
+        with self.app.test_request_context('/'):
+            db = self.db
+
+            class TestModel(db.Document):
+                lst = db.ListField(db.StringField())
+
+            field_args = {'lst': {'label': 'Custom Label',
+                                  'sub_field_args': {'widget': wtforms.widgets.HiddenInput(),
+                                                     'label': "Hidden Input"}}}
+            CustomForm = model_form(TestModel, field_args=field_args)
+
+            custom_form = CustomForm(obj=TestModel(lst=["Foo"]))
+            list_label = flask.render_template_string("{{ custom_form.lst.label }}", custom_form=custom_form)
+            self.assertTrue("Custom Label" in list_label)
+            self.assertTrue("Hidden Input" not in list_label)
+
+            sub_label = flask.render_template_string("{{ custom_form.lst }}", custom_form=custom_form)
+            self.assertTrue("Hidden Input" in sub_label)
+
+
 if __name__ == '__main__':
     unittest.main()