Locks: prevent edit actions

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(