Container: prevent parent infinite loop

1 files changed, 18 insertions, 1 deletions

diff --git a/archaeological_warehouse/models.py b/archaeological_warehouse/models.py
index a72d5a214..add786012 100644
--- a/archaeological_warehouse/models.py
+++ b/archaeological_warehouse/models.py
@@ -1274,6 +1274,22 @@ class Container(
     def _get_base_image_path(self):
         return self.location._get_base_image_path() + "/" + self.external_id
 
+    def _prevent_parent_infinite_loop(self) -> bool:
+        """
+        Check there is no infinite loop in parents. If a loop is detected, set the
+        parent to null.
+        :return: True if parent has been changed
+        """
+        parent = self.parent
+        parents = []
+        while parent:
+            if parent.id in parents:
+                self.parent = None  # break the loop arbitrary
+                return True
+            parents.append(parent.id)
+            parent = parent.parent
+        return False
+
     @classmethod
     def _change_child_location(cls, parent):
         for child in (
@@ -1742,9 +1758,10 @@ class Container(
         self.pre_save()
         super(Container, self).save(*args, **kwargs)
         self._change_child_location(self)
+        updated = False
+        updated += self._prevent_parent_infinite_loop()
         self._update_warehouse_max_division()
 
-        updated = False
         updated += self._calculate_weight()
 
         if not self.index and not self.container_type.stationary: