diff options
Diffstat (limited to 'archaeological_finds/forms.py')
| -rw-r--r-- | archaeological_finds/forms.py | 149 |
1 files changed, 113 insertions, 36 deletions
diff --git a/archaeological_finds/forms.py b/archaeological_finds/forms.py index 01765b18c..9a52882b2 100644 --- a/archaeological_finds/forms.py +++ b/archaeological_finds/forms.py @@ -270,39 +270,6 @@ class MultipleFindFormSelection(forms.Form): validators=[valid_id(models.Find)]) -class BaseTreatmentForm(forms.Form): - form_label = _(u"Base treatment") - associated_models = {'treatment_type': models.TreatmentType, - 'person': Person, - 'location': Warehouse} - treatment_type = forms.ChoiceField(label=_(u"Treatment type"), choices=[]) - person = forms.IntegerField( - label=_(u"Person"), - widget=widgets.JQueryAutoComplete( - reverse_lazy('autocomplete-person'), associated_model=Person, - new=True), - validators=[valid_id(Person)]) - location = forms.IntegerField( - label=_(u"Location"), - widget=widgets.JQueryAutoComplete( - reverse_lazy('autocomplete-warehouse'), associated_model=Warehouse, - new=True), - validators=[valid_id(Warehouse)]) - description = forms.CharField(label=_(u"Description"), - widget=forms.Textarea, required=False) - start_date = forms.DateField(label=_(u"Start date"), required=False, - widget=widgets.JQueryDate) - end_date = forms.DateField(label=_(u"End date"), required=False, - widget=widgets.JQueryDate) - - def __init__(self, *args, **kwargs): - super(BaseTreatmentForm, self).__init__(*args, **kwargs) - self.fields['treatment_type'].choices = models.TreatmentType.get_types( - exclude=['packaging']) - self.fields['treatment_type'].help_text = \ - models.TreatmentType.get_help(exclude=['packaging']) - - class FindMultipleFormSelection(forms.Form): form_label = _(u"Upstream finds") associated_models = {'finds': models.Find} @@ -321,13 +288,67 @@ class FindMultipleFormSelection(forms.Form): return self.cleaned_data +def check_exist(form_name, key): + def func(self): + request = self.request + storage = self.storage + if storage.prefix not in request.session or \ + 'step_data' not in request.session[storage.prefix] or \ + form_name not in request.session[storage.prefix]['step_data'] or\ + form_name + '-' + key not in \ + request.session[storage.prefix]['step_data'][form_name]: + return False + try: + val = request.session[storage.prefix][ + 'step_data'][form_name][form_name + '-' + key] + if val and len(val) > 0: + val = val[0] + return bool(val) + except ValueError: + return False + return func + + +def check_not_exist(form_name, key): + def func(self): + return not check_exist(form_name, key)(self) + return func + + +def check_type_field(form_name, key, model, field): + def func(self): + request = self.request + storage = self.storage + if storage.prefix not in request.session or \ + 'step_data' not in request.session[storage.prefix] or \ + form_name not in request.session[storage.prefix]['step_data'] or\ + form_name + '-' + key not in \ + request.session[storage.prefix]['step_data'][form_name]: + return False + try: + val = model.objects.get(pk=request.session[storage.prefix][ + 'step_data'][form_name][form_name + '-' + key][0]) + return bool(getattr(val, field)) + except (ValueError, model.DoesNotExist): + return False + return func + + +def check_type_not_field(form_name, key, model, field): + def func(self): + return not check_type_field(form_name, key, model, field)(self) + return func + + def check_treatment(form_name, type_key, type_list=[], not_type_list=[]): type_list = [models.TreatmentType.objects.get(txt_idx=tpe).pk for tpe in type_list] not_type_list = [models.TreatmentType.objects.get(txt_idx=tpe).pk for tpe in not_type_list] - def func(self, request, storage): + def func(self): + request = self.request + storage = self.storage if storage.prefix not in request.session or \ 'step_data' not in request.session[storage.prefix] or \ form_name not in request.session[storage.prefix]['step_data'] or\ @@ -335,8 +356,11 @@ def check_treatment(form_name, type_key, type_list=[], not_type_list=[]): request.session[storage.prefix]['step_data'][form_name]: return False try: - type = int(request.session[storage.prefix]['step_data'] - [form_name][form_name + '-' + type_key]) + tpe = request.session[storage.prefix][ + 'step_data'][form_name][form_name + '-' + type_key] + if not tpe: + return False + type = int(tpe[0]) return (not type_list or type in type_list) \ and type not in not_type_list except ValueError: @@ -375,6 +399,10 @@ class FindDeletionForm(FinalForm): class UpstreamFindFormSelection(FindFormSelection): form_label = _(u"Upstream find") + def __init__(self, *args, **kwargs): + super(UpstreamFindFormSelection, self).__init__(*args, **kwargs) + self.fields['pk'].required = True + ############################################# # Source management for archaelogical finds # ############################################# @@ -482,6 +510,55 @@ class FindBasketAddItemForm(forms.Form): basket.items.add(item) return basket + +class BaseTreatmentForm(SelectFindBasketForm): + form_label = _(u"Base treatment") + associated_models = {'treatment_type': models.TreatmentType, + 'person': Person, + 'location': Warehouse, + 'basket': models.FindBasket} + treatment_type = forms.ChoiceField(label=_(u"Treatment type"), choices=[]) + person = forms.IntegerField( + label=_(u"Person"), + widget=widgets.JQueryAutoComplete( + reverse_lazy('autocomplete-person'), associated_model=Person, + new=True), + validators=[valid_id(Person)]) + location = forms.IntegerField( + label=_(u"Location"), + widget=widgets.JQueryAutoComplete( + reverse_lazy('autocomplete-warehouse'), associated_model=Warehouse, + new=True), + validators=[valid_id(Warehouse)]) + description = forms.CharField(label=_(u"Description"), + widget=forms.Textarea, required=False) + start_date = forms.DateField(label=_(u"Start date"), required=False, + widget=widgets.JQueryDate) + end_date = forms.DateField(label=_(u"End date"), required=False, + widget=widgets.JQueryDate) + + def __init__(self, *args, **kwargs): + super(BaseTreatmentForm, self).__init__(*args, **kwargs) + self.fields['treatment_type'].choices = models.TreatmentType.get_types( + exclude=['packaging']) + self.fields['treatment_type'].help_text = \ + models.TreatmentType.get_help(exclude=['packaging']) + self.fields['basket'].required = False + self.fields['basket'].help_text = \ + _(u"Leave it blank if you want to select a single item") + + def clean(self, *args, **kwargs): + try: + treatment = models.TreatmentType.objects.get( + pk=self.cleaned_data['treatment_type'], available=True) + except models.TreatmentType.DoesNotExist: + raise forms.ValidationError(_(u"This treatment type is not " + u"available.")) + if treatment.upstream_is_many and \ + not self.cleaned_data.get('basket'): + raise forms.ValidationError(_(u"This treatment needs a basket.")) + return self.cleaned_data + """ #################################### # Source management for treatments # |