154 lines
4.5 KiB
Python
154 lines
4.5 KiB
Python
from dominate import tags
|
|
from dominate.util import raw
|
|
from flask import current_app
|
|
from markupsafe import Markup
|
|
from visitor import Visitor
|
|
|
|
|
|
def render_form(form, **kwargs):
|
|
r = WTFormsRenderer(**kwargs)
|
|
|
|
return Markup(r.visit(form))
|
|
|
|
|
|
class WTFormsRenderer(Visitor):
|
|
def __init__(self,
|
|
action='',
|
|
id=None,
|
|
method='post',
|
|
extra_classes=[],
|
|
role='form',
|
|
enctype=None):
|
|
self.action = action
|
|
self.form_id = id
|
|
self.method = method
|
|
self.extra_classes = extra_classes
|
|
self.role = role
|
|
self.enctype = enctype
|
|
|
|
def _visited_file_field(self):
|
|
if self._real_enctype is None:
|
|
self._real_enctype = u'multipart/form-data'
|
|
|
|
def _get_wrap(self, node, classes='form-group'):
|
|
# add required class, which strictly speaking isn't bootstrap, but
|
|
# a common enough customization
|
|
if node.flags.required:
|
|
classes += ' required'
|
|
|
|
div = tags.div(_class=classes)
|
|
if current_app.debug:
|
|
div.add(tags.comment(' Field: {} ({}) '.format(
|
|
node.name, node.__class__.__name__)))
|
|
|
|
return div
|
|
|
|
def _wrapped_input(self, node,
|
|
type='text',
|
|
classes=['form-control'], **kwargs):
|
|
wrap = self._get_wrap(node)
|
|
wrap.add(tags.label(node.label.text, _for=node.id))
|
|
wrap.add(tags.input(type=type, _class=' '.join(classes), **kwargs))
|
|
|
|
return wrap
|
|
|
|
def visit_BooleanField(self, node):
|
|
wrap = self._get_wrap(node, classes='checkbox')
|
|
|
|
label = wrap.add(tags.label(_for=node.id))
|
|
label.add(tags.input(type='checkbox'))
|
|
label.add(node.label.text)
|
|
|
|
return wrap
|
|
|
|
def visit_DateField(self, node):
|
|
return self._wrapped_input(node, 'date')
|
|
|
|
def visit_DateTimeField(self, node):
|
|
return self._wrapped_input(node, 'datetime-local')
|
|
|
|
def visit_DecimalField(self, node):
|
|
# FIXME: if range-validator is present, add limits?
|
|
return self._wrapped_input(node, 'number')
|
|
|
|
def visit_EmailField(self, node):
|
|
# note: WTForms does not actually have an EmailField, this function
|
|
# is called by visit_StringField based on which validators are enabled
|
|
return self._wrapped_input(node, 'email')
|
|
|
|
def visit_Field(self, node):
|
|
# FIXME: add error class
|
|
|
|
wrap = self._get_wrap(node)
|
|
|
|
# add the label
|
|
wrap.add(tags.label(node.label.text, _for=node.id))
|
|
wrap.add(raw(node()))
|
|
|
|
if node.description:
|
|
wrap.add(tags.p(node.description, _class='help-block'))
|
|
|
|
return wrap
|
|
|
|
def visit_FileField(self, node):
|
|
self._visited_file_field()
|
|
return self._wrapped_input(node, 'file', classes=[])
|
|
|
|
def visit_FloatField(self, node):
|
|
# FIXME: if range-validator is present, add limits?
|
|
return self._wrapped_input(node, 'number')
|
|
|
|
def visit_Form(self, node):
|
|
form = tags.form(_class=' '.join(['form'] + self.extra_classes))
|
|
|
|
if self.action:
|
|
form['action'] = self.action
|
|
|
|
if self.form_id:
|
|
form['id'] = self.form_id
|
|
|
|
if self.method:
|
|
form['method'] = self.method
|
|
|
|
# prepare enctype, this will be auto-updated by file fields if
|
|
# necessary
|
|
self._real_enctype = self.enctype
|
|
|
|
# render fields
|
|
for field in node:
|
|
elem = self.visit(field)
|
|
form.add(elem)
|
|
|
|
if self._real_enctype:
|
|
form['enctype'] = self._real_enctype
|
|
|
|
return form
|
|
|
|
def visit_HiddenField(self, node):
|
|
return raw(node())
|
|
|
|
def visit_IntegerField(self, node):
|
|
# FIXME: if range-validator is present, add limits?
|
|
return self._wrapped_input(node, 'number', step=1)
|
|
|
|
def visit_PasswordField(self, node):
|
|
return self._wrapped_input(node, 'password')
|
|
|
|
def visit_SubmitField(self, node):
|
|
button = tags.button(node.label.text,
|
|
_class='btn btn-default',
|
|
type='submit')
|
|
return button
|
|
|
|
def visit_TextField(self, node):
|
|
# legacy support for TextField, deprecated in WTForms 2.0
|
|
return self.visit_StringField(node)
|
|
|
|
def visit_StringField(self, node):
|
|
for v in node.validators:
|
|
if v.__class__.__name__ == 'Email':
|
|
# render email fields differently
|
|
return self.visit_EmailField(node)
|
|
|
|
return self._wrapped_input(node, 'text')
|