Add a tool to create root node in trees

Fixes #5391

specifyweb/frontend/js_src/lib/components/TreeView/Tree.tsx

@@ -5,6 +5,7 @@ import { useBooleanState } from '../../hooks/useBooleanState';
 import { useCachedState } from '../../hooks/useCachedState';
 import { useId } from '../../hooks/useId';
 import { treeText } from '../../localization/tree';
+import { ping } from '../../utils/ajax/ping';
 import type { GetSet, RA } from '../../utils/types';
 import { toggleItem } from '../../utils/utils';
 import { Button } from '../Atoms/Button';
@@ -14,6 +15,7 @@ import type {
   FilterTablesByEndsWith,
   SerializedResource,
 } from '../DataModel/helperTypes';
+import { idFromUrl } from '../DataModel/resource';
 import { deserializeResource } from '../DataModel/serializers';
 import { ResourceView } from '../Forms/ResourceView';
 import { getPref } from '../InitialContext/remotePrefs';
@@ -106,6 +108,22 @@ export function Tree<
     [baseUrl, statsThreshold]
   );
 
+  const treeDefinition = treeDefinitionItems[0].treeDef;
+  const treeDefId = idFromUrl(treeDefinition);
+  const createRootNode = async (): Promise<void> => {
+    if (treeDefId === undefined) {
+      console.error('treeDefId is undefined');
+    } else {
+      await ping(
+        `/api/specify_tree/${tableName.toLowerCase()}/${treeDefId}/add_root/`,
+        {
+          method: 'POST',
+          headers: { Accept: 'application/json' },
+        }
+      );
+    }
+  };
+
   return (
     <div
       className={`
@@ -196,6 +214,13 @@ export function Tree<
           })}
         </div>
       </div>
+      {rows.length === 0 ? (
+        <Button.Icon
+          icon="plus"
+          title={treeText.addRootNode()}
+          onClick={createRootNode}
+        />
+      ) : undefined}
       <ul role="tree rowgroup">
         {rows.map((row, index) => (
           <TreeRow

specifyweb/frontend/js_src/lib/localization/tree.ts

@@ -540,6 +540,9 @@ export const treeText = createDictionary({
   addTree: {
     'en-us': 'Add Tree',
   },
+  addRootNode: {
+    'en-us': 'Add root node',
+  },
   treePicker: {
     'en-us': 'Tree Picker',
   },

specifyweb/specify/tree_views.py

@@ -1,4 +1,5 @@
 from functools import wraps
+import http
 from typing import Literal, Tuple
 from django.db import transaction
 from django.http import HttpResponse
@@ -395,6 +396,35 @@ def repair_tree(request, tree: TREE_TABLE):
     tree_extras.renumber_tree(table)
     tree_extras.validate_tree_numbering(table)
 
+@tree_mutation
+def add_root(request, tree, treeid): 
+    "Creates a root node in a specific tree."
+
+    tree_name = tree.title()
+    tree_target = get_object_or_404(f"{tree_name}treedef", id=treeid)
+    tree_def_item = getattr(models, f"{tree_name}treedefitem")
+    item = getattr(models, tree_name)
+
+    filter_kwargs = {
+        'rankid': 0,
+        'definition_id': treeid
+        # f"{tree}treedefid": treeid
+    }
+
+    if item.objects.filter(**filter_kwargs).count() > 0:
+        raise Exception("Root node already exists.") 
+
+    root = item.objects.create(
+        name="Root",
+        isaccepted=1,
+        nodenumber=1,
+        rankid=0,
+        parent=None,
+        definition=tree_target,
+        definitionitem=tree_def_item
+    )
+
+    return http.HttpResponse('', status=204)
 
 @login_maybe_required
 @require_GET

specifyweb/specify/urls.py

@@ -40,6 +40,7 @@
         url(r'^(?P<rankid>\d+)/tree_rank_item_count/$', tree_views.tree_rank_item_count),
         url(r'^(?P<parentid>\d+)/predict_fullname/$', tree_views.predict_fullname),
         url(r'^(?P<treedef>\d+)/(?P<parentid>\w+)/stats/$', tree_views.tree_stats),
+        url(r'^(?P<treeid>\w+)/add_root/$', tree_views.add_root),
         url(r'^(?P<treedef>\d+)/(?P<parentid>\w+)/(?P<sortfield>\w+)/$', tree_views.tree_view),
         url(r'^repair/$', tree_views.repair_tree),
     ])),