diff --git a/routers/repo/grampus.go b/routers/repo/grampus.go new file mode 100755 index 000000000..3e8329dfb --- /dev/null +++ b/routers/repo/grampus.go @@ -0,0 +1,22 @@ +package repo + +import ( + "code.gitea.io/gitea/modules/base" + "net/http" + + "code.gitea.io/gitea/modules/context" +) + +const ( + tplGrampusTrainJobNew base.TplName = "repo/grampus/trainjob/new" + tplGrampusTrainJobShow base.TplName = "repo/grampus/trainjob/show" +) + +func GrampusNew(ctx *context.Context) { + err := cloudBrainNewDataPrepare(ctx) + if err != nil { + ctx.ServerError("get new train-job info failed", err) + return + } + ctx.HTML(http.StatusOK, tplGrampusTrainJobNew) +} diff --git a/routers/routes/routes.go b/routers/routes/routes.go index 4c3f5f472..a64eb0fae 100755 --- a/routers/routes/routes.go +++ b/routers/routes/routes.go @@ -1083,6 +1083,18 @@ func RegisterRoutes(m *macaron.Macaron) { m.Post("/create", reqWechatBind, reqRepoCloudBrainWriter, bindIgnErr(auth.CreateCloudBrainForm{}), repo.CloudBrainCreate) }) }, context.RepoRef()) + m.Group("/grampus", func() { + m.Group("/train-job", func() { + m.Group("/:jobid", func() { + m.Get("", reqRepoCloudBrainReader, repo.CloudBrainTrainJobShow) + m.Post("/stop", cloudbrain.AdminOrOwnerOrJobCreaterRight, repo.CloudBrainStop) + m.Post("/del", cloudbrain.AdminOrOwnerOrJobCreaterRightForTrain, repo.CloudBrainTrainJobDel) + m.Get("/download_model", cloudbrain.AdminOrOwnerOrJobCreaterRightForTrain, repo.CloudBrainDownloadModel) + }) + m.Get("/create", reqWechatBind, reqRepoCloudBrainWriter, repo.GrampusNew) + m.Post("/create", reqWechatBind, reqRepoCloudBrainWriter, bindIgnErr(auth.CreateCloudBrainForm{}), repo.GrampusCreate) + }) + }, context.RepoRef()) m.Group("/modelmanage", func() { m.Post("/create_model", reqRepoModelManageWriter, repo.SaveModel) m.Post("/create_new_model", repo.SaveNewNameModel)