czfadmin · December 22, 2018 08:53 · Feb 17, 2016
diff --git a/button_field_example.py b/button_field_example.py
@@ -0,0 +1,62 @@
+from flask import Flask, render_template_string
+from flask_wtf import Form
+from wtforms import StringField
+from wtforms.widgets import html_params, HTMLString
+
+app = Flask(__name__)
+app.secret_key = 'SHH!'
+
+
+class ButtonWidget(object):
+    """
+    Renders a multi-line text area.
+    `rows` and `cols` ought to be passed as keyword args when rendering.
+    """
+    input_type = 'submit'
+
+    html_params = staticmethod(html_params)
+
+    def __call__(self, field, **kwargs):
+        kwargs.setdefault('id', field.id)
+        kwargs.setdefault('type', self.input_type)
+        if 'value' not in kwargs:
+            kwargs['value'] = field._value()
+
+        return HTMLString('<button {params}>{label}</button>'.format(
+            params=self.html_params(name=field.name, **kwargs),
+            label=field.label.text)
+        )
+
+
+class ButtonField(StringField):
+    widget = ButtonWidget()
+
+
+class TestForm(Form):
+    choice_a = ButtonField('example')
+    choice_b = ButtonField('other')
+
+
+@app.route('/', methods=['post', 'get'])
+def home():
+    form = TestForm()
+    if form.is_submitted():
+        if form.choice_a.data:
+            return form.choice_a.data
+        else:
+            return form.choice_b.data
+
+    form.choice_a.data = 'First Choice'
+    form.choice_b.data = 'Second Choice'
+
+    return render_template_string('''
+        <form action="" method="post">
+            {{ form.hidden_tag() }}
+            {{ form.choice_a() }}
+            {{ form.choice_b }}
+        </form>
+    ''', form=form)
+
+
+if __name__ == '__main__':
+    app.run(debug=True)