Context Record, Bulk update : link to parcels #5293

1 files changed, 25 insertions, 0 deletions

diff --git a/archaeological_context_records/forms.py b/archaeological_context_records/forms.py
index f4c20d55a..44d89fcf1 100644
--- a/archaeological_context_records/forms.py
+++ b/archaeological_context_records/forms.py
@@ -702,6 +702,7 @@ class QAContextRecordFormMulti(QAForm):
         "qa_related_to": models.ContextRecord,
         "qa_unit": models.Unit,
         "qa_town": models.Town,
+        "qa_parcel": models.Parcel,
     }
     REPLACE_FIELDS = [
         "qa_unit",
@@ -739,12 +740,36 @@ class QAContextRecordFormMulti(QAForm):
         ),
         required=False
     )
+    qa_parcel = forms.ChoiceField(
+        label=_("Associated parcel"), choices=[], required=False
+    )
+    #  qa_parcel.choices += [
+        #  (p.pk, str(p)) for p in Parcel.objects.all()
+    #  ]
 
     TYPES = [
         FieldType("qa_relation_type", models.RelationType),
         FieldType("qa_unit", models.Unit),
     ]
 
+    def __init__(self, *args, **kwargs):
+        super(QAContextRecordFormMulti, self).__init__(*args, **kwargs)
+        self.items = kwargs.pop("items")
+        op_pk = 0
+        disable = False
+        for cr in self.items:
+            if op_pk != cr.operation.pk and op_pk > 0:
+                disable = self.fields["qa_parcel"].disabled = True
+                break
+            op_pk = cr.operation.pk
+        if not disable:
+            self.fields["qa_parcel"].choices += [
+                (p.pk, str(p)) for p in Parcel.objects.filter(
+                    operation=self.items[-1].operation.pk
+                ).all()
+            ]
+
+
     def clean(self):
         cleaned_data = super().clean()
         r_type = cleaned_data.get("qa_relation_type")