From 0c185f1abbe9abe0d977e1b7d1d3f0440b8d6371 Mon Sep 17 00:00:00 2001 From: Étienne Loks Date: Tue, 10 Sep 2019 18:02:50 +0200 Subject: Locks: prevent edit actions --- archaeological_context_records/forms.py | 10 +++++++++- archaeological_context_records/views.py | 7 +++++++ 2 files changed, 16 insertions(+), 1 deletion(-) (limited to 'archaeological_context_records') diff --git a/archaeological_context_records/forms.py b/archaeological_context_records/forms.py index 6bbc3099c..375943c9c 100644 --- a/archaeological_context_records/forms.py +++ b/archaeological_context_records/forms.py @@ -133,7 +133,15 @@ class RecordFormSelection(CustomFormSearch): if 'pk' not in cleaned_data or not cleaned_data['pk']: raise forms.ValidationError(_(u"You should at least select one " u"context record.")) - return cleaned_data + pk = self.cleaned_data["pk"] + try: + cr = models.ContextRecord.objects.get(pk=pk) + except models.ContextRecord.DoesNotExist: + raise forms.ValidationError(_("Invalid selection.")) + if cr.locked: + raise forms.ValidationError(_("This context record is locked for " + "edition.")) + return self.cleaned_data class RecordFormGeneral(CustomForm, ManageOldType): diff --git a/archaeological_context_records/views.py b/archaeological_context_records/views.py index fe77150f6..794cb5592 100644 --- a/archaeological_context_records/views.py +++ b/archaeological_context_records/views.py @@ -144,6 +144,13 @@ def record_modify(request, pk): ) return HttpResponseRedirect("/") + q = models.ContextRecord.objects.filter(pk=pk) + if not q.count(): + raise Http404() + cr = q.all()[0] + if cr.locked: + raise Http404() + wizards.RecordModifWizard.session_set_value( request, 'selec-record_modification', 'pk', pk, reset=True) return redirect(reverse('record_modification', -- cgit v1.2.3