757 lines
31 KiB
Python
757 lines
31 KiB
Python
from django.core import validators
|
|
from django.core.exceptions import PermissionDenied
|
|
from django.utils.html import escape
|
|
|
|
FORM_FIELD_ID_PREFIX = 'id_'
|
|
|
|
class EmptyValue(Exception):
|
|
"This is raised when empty data is provided"
|
|
pass
|
|
|
|
class Manipulator:
|
|
# List of permission strings. User must have at least one to manipulate.
|
|
# None means everybody has permission.
|
|
required_permission = ''
|
|
|
|
def __init__(self):
|
|
# List of FormField objects
|
|
self.fields = []
|
|
|
|
def __getitem__(self, field_name):
|
|
"Looks up field by field name; raises KeyError on failure"
|
|
for field in self.fields:
|
|
if field.field_name == field_name:
|
|
return field
|
|
raise KeyError, "Field %s not found" % field_name
|
|
|
|
def __delitem__(self, field_name):
|
|
"Deletes the field with the given field name; raises KeyError on failure"
|
|
for i, field in enumerate(self.fields):
|
|
if field.field_name == field_name:
|
|
del self.fields[i]
|
|
return
|
|
raise KeyError, "Field %s not found" % field_name
|
|
|
|
def check_permissions(self, user):
|
|
"""Confirms user has required permissions to use this manipulator; raises
|
|
PermissionDenied on failure."""
|
|
if self.required_permission is None:
|
|
return
|
|
if user.has_perm(self.required_permission):
|
|
return
|
|
raise PermissionDenied
|
|
|
|
def prepare(self, new_data):
|
|
"""
|
|
Makes any necessary preparations to new_data, in place, before data has
|
|
been validated.
|
|
"""
|
|
for field in self.fields:
|
|
field.prepare(new_data)
|
|
|
|
def get_validation_errors(self, new_data):
|
|
"Returns dictionary mapping field_names to error-message lists"
|
|
errors = {}
|
|
for field in self.fields:
|
|
if field.is_required and not new_data.get(field.field_name, False):
|
|
errors.setdefault(field.field_name, []).append('This field is required.')
|
|
continue
|
|
try:
|
|
validator_list = field.validator_list
|
|
if hasattr(self, 'validate_%s' % field.field_name):
|
|
validator_list.append(getattr(self, 'validate_%s' % field.field_name))
|
|
for validator in validator_list:
|
|
if field.is_required or new_data.get(field.field_name, False) or hasattr(validator, 'always_test'):
|
|
try:
|
|
if hasattr(field, 'requires_data_list'):
|
|
validator(new_data.getlist(field.field_name), new_data)
|
|
else:
|
|
validator(new_data.get(field.field_name, ''), new_data)
|
|
except validators.ValidationError, e:
|
|
errors.setdefault(field.field_name, []).extend(e.messages)
|
|
# If a CriticalValidationError is raised, ignore any other ValidationErrors
|
|
# for this particular field
|
|
except validators.CriticalValidationError, e:
|
|
errors.setdefault(field.field_name, []).extend(e.messages)
|
|
return errors
|
|
|
|
def save(self, new_data):
|
|
"Saves the changes and returns the new object"
|
|
# changes is a dictionary-like object keyed by field_name
|
|
raise NotImplementedError
|
|
|
|
def do_html2python(self, new_data):
|
|
"""
|
|
Convert the data from HTML data types to Python datatypes, changing the
|
|
object in place. This happens after validation but before storage. This
|
|
must happen after validation because html2python functions aren't
|
|
expected to deal with invalid input.
|
|
"""
|
|
for field in self.fields:
|
|
if new_data.has_key(field.field_name):
|
|
new_data.setlist(field.field_name,
|
|
[field.__class__.html2python(data) for data in new_data.getlist(field.field_name)])
|
|
else:
|
|
try:
|
|
# individual fields deal with None values themselves
|
|
new_data.setlist(field.field_name, [field.__class__.html2python(None)])
|
|
except EmptyValue:
|
|
new_data.setlist(field.field_name, [])
|
|
|
|
class FormWrapper:
|
|
"""
|
|
A wrapper linking a Manipulator to the template system.
|
|
This allows dictionary-style lookups of formfields. It also handles feeding
|
|
prepopulated data and validation error messages to the formfield objects.
|
|
"""
|
|
def __init__(self, manipulator, data, error_dict):
|
|
self.manipulator, self.data = manipulator, data
|
|
self.error_dict = error_dict
|
|
|
|
def __repr__(self):
|
|
return repr(self.data)
|
|
|
|
def __getitem__(self, key):
|
|
for field in self.manipulator.fields:
|
|
if field.field_name == key:
|
|
if hasattr(field, 'requires_data_list') and hasattr(self.data, 'getlist'):
|
|
data = self.data.getlist(field.field_name)
|
|
else:
|
|
data = self.data.get(field.field_name, None)
|
|
if data is None:
|
|
data = ''
|
|
return FormFieldWrapper(field, data, self.error_dict.get(field.field_name, []))
|
|
raise KeyError
|
|
|
|
def has_errors(self):
|
|
return self.error_dict != {}
|
|
|
|
class FormFieldWrapper:
|
|
"A bridge between the template system and an individual form field. Used by FormWrapper."
|
|
def __init__(self, formfield, data, error_list):
|
|
self.formfield, self.data, self.error_list = formfield, data, error_list
|
|
self.field_name = self.formfield.field_name # for convenience in templates
|
|
|
|
def __str__(self):
|
|
"Renders the field"
|
|
return str(self.formfield.render(self.data))
|
|
|
|
def __repr__(self):
|
|
return '<FormFieldWrapper for "%s">' % self.formfield.field_name
|
|
|
|
def field_list(self):
|
|
"""
|
|
Like __str__(), but returns a list. Use this when the field's render()
|
|
method returns a list.
|
|
"""
|
|
return self.formfield.render(self.data)
|
|
|
|
def errors(self):
|
|
return self.error_list
|
|
|
|
def html_error_list(self):
|
|
if self.errors():
|
|
return '<ul class="errorlist"><li>%s</li></ul>' % '</li><li>'.join([escape(e) for e in self.errors()])
|
|
else:
|
|
return ''
|
|
|
|
class FormFieldCollection(FormFieldWrapper):
|
|
"A utility class that gives the template access to a dict of FormFieldWrappers"
|
|
def __init__(self, formfield_dict):
|
|
self.formfield_dict = formfield_dict
|
|
|
|
def __str__(self):
|
|
return str(self.formfield_dict)
|
|
|
|
def __getitem__(self, template_key):
|
|
"Look up field by template key; raise KeyError on failure"
|
|
return self.formfield_dict[template_key]
|
|
|
|
def __repr__(self):
|
|
return "<FormFieldCollection: %s>" % self.formfield_dict
|
|
|
|
def errors(self):
|
|
"Returns list of all errors in this collection's formfields"
|
|
errors = []
|
|
for field in self.formfield_dict.values():
|
|
errors.extend(field.errors())
|
|
return errors
|
|
|
|
class FormField:
|
|
"""Abstract class representing a form field.
|
|
|
|
Classes that extend FormField should define the following attributes:
|
|
field_name
|
|
The field's name for use by programs.
|
|
validator_list
|
|
A list of validation tests (callback functions) that the data for
|
|
this field must pass in order to be added or changed.
|
|
is_required
|
|
A Boolean. Is it a required field?
|
|
Subclasses should also implement a render(data) method, which is responsible
|
|
for rending the form field in XHTML.
|
|
"""
|
|
def __str__(self):
|
|
return self.render('')
|
|
|
|
def __repr__(self):
|
|
return 'FormField "%s"' % self.field_name
|
|
|
|
def prepare(self, new_data):
|
|
"Hook for doing something to new_data (in place) before validation."
|
|
pass
|
|
|
|
def html2python(data):
|
|
"Hook for converting an HTML datatype (e.g. 'on' for checkboxes) to a Python type"
|
|
return data
|
|
html2python = staticmethod(html2python)
|
|
|
|
def render(self, data):
|
|
raise NotImplementedError
|
|
|
|
####################
|
|
# GENERIC WIDGETS #
|
|
####################
|
|
|
|
class TextField(FormField):
|
|
def __init__(self, field_name, length=30, maxlength=None, is_required=False, validator_list=[]):
|
|
self.field_name = field_name
|
|
self.length, self.maxlength = length, maxlength
|
|
self.is_required = is_required
|
|
self.validator_list = [self.isValidLength, self.hasNoNewlines] + validator_list
|
|
|
|
def isValidLength(self, data, form):
|
|
if data and self.maxlength and len(data) > self.maxlength:
|
|
raise validators.ValidationError, "Ensure your text is less than %s characters." % self.maxlength
|
|
|
|
def hasNoNewlines(self, data, form):
|
|
if data and '\n' in data:
|
|
raise validators.ValidationError, "Line breaks are not allowed here."
|
|
|
|
def render(self, data):
|
|
if data is None:
|
|
data = ''
|
|
maxlength = ''
|
|
if self.maxlength:
|
|
maxlength = 'maxlength="%s" ' % self.maxlength
|
|
if isinstance(data, unicode):
|
|
data = data.encode('utf-8')
|
|
return '<input type="text" id="%s" class="v%s%s" name="%s" size="%s" value="%s" %s/>' % \
|
|
(FORM_FIELD_ID_PREFIX + self.field_name, self.__class__.__name__, self.is_required and ' required' or '',
|
|
self.field_name, self.length, escape(data), maxlength)
|
|
|
|
def html2python(data):
|
|
return data
|
|
html2python = staticmethod(html2python)
|
|
|
|
class PasswordField(TextField):
|
|
def render(self, data):
|
|
# value is always blank because we never want to redisplay it
|
|
return '<input type="password" id="%s" class="v%s%s" name="%s" value="" />' % \
|
|
(FORM_FIELD_ID_PREFIX + self.field_name, self.__class__.__name__, self.is_required and ' required' or '',
|
|
self.field_name)
|
|
|
|
class LargeTextField(TextField):
|
|
def __init__(self, field_name, rows=10, cols=40, is_required=False, validator_list=[], maxlength=None):
|
|
self.field_name = field_name
|
|
self.rows, self.cols, self.is_required = rows, cols, is_required
|
|
self.validator_list = validator_list[:]
|
|
if maxlength:
|
|
self.validator_list.append(self.isValidLength)
|
|
self.maxlength = maxlength
|
|
|
|
def render(self, data):
|
|
if data is None:
|
|
data = ''
|
|
if isinstance(data, unicode):
|
|
data = data.encode('utf-8')
|
|
return '<textarea id="%s" class="v%s%s" name="%s" rows="%s" cols="%s">%s</textarea>' % \
|
|
(FORM_FIELD_ID_PREFIX + self.field_name, self.__class__.__name__, self.is_required and ' required' or '',
|
|
self.field_name, self.rows, self.cols, escape(data))
|
|
|
|
class HiddenField(FormField):
|
|
def __init__(self, field_name, is_required=False, validator_list=[]):
|
|
self.field_name, self.is_required = field_name, is_required
|
|
self.validator_list = validator_list[:]
|
|
|
|
def render(self, data):
|
|
return '<input type="hidden" id="%s" name="%s" value="%s" />' % \
|
|
(FORM_FIELD_ID_PREFIX + self.field_name, self.field_name, escape(data))
|
|
|
|
class CheckboxField(FormField):
|
|
def __init__(self, field_name, checked_by_default=False):
|
|
self.field_name = field_name
|
|
self.checked_by_default = checked_by_default
|
|
self.is_required, self.validator_list = False, [] # because the validator looks for these
|
|
|
|
def render(self, data):
|
|
checked_html = ''
|
|
if data or (data is '' and self.checked_by_default):
|
|
checked_html = ' checked="checked"'
|
|
return '<input type="checkbox" id="%s" class="v%s" name="%s"%s />' % \
|
|
(FORM_FIELD_ID_PREFIX + self.field_name, self.__class__.__name__,
|
|
self.field_name, checked_html)
|
|
|
|
def html2python(data):
|
|
"Convert value from browser ('on' or '') to a Python boolean"
|
|
if data == 'on':
|
|
return True
|
|
return False
|
|
html2python = staticmethod(html2python)
|
|
|
|
class SelectField(FormField):
|
|
def __init__(self, field_name, choices=[], size=1, is_required=False, validator_list=[]):
|
|
self.field_name = field_name
|
|
# choices is a list of (value, human-readable key) tuples because order matters
|
|
self.choices, self.size, self.is_required = choices, size, is_required
|
|
self.validator_list = [self.isValidChoice] + validator_list
|
|
|
|
def render(self, data):
|
|
output = ['<select id="%s" class="v%s%s" name="%s" size="%s">' % \
|
|
(FORM_FIELD_ID_PREFIX + self.field_name, self.__class__.__name__, self.is_required and ' required' or '',
|
|
self.field_name, self.size)]
|
|
str_data = str(data) # normalize to string
|
|
for value, display_name in self.choices:
|
|
selected_html = ''
|
|
if str(value) == str_data:
|
|
selected_html = ' selected="selected"'
|
|
output.append(' <option value="%s"%s>%s</option>' % (escape(value), selected_html, display_name))
|
|
output.append(' </select>')
|
|
return '\n'.join(output)
|
|
|
|
def isValidChoice(self, data, form):
|
|
str_data = str(data)
|
|
str_choices = [str(item[0]) for item in self.choices]
|
|
if str_data not in str_choices:
|
|
raise validators.ValidationError, "Select a valid choice; '%s' is not in %s." % (str_data, str_choices)
|
|
|
|
class NullSelectField(SelectField):
|
|
"This SelectField converts blank fields to None"
|
|
def html2python(data):
|
|
if not data:
|
|
return None
|
|
return data
|
|
html2python = staticmethod(html2python)
|
|
|
|
class RadioSelectField(FormField):
|
|
def __init__(self, field_name, choices=[], ul_class='', is_required=False, validator_list=[]):
|
|
self.field_name = field_name
|
|
# choices is a list of (value, human-readable key) tuples because order matters
|
|
self.choices, self.is_required = choices, is_required
|
|
self.validator_list = [self.isValidChoice] + validator_list
|
|
self.ul_class = ul_class
|
|
|
|
def render(self, data):
|
|
"""
|
|
Returns a special object, RadioFieldRenderer, that is iterable *and*
|
|
has a default str() rendered output.
|
|
|
|
This allows for flexible use in templates. You can just use the default
|
|
rendering:
|
|
|
|
{{ field_name }}
|
|
|
|
...which will output the radio buttons in an unordered list.
|
|
Or, you can manually traverse each radio option for special layout:
|
|
|
|
{% for option in field_name.field_list %}
|
|
{{ option.field }} {{ option.label }}<br />
|
|
{% endfor %}
|
|
"""
|
|
class RadioFieldRenderer:
|
|
def __init__(self, datalist, ul_class):
|
|
self.datalist, self.ul_class = datalist, ul_class
|
|
def __str__(self):
|
|
"Default str() output for this radio field -- a <ul>"
|
|
output = ['<ul%s>' % (self.ul_class and ' class="%s"' % self.ul_class or '')]
|
|
output.extend(['<li>%s %s</li>' % (d['field'], d['label']) for d in self.datalist])
|
|
output.append('</ul>')
|
|
return ''.join(output)
|
|
def __iter__(self):
|
|
for d in self.datalist:
|
|
yield d
|
|
def __len__(self):
|
|
return len(self.datalist)
|
|
datalist = []
|
|
str_data = str(data) # normalize to string
|
|
for i, (value, display_name) in enumerate(self.choices):
|
|
selected_html = ''
|
|
if str(value) == str_data:
|
|
selected_html = ' checked="checked"'
|
|
datalist.append({
|
|
'value': value,
|
|
'name': display_name,
|
|
'field': '<input type="radio" id="%s" name="%s" value="%s"%s/>' % \
|
|
(FORM_FIELD_ID_PREFIX + self.field_name + '_' + str(i), self.field_name, value, selected_html),
|
|
'label': '<label for="%s">%s</label>' % \
|
|
(FORM_FIELD_ID_PREFIX + self.field_name + '_' + str(i), display_name),
|
|
})
|
|
return RadioFieldRenderer(datalist, self.ul_class)
|
|
|
|
def isValidChoice(self, data, form):
|
|
str_data = str(data)
|
|
str_choices = [str(item[0]) for item in self.choices]
|
|
if str_data not in str_choices:
|
|
raise validators.ValidationError, "Select a valid choice; '%s' is not in %s." % (str_data, str_choices)
|
|
|
|
class NullBooleanField(SelectField):
|
|
"This SelectField provides 'Yes', 'No' and 'Unknown', mapping results to True, False or None"
|
|
def __init__(self, field_name, is_required=False, validator_list=[]):
|
|
SelectField.__init__(self, field_name, choices=[('1', 'Unknown'), ('2', 'Yes'), ('3', 'No')],
|
|
is_required=is_required, validator_list=validator_list)
|
|
|
|
def render(self, data):
|
|
if data is None: data = '1'
|
|
elif data == True: data = '2'
|
|
elif data == False: data = '3'
|
|
return SelectField.render(self, data)
|
|
|
|
def html2python(data):
|
|
return {'1': None, '2': True, '3': False}[data]
|
|
html2python = staticmethod(html2python)
|
|
|
|
class SelectMultipleField(SelectField):
|
|
requires_data_list = True
|
|
def render(self, data):
|
|
output = ['<select id="%s" class="v%s%s" name="%s" size="%s" multiple="multiple">' % \
|
|
(FORM_FIELD_ID_PREFIX + self.field_name, self.__class__.__name__, self.is_required and ' required' or '',
|
|
self.field_name, self.size)]
|
|
str_data_list = map(str, data) # normalize to strings
|
|
for value, choice in self.choices:
|
|
selected_html = ''
|
|
if str(value) in str_data_list:
|
|
selected_html = ' selected="selected"'
|
|
output.append(' <option value="%s"%s>%s</option>' % (escape(value), selected_html, choice))
|
|
output.append(' </select>')
|
|
return '\n'.join(output)
|
|
|
|
def isValidChoice(self, field_data, all_data):
|
|
# data is something like ['1', '2', '3']
|
|
str_choices = [str(item[0]) for item in self.choices]
|
|
for val in map(str, field_data):
|
|
if val not in str_choices:
|
|
raise validators.ValidationError, "Select a valid choice; '%s' is not in %s." % (val, str_choices)
|
|
|
|
def html2python(data):
|
|
if data is None:
|
|
raise EmptyValue
|
|
return data
|
|
html2python = staticmethod(html2python)
|
|
|
|
class CheckboxSelectMultipleField(SelectMultipleField):
|
|
"""
|
|
This has an identical interface to SelectMultipleField, except the rendered
|
|
widget is different. Instead of a <select multiple>, this widget outputs a
|
|
<ul> of <input type="checkbox">es.
|
|
|
|
Of course, that results in multiple form elements for the same "single"
|
|
field, so this class's prepare() method flattens the split data elements
|
|
back into the single list that validators, renderers and save() expect.
|
|
"""
|
|
requires_data_list = True
|
|
def __init__(self, field_name, choices=[], validator_list=[]):
|
|
SelectMultipleField.__init__(self, field_name, choices, size=1, is_required=False, validator_list=validator_list)
|
|
|
|
def prepare(self, new_data):
|
|
# new_data has "split" this field into several fields, so flatten it
|
|
# back into a single list.
|
|
data_list = []
|
|
for value, _ in self.choices:
|
|
if new_data.get('%s%s' % (self.field_name, value), '') == 'on':
|
|
data_list.append(value)
|
|
new_data.setlist(self.field_name, data_list)
|
|
|
|
def render(self, data):
|
|
output = ['<ul>']
|
|
str_data_list = map(str, data) # normalize to strings
|
|
for value, choice in self.choices:
|
|
checked_html = ''
|
|
if str(value) in str_data_list:
|
|
checked_html = ' checked="checked"'
|
|
field_name = '%s%s' % (self.field_name, value)
|
|
output.append('<li><input type="checkbox" id="%s%s" class="v%s" name="%s"%s /> <label for="%s%s">%s</label></li>' % \
|
|
(FORM_FIELD_ID_PREFIX, field_name, self.__class__.__name__, field_name, checked_html,
|
|
FORM_FIELD_ID_PREFIX, field_name, choice))
|
|
output.append('</ul>')
|
|
return '\n'.join(output)
|
|
|
|
####################
|
|
# FILE UPLOADS #
|
|
####################
|
|
|
|
class FileUploadField(FormField):
|
|
def __init__(self, field_name, is_required=False, validator_list=[]):
|
|
self.field_name, self.is_required = field_name, is_required
|
|
self.validator_list = [self.isNonEmptyFile] + validator_list
|
|
|
|
def isNonEmptyFile(self, field_data, all_data):
|
|
if not field_data['content']:
|
|
raise validators.CriticalValidationError, "The submitted file is empty."
|
|
|
|
def render(self, data):
|
|
return '<input type="file" id="%s" class="v%s" name="%s" />' % \
|
|
(FORM_FIELD_ID_PREFIX + self.field_name, self.__class__.__name__,
|
|
self.field_name)
|
|
|
|
def html2python(data):
|
|
if data is None:
|
|
raise EmptyValue
|
|
return data
|
|
html2python = staticmethod(html2python)
|
|
|
|
class ImageUploadField(FileUploadField):
|
|
"A FileUploadField that raises CriticalValidationError if the uploaded file isn't an image."
|
|
def __init__(self, *args, **kwargs):
|
|
FileUploadField.__init__(self, *args, **kwargs)
|
|
self.validator_list.insert(0, self.isValidImage)
|
|
|
|
def isValidImage(self, field_data, all_data):
|
|
try:
|
|
validators.isValidImage(field_data, all_data)
|
|
except validators.ValidationError, e:
|
|
raise validators.CriticalValidationError, e.messages
|
|
|
|
####################
|
|
# INTEGERS/FLOATS #
|
|
####################
|
|
|
|
class IntegerField(TextField):
|
|
def __init__(self, field_name, length=10, maxlength=None, is_required=False, validator_list=[]):
|
|
validator_list = [self.isInteger] + validator_list
|
|
TextField.__init__(self, field_name, length, maxlength, is_required, validator_list)
|
|
|
|
def isInteger(self, field_data, all_data):
|
|
try:
|
|
validators.isInteger(field_data, all_data)
|
|
except validators.ValidationError, e:
|
|
raise validators.CriticalValidationError, e.messages
|
|
|
|
def html2python(data):
|
|
if data == '' or data is None:
|
|
return None
|
|
return int(data)
|
|
html2python = staticmethod(html2python)
|
|
|
|
class SmallIntegerField(IntegerField):
|
|
def __init__(self, field_name, length=5, maxlength=5, is_required=False, validator_list=[]):
|
|
validator_list = [self.isSmallInteger] + validator_list
|
|
IntegerField.__init__(self, field_name, length, maxlength, is_required, validator_list)
|
|
|
|
def isSmallInteger(self, field_data, all_data):
|
|
if not -32768 <= int(field_data) <= 32767:
|
|
raise validators.CriticalValidationError, "Enter a whole number between -32,768 and 32,767."
|
|
|
|
class PositiveIntegerField(IntegerField):
|
|
def __init__(self, field_name, length=10, maxlength=None, is_required=False, validator_list=[]):
|
|
validator_list = [self.isPositive] + validator_list
|
|
IntegerField.__init__(self, field_name, length, maxlength, is_required, validator_list)
|
|
|
|
def isPositive(self, field_data, all_data):
|
|
if int(field_data) < 0:
|
|
raise validators.CriticalValidationError, "Enter a positive number."
|
|
|
|
class PositiveSmallIntegerField(IntegerField):
|
|
def __init__(self, field_name, length=5, maxlength=None, is_required=False, validator_list=[]):
|
|
validator_list = [self.isPositiveSmall] + validator_list
|
|
IntegerField.__init__(self, field_name, length, maxlength, is_required, validator_list)
|
|
|
|
def isPositiveSmall(self, field_data, all_data):
|
|
if not 0 <= int(field_data) <= 32767:
|
|
raise validators.CriticalValidationError, "Enter a whole number between 0 and 32,767."
|
|
|
|
class FloatField(TextField):
|
|
def __init__(self, field_name, max_digits, decimal_places, is_required=False, validator_list=[]):
|
|
self.max_digits, self.decimal_places = max_digits, decimal_places
|
|
validator_list = [self.isValidFloat] + validator_list
|
|
TextField.__init__(self, field_name, max_digits+1, max_digits+1, is_required, validator_list)
|
|
|
|
def isValidFloat(self, field_data, all_data):
|
|
v = validators.IsValidFloat(self.max_digits, self.decimal_places)
|
|
try:
|
|
v(field_data, all_data)
|
|
except validators.ValidationError, e:
|
|
raise validators.CriticalValidationError, e.messages
|
|
|
|
def html2python(data):
|
|
if data == '' or data is None:
|
|
return None
|
|
return float(data)
|
|
html2python = staticmethod(html2python)
|
|
|
|
####################
|
|
# DATES AND TIMES #
|
|
####################
|
|
|
|
class DatetimeField(TextField):
|
|
"""A FormField that automatically converts its data to a datetime.datetime object.
|
|
The data should be in the format YYYY-MM-DD HH:MM:SS."""
|
|
def __init__(self, field_name, length=30, maxlength=None, is_required=False, validator_list=[]):
|
|
self.field_name = field_name
|
|
self.length, self.maxlength = length, maxlength
|
|
self.is_required = is_required
|
|
self.validator_list = [validators.isValidANSIDatetime] + validator_list
|
|
|
|
def html2python(data):
|
|
"Converts the field into a datetime.datetime object"
|
|
import datetime
|
|
date, time = data.split()
|
|
y, m, d = date.split('-')
|
|
timebits = time.split(':')
|
|
h, mn = timebits[:2]
|
|
if len(timebits) > 2:
|
|
s = int(timebits[2])
|
|
else:
|
|
s = 0
|
|
return datetime.datetime(int(y), int(m), int(d), int(h), int(mn), s)
|
|
html2python = staticmethod(html2python)
|
|
|
|
class DateField(TextField):
|
|
"""A FormField that automatically converts its data to a datetime.date object.
|
|
The data should be in the format YYYY-MM-DD."""
|
|
def __init__(self, field_name, is_required=False, validator_list=[]):
|
|
validator_list = [self.isValidDate] + validator_list
|
|
TextField.__init__(self, field_name, length=10, maxlength=10,
|
|
is_required=is_required, validator_list=validator_list)
|
|
|
|
def isValidDate(self, field_data, all_data):
|
|
try:
|
|
validators.isValidANSIDate(field_data, all_data)
|
|
except validators.ValidationError, e:
|
|
raise validators.CriticalValidationError, e.messages
|
|
|
|
def html2python(data):
|
|
"Converts the field into a datetime.date object"
|
|
import time, datetime
|
|
try:
|
|
time_tuple = time.strptime(data, '%Y-%m-%d')
|
|
return datetime.date(*time_tuple[0:3])
|
|
except (ValueError, TypeError):
|
|
return None
|
|
html2python = staticmethod(html2python)
|
|
|
|
class TimeField(TextField):
|
|
"""A FormField that automatically converts its data to a datetime.time object.
|
|
The data should be in the format HH:MM:SS."""
|
|
def __init__(self, field_name, is_required=False, validator_list=[]):
|
|
validator_list = [self.isValidTime] + validator_list
|
|
TextField.__init__(self, field_name, length=8, maxlength=8,
|
|
is_required=is_required, validator_list=validator_list)
|
|
|
|
def isValidTime(self, field_data, all_data):
|
|
try:
|
|
validators.isValidANSITime(field_data, all_data)
|
|
except validators.ValidationError, e:
|
|
raise validators.CriticalValidationError, e.messages
|
|
|
|
def html2python(data):
|
|
"Converts the field into a datetime.time object"
|
|
import time, datetime
|
|
try:
|
|
try:
|
|
time_tuple = time.strptime(data, '%H:%M:%S')
|
|
except ValueError: # seconds weren't provided
|
|
time_tuple = time.strptime(data, '%H:%M')
|
|
return datetime.time(*time_tuple[3:6])
|
|
except ValueError:
|
|
return None
|
|
html2python = staticmethod(html2python)
|
|
|
|
####################
|
|
# INTERNET-RELATED #
|
|
####################
|
|
|
|
class EmailField(TextField):
|
|
"A convenience FormField for validating e-mail addresses"
|
|
def __init__(self, field_name, length=50, is_required=False, validator_list=[]):
|
|
validator_list = [self.isValidEmail] + validator_list
|
|
TextField.__init__(self, field_name, length, maxlength=75,
|
|
is_required=is_required, validator_list=validator_list)
|
|
|
|
def isValidEmail(self, field_data, all_data):
|
|
try:
|
|
validators.isValidEmail(field_data, all_data)
|
|
except validators.ValidationError, e:
|
|
raise validators.CriticalValidationError, e.messages
|
|
|
|
class URLField(TextField):
|
|
"A convenience FormField for validating URLs"
|
|
def __init__(self, field_name, length=50, is_required=False, validator_list=[]):
|
|
validator_list = [self.isValidURL] + validator_list
|
|
TextField.__init__(self, field_name, length=length, maxlength=200,
|
|
is_required=is_required, validator_list=validator_list)
|
|
|
|
def isValidURL(self, field_data, all_data):
|
|
try:
|
|
validators.isValidURL(field_data, all_data)
|
|
except validators.ValidationError, e:
|
|
raise validators.CriticalValidationError, e.messages
|
|
|
|
class IPAddressField(TextField):
|
|
def html2python(data):
|
|
return data or None
|
|
html2python = staticmethod(html2python)
|
|
|
|
####################
|
|
# MISCELLANEOUS #
|
|
####################
|
|
|
|
class PhoneNumberField(TextField):
|
|
"A convenience FormField for validating phone numbers (e.g. '630-555-1234')"
|
|
def __init__(self, field_name, is_required=False, validator_list=[]):
|
|
validator_list = [self.isValidPhone] + validator_list
|
|
TextField.__init__(self, field_name, length=12, maxlength=12,
|
|
is_required=is_required, validator_list=validator_list)
|
|
|
|
def isValidPhone(self, field_data, all_data):
|
|
try:
|
|
validators.isValidPhone(field_data, all_data)
|
|
except validators.ValidationError, e:
|
|
raise validators.CriticalValidationError, e.messages
|
|
|
|
class USStateField(TextField):
|
|
"A convenience FormField for validating U.S. states (e.g. 'IL')"
|
|
def __init__(self, field_name, is_required=False, validator_list=[]):
|
|
validator_list = [self.isValidUSState] + validator_list
|
|
TextField.__init__(self, field_name, length=2, maxlength=2,
|
|
is_required=is_required, validator_list=validator_list)
|
|
|
|
def isValidUSState(self, field_data, all_data):
|
|
try:
|
|
validators.isValidUSState(field_data, all_data)
|
|
except validators.ValidationError, e:
|
|
raise validators.CriticalValidationError, e.messages
|
|
|
|
def html2python(data):
|
|
return data.upper() # Should always be stored in upper case
|
|
html2python = staticmethod(html2python)
|
|
|
|
class CommaSeparatedIntegerField(TextField):
|
|
"A convenience FormField for validating comma-separated integer fields"
|
|
def __init__(self, field_name, maxlength=None, is_required=False, validator_list=[]):
|
|
validator_list = [self.isCommaSeparatedIntegerList] + validator_list
|
|
TextField.__init__(self, field_name, length=20, maxlength=maxlength,
|
|
is_required=is_required, validator_list=validator_list)
|
|
|
|
def isCommaSeparatedIntegerList(self, field_data, all_data):
|
|
try:
|
|
validators.isCommaSeparatedIntegerList(field_data, all_data)
|
|
except validators.ValidationError, e:
|
|
raise validators.CriticalValidationError, e.messages
|
|
|
|
class XMLLargeTextField(LargeTextField):
|
|
"""
|
|
A LargeTextField with an XML validator. The schema_path argument is the
|
|
full path to a Relax NG compact schema to validate against.
|
|
"""
|
|
def __init__(self, field_name, schema_path, **kwargs):
|
|
self.schema_path = schema_path
|
|
kwargs.setdefault('validator_list', []).insert(0, self.isValidXML)
|
|
LargeTextField.__init__(self, field_name, **kwargs)
|
|
|
|
def isValidXML(self, field_data, all_data):
|
|
v = validators.RelaxNGCompact(self.schema_path)
|
|
try:
|
|
v(field_data, all_data)
|
|
except validators.ValidationError, e:
|
|
raise validators.CriticalValidationError, e.messages
|