修改树状结构

src/views/ComprehensiveAnalysis.vue

@@ -98,14 +98,11 @@
               node-key="id"
               ref="tree"
               :filter-node-method="filterNode"
+              @node-click="handleNodeClick"
             >
               <span class="custom-tree-node" slot-scope="{ node }">
-                <template
-                  v-if="
-                    node.level < 4 &&
-                    node.parent.label !== '预设模型' &&
-                    node.parent.label !== '我的模型'
-                  "
+                <!-- <template
+                  v-if="node.level < 4 && node.parent.label !== '我的模型'"
                 >
                   <div
                     :class="{
@@ -114,12 +111,12 @@
                     }"
                   ></div>
                   <span>{{ node.label }}</span>
-                </template>
+                </template> -->
                 <template
-                  v-else-if="
-                    node.level === 4 ||
-                    node.parent.label === '预设模型' ||
-                    node.parent.label === '我的模型'
+                  v-if="
+                    node.data.mainType === '所有图层' ||
+                    node.data.mainType === '我的模型' ||
+                    node.data.mainType === '预设模型'
                   "
                 >
                   <el-checkbox v-model="node.checked" style="position: relative"
@@ -479,6 +476,9 @@ export default {
     this.updateCasePopupShow = false;
   },
   methods: {
+    handleNodeClick(node) {
+      console.log(node);
+    },
     // 数据字典查询 -- 根据内容获取对应的index
     // this.classTextToIndex["任务类型"]
     classDictQuery(type, cName, keyName) {
@@ -537,7 +537,8 @@ export default {
                                     id: `${item3.title}_${item3.id}`,
                                     label: item3.title,
                                     tag: item3.tag,
-                                    children:item3.columnList&&
+                                    children:
+                                      item3.columnList &&
                                       item3.columnList.length === 0
                                         ? []
                                         : item3.columnList.map((item4) => {
@@ -591,15 +592,29 @@ export default {
                           `${item2.title}_${item2.id}`,
                           `${item2.title}_${item2.id}`
                         );
-                        this.randomColor.set(
-                          `${item2.title}_${item2.id}`,
-                          publicFun.getRandomColor()
-                        );
+
                         return {
                           id: `${item2.title}_${item2.id}`,
                           label: item2.title,
-                          children: [],
-                          mainType: item1.title,
+                          children:
+                            item2.columnList.length === 0
+                              ? []
+                              : item2.columnList.map((item3) => {
+                                  treeIdMap.set(
+                                    `${item3.title}_${item3.id}`,
+                                    `${item3.title}_${item3.id}`
+                                  );
+                                  this.randomColor.set(
+                                    `${item3.title}_${item3.id}`,
+                                    publicFun.getRandomColor()
+                                  );
+                                  return {
+                                    id: `${item3.title}_${item3.id}`,
+                                    label: item3.title,
+                                    children: [],
+                                    mainType: item1.title,
+                                  };
+                                }),
                         };
                       }),
               };