From 7f46a805976074120debc54bc6780d8f3b144c52 Mon Sep 17 00:00:00 2001
From: zouap <zouap@pcl.ac.cn>
Date: Fri, 3 Dec 2021 17:12:37 +0800
Subject: [PATCH] =?UTF-8?q?=E6=8F=90=E4=BA=A4=E4=BB=A3=E7=A0=81=E3=80=82?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: zouap <zouap@pcl.ac.cn>
---
 options/locale/locale_zh-CN.ini |  1 +
 routers/repo/ai_model_manage.go | 31 +++++--------------------------
 2 files changed, 6 insertions(+), 26 deletions(-)

diff --git a/options/locale/locale_zh-CN.ini b/options/locale/locale_zh-CN.ini
index d897c131f..b728b64d9 100755
--- a/options/locale/locale_zh-CN.ini
+++ b/options/locale/locale_zh-CN.ini
@@ -894,6 +894,7 @@ modelarts.train_job_para_admin=任务参数管理
 modelarts.train_job_para.edit=编辑
 modelarts.train_job_para.connfirm=确定
 
+model.manage.create_error=相同的名称和版本的模型已经存在。
 
 template.items=模板选项
 template.git_content=Git数据(默认分支)
diff --git a/routers/repo/ai_model_manage.go b/routers/repo/ai_model_manage.go
index 833c41c14..1bb21e01c 100644
--- a/routers/repo/ai_model_manage.go
+++ b/routers/repo/ai_model_manage.go
@@ -358,33 +358,12 @@ func ShowModelInfo(ctx *context.Context) {
 }
 
 func ShowSingleModel(ctx *context.Context) {
-	id := ctx.Query("ID")
-	parentDir := ctx.Query("parentDir")
-	log.Info("Show single ModelInfo start.id=" + id)
-	task, err := models.QueryModelById(id)
-	if err != nil {
-		log.Error("no such model!", err.Error())
-		ctx.ServerError("no such model:", err)
-		return
-	}
-	log.Info("bucket=" + setting.Bucket + " key=" + task.Path[len(setting.Bucket)+1:])
-	models, err := storage.GetOneLevelAllObjectUnderDir(setting.Bucket, task.Path[len(setting.Bucket)+1:], parentDir)
-	if err != nil {
-		log.Info("get model list failed:", err)
-		ctx.ServerError("GetObsListObject:", err)
-		return
-	} else {
-		log.Info("get model file,size=" + fmt.Sprint(len(models)))
-	}
+	name := ctx.Query("name")
 
-	if err != nil {
-		ctx.ServerError("QueryTrainJobList:", err)
-	} else {
-		mapInterface := make(map[string]interface{})
-		mapInterface["task"] = task
-		mapInterface["models"] = models
-		ctx.JSON(http.StatusOK, mapInterface)
-	}
+	log.Info("Show single ModelInfo start.name=" + name)
+	models := models.QueryModelByName(name, ctx.Repo.Repository.ID)
+
+	ctx.JSON(http.StatusOK, models)
 }
 
 func ShowOneVersionOtherModel(ctx *context.Context) {