1 files changed, 27 insertions, 0 deletions
diff --git a/archaeological_operations/forms.py b/archaeological_operations/forms.py
index 79e265e93..bb62ad8d1 100644
--- a/archaeological_operations/forms.py
+++ b/archaeological_operations/forms.py
@@ -432,6 +432,33 @@ 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.IntegerField(
+        label=_("Archaeological site"),
+        widget=widgets.JQueryAutoComplete(
+            reverse_lazy('autocomplete-archaeologicalsite'),
+            associated_model=models.ArchaeologicalSite),
+        validators=[valid_id(models.ArchaeologicalSite)], 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}