From b10f1833c7b8586706a1c9d7be9e8a37cd221ef3 Mon Sep 17 00:00:00 2001 From: Étienne Loks Date: Mon, 21 Dec 2015 15:06:21 +0100 Subject: Manage context record relations. --- archaeological_operations/forms.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) (limited to 'archaeological_operations/forms.py') diff --git a/archaeological_operations/forms.py b/archaeological_operations/forms.py index 1dad60ea1..cba230338 100644 --- a/archaeological_operations/forms.py +++ b/archaeological_operations/forms.py @@ -338,6 +338,8 @@ ParcelFormSet.form_label = _(u"Parcels") class RecordRelationsForm(forms.Form): base_model = 'right_relation' + current_model = models.RelationType + current_related_model = models.Operation associated_models = {'right_record': models.Operation, 'relation_type': models.RelationType} relation_type = forms.ChoiceField(label=_(u"Relation type"), @@ -373,14 +375,14 @@ class RecordRelationsForm(forms.Form): if not data: continue try: - relation_type = models.RelationType.objects.get( + relation_type = cls.current_model.objects.get( pk=data.get('relation_type')) - except models.RelationType.DoesNotExist: + except cls.current_model.DoesNotExist: continue try: - right_record = models.Operation.objects.get( + right_record = cls.current_related_model.objects.get( pk=data.get('right_record')) - except models.Operation.DoesNotExist: + except cls.current_related_model.DoesNotExist: continue values = [unicode(relation_type), right_record.reference] if data.get('DELETE'): -- cgit v1.2.3