Locks: prevent edit actions

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):