🗃️ sites - models: add relations between sites - minor relations between items refactoring

1 files changed, 17 insertions, 1 deletions

diff --git a/ishtar_common/models.py b/ishtar_common/models.py
index 04defbd81..a6aab9416 100644
--- a/ishtar_common/models.py
+++ b/ishtar_common/models.py
@@ -874,6 +874,15 @@ class GeneralRelationType(GeneralType):
         return obj
 
 
+class RecordRelationManager(models.Manager):
+    def get_by_natural_key(self, left_record, right_record, relation_type):
+        return self.get(
+            left_record__uuid=left_record,
+            right_record__uuid=right_record,
+            relation_type__txt_idx=relation_type,
+        )
+
+
 class GeneralRecordRelations(Imported):
     class Meta:
         abstract = True
@@ -882,8 +891,15 @@ class GeneralRecordRelations(Imported):
     def general_types(cls):
         return ["relation_type"]
 
+    def natural_key(self):
+        return (
+            self.left_record.uuid,
+            self.right_record.uuid,
+            self.relation_type.txt_idx,
+        )
+
     def save(self, *args, **kwargs):
-        super(GeneralRecordRelations, self).save(*args, **kwargs)
+        super().save(*args, **kwargs)
 
         # after saving create the symmetric or the inverse relation