From 021533d01d7aea9d31bf5ae28548be6d68965720 Mon Sep 17 00:00:00 2001
From: jiang <jiang@bonus>
Date: Tue, 3 Dec 2024 09:02:06 +0800
Subject: [PATCH] =?UTF-8?q?=E6=8F=90=E4=BA=A4=E4=BB=A3=E7=A0=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/api/dataCenter/annotationTask.js          |   9 +
 src/api/dataCenter/releaseVersion.js          |   8 +
 .../annotationTask/child/allTasks.vue         |   5 +-
 .../dialog/releaseVersionDialog.vue           |  50 ++---
 .../dataSet/child/annotationTaskDialog.vue    |   4 +-
 .../dataSet/child/datasetVersion.vue          |   4 +-
 .../dataCenter/dataSet/child/exportDialog.vue |  99 ++++++++++
 .../dataCenter/dataSet/components/dataSet.vue | 178 ++++++++++--------
 8 files changed, 248 insertions(+), 109 deletions(-)
 create mode 100644 src/views/dataCenter/dataSet/child/exportDialog.vue

diff --git a/src/api/dataCenter/annotationTask.js b/src/api/dataCenter/annotationTask.js
index 71f760d0..8f62244a 100644
--- a/src/api/dataCenter/annotationTask.js
+++ b/src/api/dataCenter/annotationTask.js
@@ -71,5 +71,14 @@ export function manualAnnotate(data) {
   })
 }
 
+export function getLatestDatasetVersion(datasetId) {
+  return request({
+    url: '/ai/annotationTask/getLatestDatasetVersion/'+datasetId,
+    method: 'get'
+  })
+}
+
+
+
 
 
diff --git a/src/api/dataCenter/releaseVersion.js b/src/api/dataCenter/releaseVersion.js
index 6acee245..15a7ddb7 100644
--- a/src/api/dataCenter/releaseVersion.js
+++ b/src/api/dataCenter/releaseVersion.js
@@ -31,4 +31,12 @@ export function annotationsExport(taskId) {
   })
 }
 
+export function listSelect(query) {
+  return request({
+    url: '/ai/version/listSelect',
+    method: 'get',
+    params: query
+  })
+}
+
 
diff --git a/src/views/dataCenter/annotationTask/child/allTasks.vue b/src/views/dataCenter/annotationTask/child/allTasks.vue
index 0b9505ff..2dc8d704 100644
--- a/src/views/dataCenter/annotationTask/child/allTasks.vue
+++ b/src/views/dataCenter/annotationTask/child/allTasks.vue
@@ -70,7 +70,7 @@
       </el-table-column>
       <el-table-column label="待审核个数" show-overflow-tooltip align="center" prop="createBy">
         <template slot-scope="scope">
-          <span>{{ scope.row.totalCount - scope.row.status0Count }}</span>
+          <span>{{ scope.row.totalCount - scope.row.status0Count - scope.row.status2Count }}</span>
         </template>
       </el-table-column>
       <el-table-column label="操作" align="center" min-width="140" class-name="small-padding fixed-width">
@@ -118,8 +118,7 @@
     <add-task-dialog :open="addOpen" :get-list="getList" @dialog-cancel="handleCancel"></add-task-dialog>
 
     <release-version-dialog :get-list="getList" :dataset-id="datasetId" :open="releaseOpen" :task-id="taskId"
-                            :last-version-name="lastVersionName" @dialog-cancel="handleCancel"
-    ></release-version-dialog>
+                            :last-version-name="lastVersionName"  @dialog-cancel="handleCancel"></release-version-dialog>
 
   </div>
 </template>
diff --git a/src/views/dataCenter/annotationTask/dialog/releaseVersionDialog.vue b/src/views/dataCenter/annotationTask/dialog/releaseVersionDialog.vue
index 1533245b..ec98f04e 100644
--- a/src/views/dataCenter/annotationTask/dialog/releaseVersionDialog.vue
+++ b/src/views/dataCenter/annotationTask/dialog/releaseVersionDialog.vue
@@ -31,6 +31,7 @@
 
 
 import { add } from '@/api/dataCenter/releaseVersion'
+import { getLatestDatasetVersion } from '../../../../api/dataCenter/annotationTask'
 
 export default {
   props: {
@@ -41,7 +42,7 @@ export default {
     },
     getList: {
       type: Function,
-      required: true,
+      required: true
     },
     datasetId: {
       type: Number,
@@ -52,11 +53,6 @@ export default {
       type: Number,
       default: 0,
       required: true
-    },
-    lastVersionName: {
-      type: String,
-      default: '',
-      required: true
     }
   },
   computed: {
@@ -67,10 +63,11 @@ export default {
       set(value) {
         this.$emit('update:open', value)
       }
-    },
+    }
   },
   data() {
     return {
+      lastVersionName: null,
       form: {
         versionName: this.lastVersionName || '' // 初始化为传入的 lastVersionName
       }, // 初始化为空
@@ -83,32 +80,38 @@ export default {
     }
   },
   watch: {
-    isOpen(newVal){
-      if (newVal){
-        if (this.isValidVersion(this.lastVersionName)) {
-          this.form.versionName = this.incrementMajorVersion(this.lastVersionName);
-        } else {
-          this.form.versionName = 'V1.0.0'; // 如果格式不正确，则清空
-        }
+    isOpen(newVal) {
+      if (newVal) {
+        this.getLatestDatasetVersion(this.datasetId)
       }
-    },
+    }
   },
   methods: {
+    getLatestDatasetVersion(datasetId) {
+      getLatestDatasetVersion(datasetId).then(response => {
+       this.lastVersionName = response.msg;
+        if (this.isValidVersion(this.lastVersionName)) {
+          this.form.versionName = this.incrementMajorVersion(this.lastVersionName)
+        } else {
+          this.form.versionName = 'V1.0.0' // 如果格式不正确，则清空
+        }
+      })
+    },
     // 判断版本号格式是否正确
     isValidVersion(version) {
-      return /^V\d{1,3}(\.\d{1,3}){2}$/.test(version);
+      return /^V\d{1,3}(\.\d{1,3}){2}$/.test(version)
     },
 
     // 增加大版本号
     incrementMajorVersion(version) {
-      const match = version.match(/^V(\d{1,3})\.(\d{1,3})\.(\d{1,3})$/);
+      const match = version.match(/^V(\d{1,3})\.(\d{1,3})\.(\d{1,3})$/)
       if (match) {
-        let major = parseInt(match[1], 10);
+        let major = parseInt(match[1], 10)
         // 增加主版本号
-        major += 1;
-        return `V${major}.0.0`; // 返回新的版本号，次版本和修订号重置为 0
+        major += 1
+        return `V${major}.0.0` // 返回新的版本号，次版本和修订号重置为 0
       }
-      return version; // 如果格式不符合，则返回原始版本号
+      return version // 如果格式不符合，则返回原始版本号
     },
     // 数据提交
     submitForm() {
@@ -128,11 +131,12 @@ export default {
       this.isOpen = false
       this.reset()
       this.$emit('dialog-cancel') // 通知父组件
-      this.getList();
+      this.getList()
     },
     // 表单重置
     reset() {
-      this.form = {versionName:''} // 清空表单
+      this.lastVersionName = ''
+      this.form = { versionName: '' } // 清空表单
       this.$refs.form.resetFields() // 调用 Element UI 的重置方法
     }
   }
diff --git a/src/views/dataCenter/dataSet/child/annotationTaskDialog.vue b/src/views/dataCenter/dataSet/child/annotationTaskDialog.vue
index ea4f1998..09eadf7a 100644
--- a/src/views/dataCenter/dataSet/child/annotationTaskDialog.vue
+++ b/src/views/dataCenter/dataSet/child/annotationTaskDialog.vue
@@ -224,10 +224,10 @@ export default {
             let reviewers = []
             this.tableData.forEach((item) => {
               if (item.type === '0') {
-                reviewers.push(item)
+                reviewers.push(item.id)
               }
               if (item.type === '1') {
-                annotators.push(item)
+                annotators.push(item.id)
               }
             })
             this.form.annotators = annotators
diff --git a/src/views/dataCenter/dataSet/child/datasetVersion.vue b/src/views/dataCenter/dataSet/child/datasetVersion.vue
index 8c2aee29..e4e5121b 100644
--- a/src/views/dataCenter/dataSet/child/datasetVersion.vue
+++ b/src/views/dataCenter/dataSet/child/datasetVersion.vue
@@ -63,14 +63,12 @@
           <el-button
             size="mini"
             type="text"
-            icon="el-icon-delete"
             @click="handleExport(scope.row)"
             v-hasPermi="['dataCenter:version:delete']"
           >导出</el-button>
           <el-button
             size="mini"
             type="text"
-            icon="el-icon-delete"
             @click="handleDelete(scope.row)"
             v-hasPermi="['dataCenter:version:delete']"
           >删除</el-button>
@@ -190,7 +188,7 @@ export default {
     },
     handleExport(row){
 
-      this.download('/ai/annotations/export/' + row.taskId,{},new Date().getTime()+'.zip',{timeout: 600000})
+      this.download('/ai/annotations/export/' + row.versionId,{}, new Date().getTime()+'.zip',{timeout: 600000})
     },
     handleDelete(row){
       const ids = row.versionId || this.ids;
diff --git a/src/views/dataCenter/dataSet/child/exportDialog.vue b/src/views/dataCenter/dataSet/child/exportDialog.vue
new file mode 100644
index 00000000..441dc2e1
--- /dev/null
+++ b/src/views/dataCenter/dataSet/child/exportDialog.vue
@@ -0,0 +1,99 @@
+<template>
+  <div>
+    <el-dialog title="版本号选择" :visible.sync="isOpen" width="400px" append-to-body :modal="false" @close="cancel"
+               :close-on-click-modal="false"
+    >
+      <el-form ref="form" :model="form" :rules="rules" label-width="100px">
+        <el-form-item label="版本号" prop="versionId">
+          <el-select v-model="form.versionId" placeholder="请选择版本" clearable style="width: 100%">
+            <el-option v-for="item in versionList" :key="item.versionId" :label="item.versionName"
+                       :value="item.versionId"
+            />
+          </el-select>
+        </el-form-item>
+      </el-form>
+      <template #footer>
+        <el-button type="primary" @click="submitForm">确 定</el-button>
+        <el-button @click="cancel">取 消</el-button>
+      </template>
+    </el-dialog>
+  </div>
+</template>
+
+<script>
+
+import { listSelect } from '../../../../api/dataCenter/releaseVersion'
+
+export default {
+  props: {
+    open: {
+      type: Boolean,
+      default: false
+    },
+    datasetId: {
+      type: String,
+      default: '',
+      required: true
+    }
+  },
+  computed: {
+    isOpen: {
+      get() {
+        return this.open
+      },
+      set(value) {
+        this.$emit('update:open', value)
+      }
+    }
+  },
+  data() {
+    return {
+      versionList: [],
+      form: {},
+      rules: {
+        versionId: [
+          { required: true, message: '请选择版本', trigger: 'blur' }
+        ]
+      }
+    }
+  },
+  watch: {
+    isOpen(newVal, oldVal) {
+      if (newVal) {
+        this.listSelect()
+      }
+    }
+  },
+  methods: {
+    listSelect() {
+      listSelect({ datasetId: this.datasetId }).then(res => {
+        this.versionList = res.data
+      })
+    },
+    submitForm() {
+      this.$refs['form'].validate(valid => {
+        if (valid) {
+          this.download('/ai/annotations/export/' + this.form.versionId, {}, new Date().getTime() + '.zip', { timeout: 600000 }).then((res) => {
+            this.cancel()
+          })
+        }
+      })
+    },
+    cancel() {
+      this.isOpen = false
+      this.reset()
+      this.$emit('dialog-cancel') // 通知父组件
+    },
+    // 表单重置
+    reset() {
+      this.versionList = []
+      this.form = {} // 清空表单
+      this.$refs.form.resetFields() // 调用 Element UI 的重置方法
+    }
+  }
+}
+</script>
+
+<style scoped lang="scss">
+
+</style>
diff --git a/src/views/dataCenter/dataSet/components/dataSet.vue b/src/views/dataCenter/dataSet/components/dataSet.vue
index 7a6272fa..d0a43b6e 100644
--- a/src/views/dataCenter/dataSet/components/dataSet.vue
+++ b/src/views/dataCenter/dataSet/components/dataSet.vue
@@ -25,7 +25,8 @@
           size="mini"
           @click="handleAdd"
           v-hasPermi="['dataCenter:dataSet:create']"
-        >新增</el-button>
+        >新增
+        </el-button>
       </el-col>
       <el-col :span="1.5">
         <el-button
@@ -36,28 +37,32 @@
           :disabled="multiple"
           @click="handleDelete"
           v-hasPermi="['dataCenter:dataSet:delete']"
-        >删除</el-button>
+        >删除
+        </el-button>
       </el-col>
       <right-toolbar :showSearch.sync="showSearch" @queryTable="getList"></right-toolbar>
     </el-row>
 
-    <el-table  style="width: 100%" v-loading="loading"  :data="list"  :height="tableHeight"  @selection-change="handleSelectionChange">
-      <el-table-column type="selection" width="50" align="center" />
-      <el-table-column type="index" label="序号" :index="indexMethod" width="50" />
+    <el-table style="width: 100%" v-loading="loading" :data="list" :height="tableHeight"
+              @selection-change="handleSelectionChange"
+    >
+      <el-table-column type="selection" width="50" align="center"/>
+      <el-table-column type="index" label="序号" :index="indexMethod" width="50"/>
       <el-table-column label="名称" align="center" show-overflow-tooltip prop="datasetName">
         <template slot-scope="scope">
           <router-link :to="{
       path: '/dataCenter/dataSet-details/index/' + scope.row.datasetId,
       query: { dataType: scope.row.dataType,_t: Date.now() }
-    }" class="link-type" @click="">
+    }" class="link-type" @click=""
+          >
             <span>{{ scope.row.datasetName }}</span>
           </router-link>
         </template>
       </el-table-column>
-      <el-table-column label="版本名称" align="center" show-overflow-tooltip prop="latestVersionName" />
+      <el-table-column label="版本名称" align="center" show-overflow-tooltip prop="latestVersionName"/>
       <el-table-column label="文件数量" align="center" show-overflow-tooltip prop="createBy" min-width="100">
         <template slot-scope="scope">
-          <span>{{scope.row.annotatedCount }}</span>
+          <span>{{ scope.row.annotatedCount }}</span>
         </template>
 
       </el-table-column>
@@ -67,8 +72,8 @@
           <span>{{ parseTime(scope.row.createTime) }}</span>
         </template>
       </el-table-column>
-      <el-table-column label="描述" show-overflow-tooltip align="center" prop="datasetDesc" />
-      <el-table-column label="是否公开" show-overflow-tooltip align="center" prop="createBy" >
+      <el-table-column label="描述" show-overflow-tooltip align="center" prop="datasetDesc"/>
+      <el-table-column label="是否公开" show-overflow-tooltip align="center" prop="createBy">
         <template slot-scope="scope">
           <el-switch v-model="scope.row.isPublic" active-value="1" inactive-value="0"
                      @change="change(scope.row)"
@@ -83,33 +88,37 @@
             type="text"
             @click="handleImport(scope.row)"
             v-hasPermi="['dataCenter:dataSet:import']"
-          >导入</el-button>
+          >导入
+          </el-button>
           <el-button
             size="mini"
             type="text"
             @click="handleAnnotation(scope.row)"
             v-hasPermi="['dataCenter:task:create']"
-          >标注</el-button>
+          >标注
+          </el-button>
           <el-button
             size="mini"
             type="text"
-
             @click="downloadFile(scope.row)"
             v-hasPermi="['dataCenter:dataSet:export']"
-          >导出</el-button>
+          >导出
+          </el-button>
           <el-button
             size="mini"
             type="text"
 
             @click="handleVersion(scope.row)"
             v-hasPermi="['dataCenter:version:list']"
-          >版本查看</el-button>
+          >版本查看
+          </el-button>
           <el-button
             size="mini"
             type="text"
             @click="handleDelete(scope.row)"
             v-hasPermi="['dataCenter:dataSet:delete']"
-          >删除</el-button>
+          >删除
+          </el-button>
         </template>
       </el-table-column>
     </el-table>
@@ -124,9 +133,18 @@
     <!-- 添加对话框 -->
     <add-data-set-dialog :get-list="getList" :open="addOpen" @dialog-cancel="handleCancel"/>
     <!--创建标注任务-->
-    <annotation-task-dialog :datasetNum="datasetNum" :dataset-id="datasetId" :open="annotationOpen" @dialog-cancel="handleCancel"/>
+    <annotation-task-dialog :datasetNum="datasetNum" :dataset-id="datasetId" :open="annotationOpen"
+                            @dialog-cancel="handleCancel"
+    />
     <!-- 导入 -->
-    <import-dialog :get-list="getList" :open="importOpen" :data-type="dataType" :dataset-id="datasetId.toString()"  @dialog-cancel="handleCancel"/>
+    <import-dialog :get-list="getList" :open="importOpen" :data-type="dataType" :dataset-id="datasetId.toString()"
+                   @dialog-cancel="handleCancel"
+    />
+
+    <export-dialog :get-list="getList" :open="exportOpen"  :dataset-id="datasetId.toString()"
+                   @dialog-cancel="handleCancel"
+    />
+
   </div>
 </template>
 <script>
@@ -134,17 +152,16 @@
 import addDataSetDialog from '../child/addDataSetDialog.vue'
 import annotationTaskDialog from '../child/annotationTaskDialog.vue'
 import importDialog from '../child/importDialog.vue'
-
-import { list,del,setPublic} from '@/api/dataCenter/dataSet'
-
+import exportDialog from '../child/exportDialog.vue'
+import { list, del, setPublic } from '@/api/dataCenter/dataSet'
 
 export default {
-  components: { addDataSetDialog,annotationTaskDialog,importDialog},
-  name: "DataSet",
+  components: { addDataSetDialog, annotationTaskDialog, importDialog, exportDialog },
+  name: 'DataSet',
   data() {
     return {
-      dataType:'0',
-      datasetNum:0,
+      dataType: '0',
+      datasetNum: 0,
       datasetId: 0,
       // 遮罩层
       loading: true,
@@ -164,91 +181,93 @@ export default {
       // 是否显示弹出层
       addOpen: false,
       releaseOpen: false,
-      annotationOpen:false,
-      importOpen:false,
+      annotationOpen: false,
+      importOpen: false,
+      exportOpen: false,
       // 查询参数
       queryParams: {
         pageNum: 1,
         pageSize: 10,
-        datasetName: null,
-      },
-    };
+        datasetName: null
+      }
+    }
   },
   created() {
-    this.getList();
+    this.getList()
     // 初始化表格高度
-    this.updateTableHeight();
+    this.updateTableHeight()
     // 监听窗口大小变化
-    window.addEventListener("resize", this.updateTableHeight);
+    window.addEventListener('resize', this.updateTableHeight)
   },
   methods: {
-    indexMethod(index){
-      return (this.queryParams.pageNum - 1) * this.queryParams.pageSize + index + 1;
+    indexMethod(index) {
+      return (this.queryParams.pageNum - 1) * this.queryParams.pageSize + index + 1
     },
     updateTableHeight() {
       // 设置表格高度为窗口高度减去其他元素高度
-      const headerHeight = 300; // 头部高度，可以调整
-      const footerHeight = 80; // 底部高度，可以调整
-      this.tableHeight = window.innerHeight - headerHeight - footerHeight;
+      const headerHeight = 300 // 头部高度，可以调整
+      const footerHeight = 80 // 底部高度，可以调整
+      this.tableHeight = window.innerHeight - headerHeight - footerHeight
     },
     /**获取数据 **/
-    getList(){
-      this.loading =true
+    getList() {
+      this.loading = true
       list(this.queryParams).then(response => {
-        console.log(response);
-        this.list = response.rows;
-        this.total = response.total;
-        this.loading =false
+        console.log(response)
+        this.list = response.rows
+        this.total = response.total
+        this.loading = false
       })
     },
     /** 搜索按钮操作 */
     handleQuery() {
-      this.queryParams.pageNum = 1;
-      this.getList();
+      this.queryParams.pageNum = 1
+      this.getList()
     },
     /** 重置按钮操作 */
     resetQuery() {
-      this.resetForm("queryForm");
-      this.handleQuery();
+      this.resetForm('queryForm')
+      this.handleQuery()
     },
-    handleAdd(){
-      this.addOpen = true;
+    handleAdd() {
+      this.addOpen = true
     },
-    handleAnnotation(row){
-      this.annotationOpen = true;
-      this.datasetId = row.datasetId;
-      this.datasetNum = row.notAnnotatedCount;
+    handleAnnotation(row) {
+      this.annotationOpen = true
+      this.datasetId = row.datasetId
+      this.datasetNum = row.notAnnotatedCount
     },
-    handleImport(row){
-      this.importOpen = true;
-      this.datasetId = row.datasetId;
-      this.dataType = row.dataType;
+    handleImport(row) {
+      this.importOpen = true
+      this.datasetId = row.datasetId
+      this.dataType = row.dataType
     },
-    handleUpdate(){
+    handleUpdate() {
 
     },
-    handleVersion(row){
-      this.$tab.openPage("数据集版本管理", '/dataCenter/dataSet-version/index/' + row.datasetId,
-        {_t: Date.now() });
+    handleVersion(row) {
+      this.$tab.openPage('数据集版本管理', '/dataCenter/dataSet-version/index/' + row.datasetId,
+        { _t: Date.now() })
     },
     getPercentage(row) {
       if (row.annotatedCount > 0 && row.notAnnotatedCount >= 0) {
-        let percentage = ((row.annotatedCount - row.notAnnotatedCount) / row.annotatedCount) * 100;
+        let percentage = ((row.annotatedCount - row.notAnnotatedCount) / row.annotatedCount) * 100
         // 确保 percentage 在 0 到 100 之间
-        return Math.min(Math.max(percentage, 0), 100);
+        return Math.min(Math.max(percentage, 0), 100)
       }
-      return 0; // 如果没有标注数据，返回 0%
+      return 0 // 如果没有标注数据，返回 0%
     },
-    handleDelete(row){
-      const ids = row.datasetId || this.ids;
+    handleDelete(row) {
+      const ids = row.datasetId || this.ids
       this.$modal.confirm('是否确认删除数据项？').then(function() {
-        return del(ids);
+        return del(ids)
       }).then(() => {
-        this.getList();
-        this.$modal.msgSuccess("删除成功");
-      }).catch(() => {});
+        this.getList()
+        this.$modal.msgSuccess('删除成功')
+      }).catch(() => {
+      })
     },
-    change(row){
+    change(row) {
       let text = row.isPublic === '0' ? '不公开' : '公开'
       this.$modal.confirm('确认要' + text + '吗？').then(function() {
         return setPublic(row.datasetId, row.isPublic)
@@ -261,18 +280,19 @@ export default {
     // 多选框选中数据
     handleSelectionChange(selection) {
       this.ids = selection.map(item => item.datasetId)
-      this.single = selection.length!==1
+      this.single = selection.length !== 1
       this.multiple = !selection.length
     },
     // 处理取消事件
     handleCancel() {
-      this.addOpen = false;
-      this.annotationOpen = false;
-      this.importOpen = false;
+      this.addOpen = false
+      this.annotationOpen = false
+      this.importOpen = false
+      this.exportOpen = false
     },
     downloadFile(row) {
-      const datasetId = row.datasetId;
-      this.download('/ai/dataset/export/'+datasetId,{},`${new Date().getTime()}.zip`,{timeout: 600000})
+      this.exportOpen = true
+      this.datasetId = row.datasetId
     }
   }
 }
@@ -281,10 +301,12 @@ export default {
 .demo-table-expand {
   font-size: 0;
 }
+
 .demo-table-expand label {
   width: 90px;
   color: #99a9bf;
 }
+
 .demo-table-expand .el-form-item {
   margin-right: 0;
   margin-bottom: 0;