🐛 operation/context record forms: filter site fields (refs #5873, #5874)

1 files changed, 5 insertions, 1 deletions

diff --git a/archaeological_operations/forms.py b/archaeological_operations/forms.py
index 6b0959a91..af3c4bb3d 100644
--- a/archaeological_operations/forms.py
+++ b/archaeological_operations/forms.py
@@ -872,7 +872,7 @@ class OperationFormGeneral(CustomForm, ManageOldType):
         towns = None
         if kwargs.get("data", None) and kwargs['data'].get("TOWNS", None):
             towns = kwargs['data']['TOWNS']
-        super(OperationFormGeneral, self).__init__(*args, **kwargs)
+        super().__init__(*args, **kwargs)
         profile = get_current_profile()
         if not profile.files:
             for key in self.FILE_FIELDS:
@@ -882,6 +882,10 @@ class OperationFormGeneral(CustomForm, ManageOldType):
                 self.remove_field(key)
         if not profile.underwater:
             self._remove_fields(("drassm_code",))
+        data = kwargs.get("data", {}) or kwargs.get("initial", {})
+        has_site = any(1 for k, v in data.items() if v and k.split("-")[-1] == "archaeological_site")
+        if not has_site and not profile.archaeological_site:
+            self._remove_fields(("archaeological_site",))
         if 'collaborator' in self.fields:
             self.fields['collaborator'].widget.attrs['full-width'] = True
         if towns and towns != -1: