from __future__ import absolute_import from django import forms from django.utils.translation import ugettext_lazy as _ from django.forms.formsets import formset_factory from common.widgets import ScrollableCheckboxSelectMultiple from .conf.settings import AVAILABLE_MODELS, AVAILABLE_FUNCTIONS from .models import MetadataSet, MetadataType, DocumentTypeDefaults class MetadataForm(forms.Form): 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) #self.document_type = kwargs['initial'].pop('document_type', None) # FIXME: #required = self.document_type.documenttypemetadatatype_set.get(metadata_type=self.metadata_type).required required = False required_string = u'' if required: self.fields['value'].required = True required_string = ' (%s)' % _(u'required') else: #TODO: FIXME: not working correctly 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.default: try: self.fields['value'].initial = eval(self.metadata_type.default, AVAILABLE_FUNCTIONS) except Exception, err: self.fields['value'].initial = err 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, err: self.fields['value'].initial = err self.fields['value'].widget = forms.TextInput(attrs={'readonly': 'readonly'}) 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')) class MetadataRemoveForm(MetadataForm): update = forms.BooleanField(initial=False, label=_(u'Remove'), required=False) class MetadataSelectionForm(forms.Form): def __init__(self, *args, **kwargs): #document_type = kwargs.pop('document_type', None) super(MetadataSelectionForm, self).__init__(*args, **kwargs) document_type = getattr(self, 'initial', {}).get('document_type', None) if document_type: try: defaults = document_type.documenttypedefaults_set.get() self.fields['metadata_sets'].initial = defaults.default_metadata_sets.all() self.fields['metadata_types'].initial = defaults.default_metadata.all() except DocumentTypeDefaults.DoesNotExist: pass metadata_sets = forms.ModelMultipleChoiceField( queryset=MetadataSet.objects.all(), label=_(u'Metadata sets'), required=False, #widget=forms.widgets.SelectMultiple(attrs={'size': 10, 'class': 'choice_form'}) widget=ScrollableCheckboxSelectMultiple(attrs={'size': 10, 'class': 'choice_form'}) ) metadata_types = forms.ModelMultipleChoiceField( queryset=MetadataType.objects.all(), label=_(u'Metadata'), required=False, #widget=forms.widgets.SelectMultiple(attrs={'size': 10, 'class': 'choice_form'}) widget=ScrollableCheckboxSelectMultiple(attrs={'size': 10, 'class': 'choice_form'}) ) MetadataRemoveFormSet = formset_factory(MetadataRemoveForm, extra=0) class MetadataTypeForm(forms.ModelForm): class Meta: model = MetadataType class MetadataSetForm(forms.ModelForm): class Meta: model = MetadataSet