Add manage.py validation check

archesproject · Nov 21, 2024 · 6b866aa · 6b866aa
1 parent 4171f6b
commit 6b866aa
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/arches/app/const.py b/arches/app/const.py
@@ -3,14 +3,16 @@
 
 IntegrityCheckDescriptions = {
     1005: "Nodes with ontologies found in graphs without ontologies",
-    1012: "Node Groups without matching nodes",
+    1012: "Node Groups without contained nodes",
+    1013: "Node Groups without a root node",
 }
 
 
 @unique
 class IntegrityCheck(Enum):
     NODE_HAS_ONTOLOGY_GRAPH_DOES_NOT = 1005
     NODELESS_NODE_GROUP = 1012
+    NODEGROUP_WITHOUT_ROOT_NODE = 1013
 
     def __str__(self):
         return IntegrityCheckDescriptions[self.value]

diff --git a/arches/management/commands/validate.py b/arches/management/commands/validate.py
@@ -116,6 +116,11 @@ def handle(self, *args, **options):
             ),
             fix_action=DELETE_QUERYSET,
         )
+        self.check_integrity(
+            check=IntegrityCheck.NODEGROUP_WITHOUT_ROOT_NODE,  # 1013
+            queryset=models.NodeGroup.objects.filter(node__gt=0, root_node=None),
+            fix_action=None,
+        )
 
     def check_integrity(self, check, queryset, fix_action):
         # 500 not set as a default earlier: None distinguishes whether verbose output implied