From 08ff65bf7b950ccfb7b5523e10b5e03709a29c4c Mon Sep 17 00:00:00 2001 From: liuzx Date: Wed, 12 Jan 2022 17:24:14 +0800 Subject: [PATCH] update --- routers/repo/modelarts.go | 12 ++++++++++++ routers/repo/setting.go | 11 ++--------- 2 files changed, 14 insertions(+), 9 deletions(-) diff --git a/routers/repo/modelarts.go b/routers/repo/modelarts.go index ef32e4292..65272c1d6 100755 --- a/routers/repo/modelarts.go +++ b/routers/repo/modelarts.go @@ -2102,3 +2102,15 @@ func DownloadMultiResultFile(ctx *context.Context) { ctx.ServerError("no file to download.", err) } } + +func SetJobCount(ctx *context.Context) { + repoId := ctx.Repo.Repository.ID + _, jobCount, err := models.Cloudbrains(&models.CloudbrainsOptions{ + RepoID: repoId, + }) + if err != nil { + ctx.ServerError("Get job faild:", err) + return + } + ctx.Data["jobCount"] = jobCount +} diff --git a/routers/repo/setting.go b/routers/repo/setting.go index afed8a5a9..7bb1a477b 100644 --- a/routers/repo/setting.go +++ b/routers/repo/setting.go @@ -51,6 +51,7 @@ func Settings(ctx *context.Context) { ctx.Data["PageIsSettingsOptions"] = true ctx.Data["ForcePrivate"] = setting.Repository.ForcePrivate SetModelCount(ctx) + SetJobCount(ctx) ctx.HTML(200, tplSettingsOptions) } @@ -59,17 +60,9 @@ func SettingsPost(ctx *context.Context, form auth.RepoSettingForm) { ctx.Data["Title"] = ctx.Tr("repo.settings") ctx.Data["PageIsSettingsOptions"] = true SetModelCount(ctx) + SetJobCount(ctx) repo := ctx.Repo.Repository - _, jobCount, err := models.Cloudbrains(&models.CloudbrainsOptions{ - RepoID: repo.ID, - }) - if err != nil { - ctx.ServerError("Get job faild:", err) - return - } - ctx.Data["jobCount"] = jobCount - switch ctx.Query("action") { case "update": if ctx.HasError() {