From 8de31e8bf0c88b4ba85b8e1f41820b80f7e12a2e Mon Sep 17 00:00:00 2001
From: zhoupzh <zhoupzh@pcl.ac.cn>
Date: Wed, 7 Dec 2022 16:01:12 +0800
Subject: [PATCH] fix issue

---
 templates/repo/cloudbrain/new.tmpl            |  4 ++-
 templates/repo/modelarts/notebook/new.tmpl    |  2 +-
 .../js/components/dataset/selectDataset.vue   | 30 +++++++++++++++++++
 web_src/js/features/i18nVue.js                |  2 ++
 4 files changed, 36 insertions(+), 2 deletions(-)

diff --git a/templates/repo/cloudbrain/new.tmpl b/templates/repo/cloudbrain/new.tmpl
index b75e75cdf..cb18517b3 100755
--- a/templates/repo/cloudbrain/new.tmpl
+++ b/templates/repo/cloudbrain/new.tmpl
@@ -23,7 +23,8 @@
     {{template "repo/header" .}}
     <div class="repository new repo ui middle very relaxed page grid">
         <div class="column">
-            <div class="cloudbrain-type" style="display: none;" data-cloudbrain-type="{{.datasetType}}" data-repo-link="{{.RepoLink}}" data-queue="{{.QueuesDetail}}" data-queue-start="{{.i18n.Tr "repo.wait_count_start"}}" data-queue-end="{{.i18n.Tr "repo.wait_count_end"}}"></div>
+            <div class="cloudbrain-type" style="display: none;" data-cloudbrain-type="{{.datasetType}}" data-repo-link="{{.RepoLink}}" data-queue="{{.QueuesDetail}}" 
+                data-queue-start="{{.i18n.Tr "repo.wait_count_start"}}" data-queue-end="{{.i18n.Tr "repo.wait_count_end"}}" data-exceed-size="{{DebugAttachSize}}"></div>
             {{if eq .NotStopTaskCount 0}}
                 {{template "base/alert" .}}
             {{end}}
@@ -216,6 +217,7 @@
 {{template "base/footer" .}}
 <script src="{{StaticUrlPrefix}}/js/specsuse.js?v={{MD5 AppVer}}" type="text/javascript"></script>
 <script>
+    console.log({{DebugAttachSize}});
     let form = document.getElementById('form_id');
     $('#messageInfo').css('display', 'none')
     function clearValue() {
diff --git a/templates/repo/modelarts/notebook/new.tmpl b/templates/repo/modelarts/notebook/new.tmpl
index e149d6095..5be95b932 100755
--- a/templates/repo/modelarts/notebook/new.tmpl
+++ b/templates/repo/modelarts/notebook/new.tmpl
@@ -10,7 +10,7 @@
     {{template "repo/header" .}}
     <div class="repository new repo ui middle very relaxed page grid">
         <div class="column">
-            <div class="cloudbrain-type" style="display: none;" data-cloudbrain-type="{{.datasetType}}" data-repo-link="{{.RepoLink}}"></div>
+            <div class="cloudbrain-type" style="display: none;" data-cloudbrain-type="{{.datasetType}}" data-repo-link="{{.RepoLink}}" data-exceed-size="{{DebugAttachSize}}"></div>
             {{if eq .NotStopTaskCount 0}}
                 {{template "base/alert" .}}
             {{end}}
diff --git a/web_src/js/components/dataset/selectDataset.vue b/web_src/js/components/dataset/selectDataset.vue
index 057298d13..eef2ee322 100755
--- a/web_src/js/components/dataset/selectDataset.vue
+++ b/web_src/js/components/dataset/selectDataset.vue
@@ -189,6 +189,12 @@
                       >
                         {{ i18n.dataset_unzip_failed }}
                       </span>
+                      <span
+                        class="unzip-failed"
+                        v-if="data.Size > exceedSize"
+                      >
+                        {{ i18n.dataset_exceeds_failed }}
+                      </span>
                     </span>
                   </span>
                 </el-tree>
@@ -295,6 +301,12 @@
                       >
                         {{ i18n.dataset_unzip_failed }}
                       </span>
+                      <span
+                        class="unzip-failed"
+                        v-if="data.Size > exceedSize"
+                      >
+                        {{ i18n.dataset_exceeds_failed }}
+                      </span>
                     </span>
                   </span>
                 </el-tree>
@@ -401,6 +413,12 @@
                       >
                         {{ i18n.dataset_unzip_failed }}
                       </span>
+                      <span
+                        class="unzip-failed"
+                        v-if="data.Size > exceedSize"
+                      >
+                        {{ i18n.dataset_exceeds_failed }}
+                      </span>
                     </span>
                   </span>
                 </el-tree>
@@ -507,6 +525,12 @@
                       >
                         {{ i18n.dataset_unzip_failed }}
                       </span>
+                      <span
+                        class="unzip-failed"
+                        v-if="data.Size > exceedSize"
+                      >
+                        {{ i18n.dataset_exceeds_failed }}
+                      </span>
                     </span>
                   </span>
                 </el-tree>
@@ -639,6 +663,7 @@ export default {
       paramsFavorite: { page: 1, q: "" },
       loadingFavorite: false,
       initFavoriteTreeNode: [],
+      exceedSize:0,
     };
   },
   methods: {
@@ -746,6 +771,7 @@ export default {
         .then((res) => {
           this.loadingCurrent = false;
           let data = JSON.parse(res.data.data);
+          console.log(data);
           this.currentDatasetList = this.transformeTreeData(
             data,
             "currentTree",
@@ -897,6 +923,9 @@ export default {
             if (curchild.DecompressState !== 1) {
               curchild.disabled = true;
             }
+            if(curchild.Size>this.exceedSize && this.exceedSize){
+              curchild.disabled = true;
+            }
             curchild.ref = ref;
             curchild.label = curchild.Name;
             preChild.push(curchild);
@@ -985,6 +1014,7 @@ export default {
   mounted() {
     this.type = $(".cloudbrain-type").data("cloudbrain-type");
     this.repoLink = $(".cloudbrain-type").data("repo-link");
+    this.exceedSize = $(".cloudbrain-type").data("exceed-size");
     if ($(".cloudbrain-type").data("dataset-uuid")) {
       this.hasSelectDatasetList = $(".cloudbrain-type")
         .data("dataset-uuid")
diff --git a/web_src/js/features/i18nVue.js b/web_src/js/features/i18nVue.js
index 5e8c55e62..05b40c09a 100644
--- a/web_src/js/features/i18nVue.js
+++ b/web_src/js/features/i18nVue.js
@@ -74,6 +74,7 @@ export const i18nVue = {
     dataset_search_placeholder: "搜数据集名称/描述...",
     dataset_unziping: "正在解压缩",
     dataset_unzip_failed: "解压失败",
+    dataset_exceeds_failed: "数据集大小超过20G",
     dataset_my_upload: "我上传的",
     dataset_current_repo: "本项目",
     dataset_public: "公开数据集",
@@ -190,6 +191,7 @@ export const i18nVue = {
     dataset_search_placeholder: "Search dataset name/description ...",
     dataset_unziping: "Decompressing",
     dataset_unzip_failed: "Decompression failed",
+    dataset_exceeds_failed: "Dataset size exceeds 20G",
     dataset_my_upload: "Upload by me",
     dataset_current_repo: "Current Repository",
     dataset_public: "Public dataset",