Context records: fix post delete and post save of records relations

1 files changed, 9 insertions, 8 deletions

diff --git a/archaeological_context_records/models.py b/archaeological_context_records/models.py
index 8cc10172f..59716be8b 100644
--- a/archaeological_context_records/models.py
+++ b/archaeological_context_records/models.py
@@ -284,22 +284,22 @@ class RecordRelations(models.Model):
         verbose_name_plural = _(u"Record relations")
 
     def save(self, *args, **kwargs):
-        obj = super(RelationType, self).save(*args, **kwargs)
+        super(RecordRelations, self).save(*args, **kwargs)
 
         # after saving create the symetric or the inverse relation
 
-        sym_rel_type = obj.relation_type
-        if not obj.relation_type.symmetrical:
-            sym_rel_type = obj.relation_type.inverse_relation
+        sym_rel_type = self.relation_type
+        if not self.relation_type.symmetrical:
+            sym_rel_type = self.relation_type.inverse_relation
 
         # no symetric/inverse is defined
         if not sym_rel_type:
             return
 
         dct = {'right_record':self.left_record, 'left_record':self.right_record,
-               'relation_type':self.relation_type}
+               'relation_type':sym_rel_type}
         RecordRelations.objects.get_or_create(**dct)
-        return obj
+        return self
 
 def post_delete_record_relation(sender, instance, **kwargs):
     # delete symmetrical or inverse relation
@@ -311,8 +311,9 @@ def post_delete_record_relation(sender, instance, **kwargs):
     if not sym_rel_type:
         return
 
-    dct = {'right_record':self.left_record, 'left_record':self.right_record,
-           'relation_type':self.relation_type}
+    dct = {'right_record':instance.left_record,
+           'left_record':instance.right_record,
+           'relation_type':sym_rel_type}
     RecordRelations.objects.filter(**dct).delete()
 
 post_delete.connect(post_delete_record_relation, sender=RecordRelations)