90 lines
2.3 KiB
Python
90 lines
2.3 KiB
Python
from flask_wtf import Form
|
|
|
|
|
|
class ModelForm(Form):
|
|
"""
|
|
wtforms_components exposes ModelForm but their ModelForm does not inherit
|
|
from flask_wtf's Form, but instead WTForm's Form.
|
|
|
|
However, in order to get CSRF protection handled by default we need to
|
|
inherit from flask_wtf's Form. So let's just copy his class directly.
|
|
|
|
We modified it by removing the format argument so that wtforms_component
|
|
uses its own default which is to pass in request.form automatically.
|
|
"""
|
|
def __init__(self, obj=None, prefix='', **kwargs):
|
|
Form.__init__(
|
|
self, obj=obj, prefix=prefix, **kwargs
|
|
)
|
|
self._obj = obj
|
|
|
|
|
|
def choices_from_dict(source, prepend_blank=True):
|
|
"""
|
|
Convert a dict to a format that's compatible with WTForm's choices. It also
|
|
optionally prepends a "Please select one..." value.
|
|
|
|
Example:
|
|
# Convert this data structure:
|
|
STATUS = OrderedDict([
|
|
('unread', 'Unread'),
|
|
('open', 'Open'),
|
|
('contacted', 'Contacted'),
|
|
('closed', 'Closed')
|
|
])
|
|
|
|
# Into this:
|
|
choices = [('', 'Please select one...'), ('unread', 'Unread) ...]
|
|
|
|
:param source: Input source
|
|
:type source: dict
|
|
:param prepend_blank: An optional blank item
|
|
:type prepend_blank: bool
|
|
:return: list
|
|
"""
|
|
choices = []
|
|
|
|
if prepend_blank:
|
|
choices.append(('', 'Please select one...'))
|
|
|
|
for key, value in source.items():
|
|
pair = (key, value)
|
|
choices.append(pair)
|
|
|
|
return choices
|
|
|
|
|
|
def choices_from_list(source, prepend_blank=True):
|
|
"""
|
|
Convert a list to a format that's compatible with WTForm's choices. It also
|
|
optionally prepends a "Please select one..." value.
|
|
|
|
Example:
|
|
# Convert this data structure:
|
|
TIMEZONES = (
|
|
'Africa/Abidjan',
|
|
'Africa/Accra',
|
|
'Africa/Addis_Ababa'
|
|
)
|
|
|
|
# Into this:
|
|
choices = [('', 'Please select one...'),
|
|
('Africa/Abidjan', 'Africa/Abidjan) ...]
|
|
|
|
:param source: Input source
|
|
:type source: list or tuple
|
|
:param prepend_blank: An optional blank item
|
|
:type prepend_blank: bool
|
|
:return: list
|
|
"""
|
|
choices = []
|
|
|
|
if prepend_blank:
|
|
choices.append(('', 'Please select one...'))
|
|
|
|
for item in source:
|
|
pair = (item, item)
|
|
choices.append(pair)
|
|
|
|
return choices
|