diff options
Diffstat (limited to 'archaeological_operations/forms.py')
| -rw-r--r-- | archaeological_operations/forms.py | 88 |
1 files changed, 81 insertions, 7 deletions
diff --git a/archaeological_operations/forms.py b/archaeological_operations/forms.py index 1fc081ea4..2071dd00f 100644 --- a/archaeological_operations/forms.py +++ b/archaeological_operations/forms.py @@ -27,16 +27,19 @@ from django.conf import settings from django.core import validators from django.core.exceptions import ObjectDoesNotExist from django.db.models import Max -from django.forms.formsets import DELETION_FIELD_NAME +from django.forms.formsets import DELETION_FIELD_NAME, TOTAL_FORM_COUNT from django.shortcuts import render_to_response from django.template import RequestContext from django.utils.translation import ugettext_lazy as _ +from django.utils.safestring import mark_safe from ishtar_common.models import valid_id, PersonType, Person, Town from archaeological_files.models import File import models -from widgets import ParcelWidget + +from widgets import ParcelWidget, SelectParcelWidget from ishtar_common import widgets + from ishtar_common.forms import BaseFormSet, FinalForm, FormSet, \ ClosingDateFormSelection, formset_factory, get_now, reverse_lazy, \ get_form_selection, TableSelect @@ -44,6 +47,8 @@ from ishtar_common.forms_common import TownForm, TownFormSet, TownFormset, \ AuthorFormset, SourceForm, SourceSelect, \ SourceDeletionForm, get_town_field +from archaeological_operations.utils import parse_parcels + class ParcelField(forms.MultiValueField): def __init__(self, *args, **kwargs): if 'widget' not in kwargs: @@ -96,15 +101,84 @@ class ParcelForm(forms.Form): u"fields are required.")) return self.cleaned_data +class ParcelSelectionForm(forms.Form): + _parcel_selection = forms.CharField(label=_(u"Full text input"), + widget=SelectParcelWidget(attrs={'class':'parcel-select'}), + max_length=100, help_text=_(u"example: \"2013: XD:1 to "\ + u"13,24,33 to 39, YD:24\" or \"AB:24,AC:42\""), + required=False) + class ParcelFormSet(FormSet): + SELECTION_FORM = ParcelSelectionForm + + def __init__(self, *args, **kwargs): + super(FormSet, self).__init__(*args, **kwargs) + if self.forms[0].__class__.__name__ == 'ParcelForm': + self.selection_form = ParcelSelectionForm() + + def as_table(self): + # add dynamic widget + render = self.selection_form.as_table() + render += super(FormSet, self).as_table() + return mark_safe(render) + + def as_p(self): + # add dynamic widget + render = self.selection_form.as_p() + render += super(FormSet, self).as_p() + return mark_safe(render) + + def as_ul(self): + # add dynamic widget + render = self.selection_form.as_ul() + render += super(FormSet, self).as_ul() + return mark_safe(render) + def add_fields(self, form, index): super(FormSet, self).add_fields(form, index) def clean(self): - """Checks that no parcels are duplicated.""" - return self.check_duplicate(('town', 'section', - 'parcel_number', 'year'), - _(u"There are identical parcels.")) + # manage parcel selection + last_town, parcels = None, [] + if hasattr(self, 'cleaned_data') and self.cleaned_data: + for parcel in reversed(self.cleaned_data): + if parcel.get('town'): + last_town = parcel.get('town') + break + if not last_town: + towns = self.forms[0].fields['town'].choices + if towns: + towns.pop(0) # remove first empty + if towns: + last_town = towns[0][0] + if self.data.get('_parcel_selection'): + parcels = parse_parcels(self.data['_parcel_selection']) + if last_town: + for idx, parcel in enumerate(parcels): + parcel['town'] = last_town + parcel['DELETE'] = False + parcels[idx] = parcel + c_max = self.total_form_count() + # pop the last extra form + extra_form = self.forms.pop() + for idx, parcel in enumerate(parcels): + form = self._construct_form(idx + c_max) + for k in parcel: + self.data[form.prefix+'-'+k] = parcel[k] + # reconstruct with correct binded data + form = self._construct_form(idx + c_max) + form.cleaned_data = parcel + self.forms.append(form) + self._errors.append(None) + self.forms.append(extra_form) + self.data[self.prefix+'-'+TOTAL_FORM_COUNT] = c_max + len(parcels) + self.management_form.data = self.data + self.management_form.is_valid() + # Checks that no parcels are duplicated. + self.check_duplicate(('town', 'section', 'parcel_number', + 'year'), _(u"There are identical parcels.")) + if hasattr(self, 'cleaned_data') and self.cleaned_data: + return self.cleaned_data ParcelFormSet = formset_factory(ParcelForm, can_delete=True, formset=ParcelFormSet) @@ -171,7 +245,7 @@ class OperationFormSelection(forms.Form): return cleaned_data class OperationCodeInput(forms.TextInput): - """Manage auto complete whene changing year in form""" + """Manage auto complete when changing year in form""" def render(self, *args, **kwargs): name, value = args base_name = '-'.join(name.split('-')[:-1]) |