Container trees: INSERT rules

1 files changed, 12 insertions, 2 deletions

diff --git a/archaeological_warehouse/models.py b/archaeological_warehouse/models.py
index 2df5cbe77..5cadc46da 100644
--- a/archaeological_warehouse/models.py
+++ b/archaeological_warehouse/models.py
@@ -460,10 +460,20 @@ class ContainerTree(models.Model):
         DO INSTEAD DELETE FROM archaeological_warehouse_container where id=NULL;
     CREATE RULE containers_tree_update AS
         ON UPDATE TO containers_tree
-        DO INSTEAD UPDATE archaeological_warehouse_container set id=id WHERE id=NULL;
+        DO INSTEAD UPDATE archaeological_warehouse_container set id=id
+            WHERE id=NULL;
     CREATE RULE container_tree_update AS
         ON UPDATE TO container_tree
-        DO INSTEAD UPDATE archaeological_warehouse_container set id=id WHERE id=NULL;
+        DO INSTEAD UPDATE archaeological_warehouse_container set id=id
+            WHERE id=NULL;
+    CREATE RULE containers_tree_insert AS
+        ON INSERT TO containers_tree
+        DO INSTEAD UPDATE archaeological_warehouse_container set id=id
+            WHERE id=NULL;
+    CREATE RULE container_tree_insert AS
+        ON INSERT TO container_tree
+        DO INSTEAD UPDATE archaeological_warehouse_container set id=id
+            WHERE id=NULL;
     """
     DELETE_SQL = """
     DROP VIEW IF EXISTS container_tree;