1 files changed, 23 insertions, 0 deletions
diff --git a/archaeological_operations/forms.py b/archaeological_operations/forms.py
index 79e265e93..f32399ecd 100644
--- a/archaeological_operations/forms.py
+++ b/archaeological_operations/forms.py
@@ -432,6 +432,29 @@ RecordRelationsFormSet.form_admin_name = _("Operation - Relations")
 RecordRelationsFormSet.form_slug = "operation-relations"
 
 
+class SiteRecordRelationsForm(RecordRelationsForm):
+    current_model = models.SiteRelationType
+    current_related_model = models.ArchaeologicalSite
+    associated_models = {
+        "right_record": models.ArchaeologicalSite,
+        "relation_type": models.SiteRelationType,
+    }
+    ERROR_MISSING = _("You should select an archaeological site and a relation type.")
+    ERROR_SAME = _("An archaeological site cannot be related to himself.")
+
+    right_record = forms.ChoiceField(
+        label=_("Archaeological site"), choices=[], required=False
+    )
+
+
+SiteRecordRelationsFormSet = formset_factory(
+    SiteRecordRelationsForm, can_delete=True, formset=RecordRelationsFormSetBase
+)
+SiteRecordRelationsFormSet.form_label = _("Archaeological site - Relations")
+RecordRelationsFormSet.form_admin_name = _("Archaeological site - Relations")
+RecordRelationsFormSet.form_slug = "archaeologicalsite-recordrelations"
+
+
 class OpeSiteRelationsForm(ManageOldType):
     associated_models = {'right_record': models.ArchaeologicalSite}