Consolidate migrations

archesproject · Nov 14, 2024 · 13d1c49 · 13d1c49
1 parent b674f6b
commit 13d1c49
Show file tree

Hide file tree

Showing 3 changed files with 27 additions and 53 deletions.
diff --git a/arches/app/models/migrations/11613_add_node_constraints.py b/arches/app/models/migrations/11613_add_node_constraints.py
diff --git a/arches/app/models/migrations/11613_node_grouping_node.py b/arches/app/models/migrations/11613_node_grouping_node.py
@@ -47,4 +47,31 @@ def remove_grouping_node(apps, schema_editor):
             ),
         ),
         migrations.RunPython(set_grouping_node, remove_grouping_node),
+        migrations.AddConstraint(
+            model_name="node",
+            constraint=models.CheckConstraint(
+                condition=models.Q(
+                    ("istopnode", True), ("nodegroup__isnull", False), _connector="OR"
+                ),
+                name="has_nodegroup_or_istopnode",
+            ),
+        ),
+        migrations.AddConstraint(
+            model_name="node",
+            constraint=models.CheckConstraint(
+                condition=models.Q(
+                    ("istopnode", True),
+                    ("grouping_node__isnull", False),
+                    _connector="OR",
+                ),
+                name="has_grouping_node_or_istopnode",
+            ),
+        ),
+        migrations.AddConstraint(
+            model_name="node",
+            constraint=models.CheckConstraint(
+                condition=models.Q(("grouping_node_id", models.F("nodegroup_id"))),
+                name="grouping_node_matches_nodegroup",
+            ),
+        ),
     ]
diff --git a/arches/app/models/migrations/11613_node_nodegroup_root_matches_nodegroup.py b/arches/app/models/migrations/11613_node_nodegroup_root_matches_nodegroup.py