权限管理

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) {