122 lines
5.1 KiB
Python
122 lines
5.1 KiB
Python
from __future__ import absolute_import
|
|
|
|
from django import forms
|
|
from django.core.exceptions import ValidationError
|
|
from django.forms.formsets import formset_factory
|
|
from django.utils.translation import ugettext_lazy as _
|
|
|
|
from .models import MetadataType
|
|
from .settings import AVAILABLE_FUNCTIONS, AVAILABLE_MODELS, AVAILABLE_VALIDATORS
|
|
|
|
|
|
class MetadataForm(forms.Form):
|
|
|
|
def clean_value(self):
|
|
metadata_type = MetadataType.objects.get(pk=self.cleaned_data['id'])
|
|
|
|
try:
|
|
validation_function = AVAILABLE_VALIDATORS[metadata_type.validation]
|
|
except KeyError:
|
|
# User entered a validation function name, but was not found
|
|
# Return value entered as is
|
|
return self.cleaned_data['value']
|
|
else:
|
|
try:
|
|
# If it is a parsing function we should get a value
|
|
# If it is a validation function we get nothing on success
|
|
result = validation_function(self.cleaned_data['value'])
|
|
except Exception as exception:
|
|
# If it is a validation function and an exception is raise
|
|
# we wrap that into a new ValidationError exception
|
|
# If the function exception is a ValidationError itself the
|
|
# error messages will be in a 'messages' property, so we
|
|
# contatenate them.
|
|
# Otherwise we extract whatever single message the exception
|
|
# included.
|
|
try:
|
|
message = u', '.join(exception.messages)
|
|
except AttributeError:
|
|
message = unicode(exception)
|
|
|
|
raise ValidationError(_('Invalid value: %(message)s'), params={'message': message}, code='invalid')
|
|
else:
|
|
# Return the result if it was a parsing function
|
|
# If it was a validation function and passed correctly we return
|
|
# the original input value
|
|
return result or self.cleaned_data['value']
|
|
|
|
# If a validation function was never specified we return the original
|
|
# value
|
|
return self.cleaned_data['value']
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super(MetadataForm, self).__init__(*args, **kwargs)
|
|
|
|
# Set form fields initial values
|
|
if 'initial' in kwargs:
|
|
self.metadata_type = kwargs['initial'].pop('metadata_type', None)
|
|
required = kwargs['initial'].pop('required', False)
|
|
required_string = u''
|
|
|
|
if required:
|
|
self.fields['value'].required = True
|
|
required_string = u' (%s)' % _('Required')
|
|
else:
|
|
self.fields['value'].required = False
|
|
|
|
self.fields['name'].initial = '%s%s' % ((self.metadata_type.title if self.metadata_type.title else self.metadata_type.name), required_string)
|
|
self.fields['id'].initial = self.metadata_type.pk
|
|
|
|
if self.metadata_type.lookup:
|
|
try:
|
|
choices = eval(self.metadata_type.lookup, AVAILABLE_MODELS)
|
|
self.fields['value'] = forms.ChoiceField(label=self.fields['value'].label)
|
|
choices = zip(choices, choices)
|
|
if not required:
|
|
choices.insert(0, ('', '------'))
|
|
self.fields['value'].choices = choices
|
|
self.fields['value'].required = required
|
|
except Exception as exception:
|
|
self.fields['value'].initial = exception
|
|
self.fields['value'].widget = forms.TextInput(attrs={'readonly': 'readonly'})
|
|
|
|
if self.metadata_type.default:
|
|
try:
|
|
self.fields['value'].initial = eval(self.metadata_type.default, AVAILABLE_FUNCTIONS)
|
|
except Exception as exception:
|
|
self.fields['value'].initial = exception
|
|
|
|
id = forms.CharField(label=_(u'ID'), widget=forms.HiddenInput)
|
|
|
|
name = forms.CharField(label=_(u'Name'), required=False, widget=forms.TextInput(attrs={'readonly': 'readonly'}))
|
|
value = forms.CharField(label=_(u'Value'), required=False)
|
|
update = forms.BooleanField(initial=True, label=_(u'Update'), required=False)
|
|
|
|
MetadataFormSet = formset_factory(MetadataForm, extra=0)
|
|
|
|
|
|
class AddMetadataForm(forms.Form):
|
|
metadata_type = forms.ModelChoiceField(queryset=MetadataType.objects.all(), label=_(u'Metadata type'))
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
document_type = kwargs.pop('document_type')
|
|
super(AddMetadataForm, self).__init__(*args, **kwargs)
|
|
self.fields['metadata_type'].queryset = document_type.metadata.all()
|
|
|
|
|
|
class MetadataRemoveForm(MetadataForm):
|
|
update = forms.BooleanField(initial=False, label=_(u'Remove'), required=False)
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super(MetadataRemoveForm, self).__init__(*args, **kwargs)
|
|
self.fields.pop('value')
|
|
|
|
|
|
MetadataRemoveFormSet = formset_factory(MetadataRemoveForm, extra=0)
|
|
|
|
|
|
class MetadataTypeForm(forms.ModelForm):
|
|
class Meta:
|
|
fields = ('name', 'title', 'default', 'lookup', 'validation')
|
|
model = MetadataType
|