From fd2699373419ce14aaae5402a8d6fdf612726906 Mon Sep 17 00:00:00 2001
From: zouap <zouap@pcl.ac.cn>
Date: Thu, 13 Oct 2022 09:16:50 +0800
Subject: [PATCH] =?UTF-8?q?=E5=8E=BB=E6=8E=89=E6=A8=A1=E5=9E=8B=E4=BF=A1?=
 =?UTF-8?q?=E6=81=AF=E4=B8=ADIP=E5=9C=B0=E5=9D=80?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: zouap <zouap@pcl.ac.cn>
---
 routers/repo/ai_model_manage.go | 15 +++++++++++++--
 1 file changed, 13 insertions(+), 2 deletions(-)

diff --git a/routers/repo/ai_model_manage.go b/routers/repo/ai_model_manage.go
index 6aec541e3..5b358b83b 100644
--- a/routers/repo/ai_model_manage.go
+++ b/routers/repo/ai_model_manage.go
@@ -8,6 +8,7 @@ import (
 	"net/http"
 	"net/url"
 	"path"
+	"regexp"
 	"strings"
 
 	"code.gitea.io/gitea/models"
@@ -85,6 +86,7 @@ func saveModelByParameters(jobId string, versionName string, name string, versio
 	accuracy["Precision"] = ""
 	accuracyJson, _ := json.Marshal(accuracy)
 	log.Info("accuracyJson=" + string(accuracyJson))
+	aiTask.ContainerIp = ""
 	aiTaskJson, _ := json.Marshal(aiTask)
 
 	model := &models.AiModelManage{
@@ -635,16 +637,23 @@ func ShowSingleModel(ctx *context.Context) {
 	userNameMap := queryUserName(userIds)
 
 	for _, model := range models {
+		removeIpInfo(model)
 		value := userNameMap[model.UserId]
 		if value != nil {
 			model.UserName = value.Name
 			model.UserRelAvatarLink = value.RelAvatarLink()
-			model.TrainTaskInfo = ""
 		}
 	}
 	ctx.JSON(http.StatusOK, models)
 }
 
+func removeIpInfo(model *models.AiModelManage) {
+	reg, _ := regexp.Compile(`[[:digit:]]{1,3}\.[[:digit:]]{1,3}\.[[:digit:]]{1,3}\.[[:digit:]]{1,3}`)
+	taskInfo := model.TrainTaskInfo
+	taskInfo = reg.ReplaceAllString(taskInfo, "")
+	model.TrainTaskInfo = taskInfo
+}
+
 func queryUserName(intSlice []int64) map[int64]*models.User {
 	keys := make(map[int64]string)
 	uniqueElements := []int64{}
@@ -678,6 +687,7 @@ func ShowOneVersionOtherModel(ctx *context.Context) {
 	userNameMap := queryUserName(userIds)
 
 	for _, model := range aimodels {
+		removeIpInfo(model)
 		value := userNameMap[model.UserId]
 		if value != nil {
 			model.UserName = value.Name
@@ -788,13 +798,13 @@ func ShowModelPageInfo(ctx *context.Context) {
 	for i, model := range modelResult {
 		model.IsCanOper = isOper(ctx, model.UserId)
 		model.IsCanDelete = isCanDelete(ctx, model.UserId)
-		model.TrainTaskInfo = ""
 		userIds[i] = model.UserId
 	}
 
 	userNameMap := queryUserName(userIds)
 
 	for _, model := range modelResult {
+		removeIpInfo(model)
 		value := userNameMap[model.UserId]
 		if value != nil {
 			model.UserName = value.Name
@@ -868,6 +878,7 @@ func QueryModelListForPredict(ctx *context.Context) {
 
 	nameMap := make(map[string][]*models.AiModelManage)
 	for _, model := range modelResult {
+		removeIpInfo(model)
 		if _, value := nameMap[model.Name]; !value {
 			models := make([]*models.AiModelManage, 0)
 			models = append(models, model)