Browse Source

add param check

pull/625/head
lewis 4 years ago
parent
commit
b082f861f2
2 changed files with 24 additions and 0 deletions
  1. +23
    -0
      routers/repo/modelarts.go
  2. +1
    -0
      templates/repo/modelarts/trainjob/new.tmpl

+ 23
- 0
routers/repo/modelarts.go View File

@@ -329,6 +329,8 @@ func TrainJobNew(ctx *context.Context) {
ctx.HTML(200, tplModelArtsTrainJobNew) ctx.HTML(200, tplModelArtsTrainJobNew)
} }


//todo: show the error in html
//todo; param check
func TrainJobCreate(ctx *context.Context, form auth.CreateModelArtsTrainJobForm) { func TrainJobCreate(ctx *context.Context, form auth.CreateModelArtsTrainJobForm) {
ctx.Data["PageIsCloudBrain"] = true ctx.Data["PageIsCloudBrain"] = true
jobName := form.JobName jobName := form.JobName
@@ -346,6 +348,13 @@ func TrainJobCreate(ctx *context.Context, form auth.CreateModelArtsTrainJobForm)
logObsPath := "/" + setting.Bucket + modelarts.JobPath + jobName + modelarts.LogPath logObsPath := "/" + setting.Bucket + modelarts.JobPath + jobName + modelarts.LogPath
dataPath := "/" + setting.Bucket + "/" + setting.BasePath + path.Join(uuid[0:1], uuid[1:2]) + "/" + uuid + "/" dataPath := "/" + setting.Bucket + "/" + setting.BasePath + path.Join(uuid[0:1], uuid[1:2]) + "/" + uuid + "/"


//param check
if err := paramCheckCreateTrainJob(form); err != nil {
log.Error("paramCheckCreateTrainJob failed:(%v)", err)
ctx.RenderWithErr(err.Error(), tplModelArtsTrainJobNew, &form)
return
}

if err := git.Clone(repo.RepoPath(), codeLocalPath, git.CloneRepoOptions{}); err != nil { if err := git.Clone(repo.RepoPath(), codeLocalPath, git.CloneRepoOptions{}); err != nil {
log.Error("Failed to clone repository: %s (%v)", repo.FullName(), err) log.Error("Failed to clone repository: %s (%v)", repo.FullName(), err)
ctx.RenderWithErr("Failed to clone repository", tplModelArtsTrainJobNew, &form) ctx.RenderWithErr("Failed to clone repository", tplModelArtsTrainJobNew, &form)
@@ -467,3 +476,17 @@ func obsMkdir(dir string) error {


return nil return nil
} }

func paramCheckCreateTrainJob(form auth.CreateModelArtsTrainJobForm) error {
if !strings.HasSuffix(form.BootFile, ".py") {
log.Error("the boot file(%s) must be a python file", form.BootFile)
return errors.New("the boot file is not a python file")
}

if form.WorkServerNumber > 25 || form.WorkServerNumber < 1{
log.Error("the WorkServerNumber(%d) must be in (1,25)", form.WorkServerNumber)
return errors.New("the WorkServerNumber(%d) must be in (1,25)")
}

return nil
}

+ 1
- 0
templates/repo/modelarts/trainjob/new.tmpl View File

@@ -98,6 +98,7 @@
<div class="repository"> <div class="repository">
{{template "repo/header" .}} {{template "repo/header" .}}
<div class="ui container"> <div class="ui container">
{{template "base/alert" .}}
<h4 class="ui top attached header"> <h4 class="ui top attached header">
{{.i18n.Tr "repo.modelarts.train_job.new"}} {{.i18n.Tr "repo.modelarts.train_job.new"}}
</h4> </h4>


Loading…
Cancel
Save