diff options
Diffstat (limited to 'archaeological_finds/forms.py')
| -rw-r--r-- | archaeological_finds/forms.py | 17 |
1 files changed, 17 insertions, 0 deletions
diff --git a/archaeological_finds/forms.py b/archaeological_finds/forms.py index f4b115492..4de4bdaf9 100644 --- a/archaeological_finds/forms.py +++ b/archaeological_finds/forms.py @@ -1239,6 +1239,7 @@ class FindFormSelection(CustomFormSearch): form_label = _("Find search") associated_models = {'pk': models.Find} currents = {'pk': models.Find} + pk_key = 'pk' pk = forms.IntegerField( label="", required=False, @@ -1249,6 +1250,22 @@ class FindFormSelection(CustomFormSearch): source_full=reverse_lazy('get-find-full')), validators=[valid_id(models.Find)]) + def clean(self): + pk = self.cleaned_data[self.pk_key] + if "," in pk: + pks = [k.strip() for k in pk.split(',')] + else: + pks = [pk] + for pk in pks: + try: + find = models.Find.objects.get(pk=pk) + except models.Find.DoesNotExist: + raise forms.ValidationError(_("Invalid selection.")) + if find.locked: + raise forms.ValidationError( + _("This find is locked for edition.")) + return self.cleaned_data + class FindFormSelectionWarehouseModule(FindFormSelection): pk = forms.IntegerField( |