diff options
Diffstat (limited to 'archaeological_operations/forms.py')
| -rw-r--r-- | archaeological_operations/forms.py | 20 |
1 files changed, 18 insertions, 2 deletions
diff --git a/archaeological_operations/forms.py b/archaeological_operations/forms.py index 5cc544171..1e12c3614 100644 --- a/archaeological_operations/forms.py +++ b/archaeological_operations/forms.py @@ -616,7 +616,15 @@ class OperationFormSelection(CustomFormSearch): cleaned_data = self.cleaned_data if 'pk' not in cleaned_data or not cleaned_data['pk']: raise forms.ValidationError(_(u"You should select an operation.")) - return cleaned_data + pk = self.cleaned_data["pk"] + try: + item = models.Operation.objects.get(pk=pk) + except models.Operation.DoesNotExist: + raise forms.ValidationError(_("Invalid selection.")) + if item.locked: + raise forms.ValidationError(_("This operation is locked for " + "edition.")) + return self.cleaned_data class OperationCodeInput(forms.TextInput): @@ -1388,7 +1396,15 @@ class SiteFormSelection(IshtarForm): cleaned_data = self.cleaned_data if 'pk' not in cleaned_data or not cleaned_data['pk']: raise forms.ValidationError(_(u"You should select an item.")) - return cleaned_data + pk = self.cleaned_data["pk"] + try: + item = models.ArchaeologicalSite.objects.get(pk=pk) + except models.ArchaeologicalSite.DoesNotExist: + raise forms.ValidationError(_("Invalid selection.")) + if item.locked: + raise forms.ValidationError(_("This site is locked for " + "edition.")) + return self.cleaned_data class SiteForm(CustomForm, ManageOldType): |