From dd59dd2acf005323d51a78522a22fa12d7c45afa Mon Sep 17 00:00:00 2001
From: liuzx <liuzx@pcl.ac.cn>
Date: Thu, 25 Nov 2021 12:02:59 +0800
Subject: [PATCH] =?UTF-8?q?=E6=9D=83=E9=99=90=E7=AE=A1=E7=90=86?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 routers/repo/modelarts.go | 29 ++++++++++++++++-------------
 1 file changed, 16 insertions(+), 13 deletions(-)

diff --git a/routers/repo/modelarts.go b/routers/repo/modelarts.go
index a820c56fa..484715cb5 100755
--- a/routers/repo/modelarts.go
+++ b/routers/repo/modelarts.go
@@ -1160,6 +1160,14 @@ func TrainJobShow(ctx *context.Context) {
 		ctx.RenderWithErr(err.Error(), tplModelArtsTrainJobShow, nil)
 		return
 	}
+	//设置权限
+	canNewJob, err := canUserCreateTrainJobVersion(ctx, VersionListTasks[0].UserID)
+	if err != nil {
+		ctx.ServerError("canNewJob failed", err)
+		return
+	}
+	ctx.Data["canNewJob"] = canNewJob
+
 	//将运行参数转化为epoch_size = 3, device_target = Ascend的格式
 	for i, _ := range VersionListTasks {
 
@@ -1311,19 +1319,14 @@ func canUserCreateTrainJob(uid int64) (bool, error) {
 
 	return org.IsOrgMember(uid)
 }
-func canUserCreateTrainJobVersion(ctx *context.Context, jobID string, versionName string) (bool, error) {
-	// task, err := models.GetCloudbrainByJobIDAndVersionName(jobID, versionName)
-	// if err != nil {
-	// 	return false, err
-	// }
-	// if ctx.User.ID == task.UserID {
-	// 	canNewJob := true
-	// 	return canNewJob, nil
-	// } else {
-	// 	canNewJob := false
-	// 	return canNewJob, nil
-	// }
-	return true, nil
+func canUserCreateTrainJobVersion(ctx *context.Context, userID int64) (bool, error) {
+	if ctx.User.ID == userID || ctx.User.IsAdmin {
+		canNewJob := true
+		return canNewJob, nil
+	} else {
+		canNewJob := false
+		return canNewJob, nil
+	}
 }
 
 func TrainJobGetConfigList(ctx *context.Context) {