提交代码。

Signed-off-by: zouap <zouap@pcl.ac.cn>

models/ai_model_manage.go

@@ -86,6 +86,22 @@ func DeleteModelById(id string) error {
 
 }
 
+func ModifyModelDescription(id string, description string) error {
+	sess := x.NewSession()
+	defer sess.Close()
+
+	re, err := sess.Update(&AiModelManage{
+		ID:          id,
+		Description: description,
+	})
+	if err != nil {
+		return err
+	}
+	log.Info("success to update description from  db.re=" + fmt.Sprint((re)))
+	return nil
+
+}
+
 func QueryModelByName(name string, uid int64) []*AiModelManage {
 	sess := x.NewSession()
 	defer sess.Close()

routers/private/internal.go

@@ -47,5 +47,6 @@ func RegisterRoutes(m *macaron.Macaron) {
 		m.Post("/tool/create_model", CreateModel)
 		m.Delete("/tool/delete_model", DeleteModel)
 		m.Get("/tool/show_model", ShowModel)
+		m.Put("/tool/modify_model", ModifyModel)
 	}, CheckInternalToken)
 }

routers/private/tool.go

@@ -76,3 +76,14 @@ func ShowModel(ctx *macaron.Context) {
 		ctx.JSON(500, "query error.")
 	}
 }
+
+func ModifyModel(ctx *macaron.Context) {
+	id := ctx.Query("id")
+	description := ctx.Query("Description")
+	err := repo.ModifyModel(id, description)
+	if err == nil {
+		ctx.JSON(200, "Success.")
+	} else {
+		ctx.JSON(500, "Failed.")
+	}
+}

routers/repo/ai_model_manage.go

@@ -199,3 +199,28 @@ func ShowModelInfo(ctx *context.Context) {
 	ctx.Data["Tasks"] = modelResult
 	ctx.HTML(200, "")
 }
+
+func ModifyModel(id string, description string) error {
+	err := models.ModifyModelDescription(id, description)
+	if err == nil {
+		log.Info("modify success.")
+	} else {
+		log.Info("Failed to modify.id=" + id + "  desc=" + description)
+	}
+	return err
+}
+
+func ModifyModelInfo(ctx *context.Context) {
+	log.Info("delete model start.")
+	id := ctx.Query("ID")
+	description := ctx.Query("Description")
+
+	err := ModifyModel(id, description)
+
+	if err == nil {
+		ctx.HTML(200, "success")
+	} else {
+		ctx.HTML(500, "Failed.")
+	}
+
+}