diff options
Diffstat (limited to 'archaeological_context_records')
-rw-r--r-- | archaeological_context_records/forms.py | 10 | ||||
-rw-r--r-- | archaeological_context_records/views.py | 7 |
2 files changed, 16 insertions, 1 deletions
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', |