新增自定义节点显示修改

src/views/data-fill/division.vue

@@ -695,6 +695,7 @@ const ElTreeMenuClick = async ({ key, node, data, keys }) => {
 const loadMenu = ({ node, item, level }, resolve)=>{
     ElTreeMenu.value = [
     { icon: 'add-circle', label: '新增节点', key: 'add' },
+    { icon: 'add-circle', label: '新增自定义节点', key: 'add1' },
     { icon: 'file-copy-2', label: '复制节点', key: 'copy' },
     { icon: 'draft', label: '修改节点', key: 'edit' },
     { icon: 'sort-asc', label: '调整排序', key: 'sort' },
@@ -702,17 +703,23 @@ const loadMenu = ({ node, item, level }, resolve)=>{
 ]
     let menusArr = ElTreeMenu.value
     const { isCustomChild, notExsitChild } = item //isCustomChild===1//代表子级是自定义节点
-    const { isCustom } = item
-    if (isCustom === 1 && isCustomChild === 0 && notExsitChild) {//无子级，显示两个按钮
-        menusArr.unshift( { icon: 'add-circle', label: '新增自定义节点', key: 'add1' })
-        resolve(menusArr)
-    } else if (isCustom === 1 && isCustomChild === 1) {//自定义节点类型下如果有自定义节点，就不允许新增节点
-        menusArr.unshift( { icon: 'add-circle', label: '新增自定义节点', key: 'add1' })
-        let menusArr1 = arrDelKey(menusArr, 'label', '新增节点') // [{id:1}]
-        resolve(menusArr1)
-    } else if (isCustom === 1 && isCustomChild === 0 && !notExsitChild) { //自定义节点类型下如果有划分节点，就不允许新增自定义节点
-        resolve(menusArr)
-    } else if (level === 1) {
+    // const { isCustom } = item
+    // if (isCustom === 1 && isCustomChild === 0 && notExsitChild) {//无子级，显示两个按钮
+    //     menusArr.unshift( { icon: 'add-circle', label: '新增自定义节点', key: 'add1' })
+    //     resolve(menusArr)
+    // } else if (isCustom === 1 && isCustomChild === 1) {//自定义节点类型下如果有自定义节点，就不允许新增节点
+    //     menusArr.unshift( { icon: 'add-circle', label: '新增自定义节点', key: 'add1' })
+    //     let menusArr1 = arrDelKey(menusArr, 'label', '新增节点') // [{id:1}]
+    //     resolve(menusArr1)
+    // } else if (isCustom === 1 && isCustomChild === 0 && !notExsitChild) { //自定义节点类型下如果有划分节点，就不允许新增自定义节点
+    //     resolve(menusArr)
+    // } else if (level === 1) {
+    //     menusArr = [ { icon: 'add-circle', label: '新增自定义节点', key: 'add1' }]
+    //     resolve(menusArr)
+    // } else {
+    //     resolve(menusArr)
+    // }
+     if (level === 1) {
         menusArr = [ { icon: 'add-circle', label: '新增自定义节点', key: 'add1' }]
         resolve(menusArr)
     } else {

src/views/data-fill/wbs.vue

@@ -796,7 +796,10 @@ const setElTreeMenu = (contractType) => {
 
         if (HcIsButton('wbs_tree_add')) {
             newArr.push({ icon: 'add-circle', label: '新增节点', key: 'add' })
+            newArr.push( { icon: 'add-circle', label: '新增自定义节点', key: 'add1' })
         }
+        
+       
         if (HcIsButton('wbs_tree_edit')) {
             newArr.push({ icon: 'draft', label: '编辑节点', key: 'edit' })
         }
@@ -823,6 +826,7 @@ const setElTreeMenu = (contractType) => {
 
         if (HcIsButton('wbs_tree_add')) {
             newArr.push({ icon: 'add-circle', label: '新增节点', key: 'add' })
+            newArr.push( { icon: 'add-circle', label: '新增自定义节点', key: 'add1' })
         }
         if (HcIsButton('wbs_tree_copy')) {
             newArr.push({ icon: 'file-copy-2', label: '复制节点', key: 'copy' })
@@ -2091,17 +2095,23 @@ const loadMenu = ({ node, item, level }, resolve)=>{
     setElTreeMenu(contractInfo.value?.contractType)
     let menusArr = ElTreeMenu.value
     const { isCustomChild, notExsitChild } = item //isCustomChild===1//代表子级是自定义节点
-    const { isCustom } = item
-    if (isCustom === 1 && isCustomChild === 0 && notExsitChild) {//无子级，显示两个按钮
-        menusArr.unshift( { icon: 'add-circle', label: '新增自定义节点', key: 'add1' })
-        resolve(menusArr)
-    } else if (isCustom === 1 && isCustomChild === 1) {//自定义节点类型下如果有自定义节点，就不允许新增节点
-        menusArr.unshift( { icon: 'add-circle', label: '新增自定义节点', key: 'add1' })
-        let menusArr1 = arrDelKey(menusArr, 'label', '新增节点') // [{id:1}]
-        resolve(menusArr1)
-    } else if (isCustom === 1 && isCustomChild === 0 && !notExsitChild) { //自定义节点类型下如果有划分节点，就不允许新增自定义节点
-        resolve(menusArr)
-    } else if (level === 1) {
+    // const { isCustom } = item
+    // if (isCustom === 1 && isCustomChild === 0 && notExsitChild) {//无子级，显示两个按钮
+    //     menusArr.unshift( { icon: 'add-circle', label: '新增自定义节点', key: 'add1' })
+    //     resolve(menusArr)
+    // } else if (isCustom === 1 && isCustomChild === 1) {//自定义节点类型下如果有自定义节点，就不允许新增节点
+    //     menusArr.unshift( { icon: 'add-circle', label: '新增自定义节点', key: 'add1' })
+    //     let menusArr1 = arrDelKey(menusArr, 'label', '新增节点') // [{id:1}]
+    //     resolve(menusArr1)
+    // } else if (isCustom === 1 && isCustomChild === 0 && !notExsitChild) { //自定义节点类型下如果有划分节点，就不允许新增自定义节点
+    //     resolve(menusArr)
+    // } else if (level === 1) {
+    //     menusArr = [ { icon: 'add-circle', label: '新增自定义节点', key: 'add1' }]
+    //     resolve(menusArr)
+    // } else {
+    //     resolve(menusArr)
+    // }
+    if (level === 1) {
         menusArr = [ { icon: 'add-circle', label: '新增自定义节点', key: 'add1' }]
         resolve(menusArr)
     } else {