提交测试代码。

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

routers/repo/ai_model_manage.go

@@ -110,12 +110,17 @@ func downloadModelFromCloudBrainTwo(modelUUID string, jobName string, parentDir
 	}
 	var size int64
 	prefix := strings.TrimPrefix(path.Join(setting.TrainJobModelPath, jobName, setting.OutPutPath, parentDir), "/") + "/"
+	destKeyNamePrefix := Model_prefix + models.AttachmentRelativePath(modelUUID) + "/"
 	for _, modelFile := range modelDbResult {
-		destKeyNamePrefix := Model_prefix + models.AttachmentRelativePath(modelUUID) + "/"
-
-		log.Info("copy file, bucket=" + setting.Bucket + ", src keyname=" + prefix + modelFile.FileName)
-		log.Info("Dest key name=" + destKeyNamePrefix + modelFile.FileName)
-		err := storage.ObsCopyFile(setting.Bucket, prefix+modelFile.FileName, setting.Bucket, destKeyNamePrefix+modelFile.FileName)
+		if modelFile.IsDir {
+			log.Info("copy dir, continue. dir=" + modelFile.FileName)
+			continue
+		}
+		srcKeyName := prefix + modelFile.ParenDir + modelFile.FileName
+		log.Info("copy file, bucket=" + setting.Bucket + ", src keyname=" + srcKeyName)
+		destKeyName := destKeyNamePrefix + modelFile.ParenDir + modelFile.FileName
+		log.Info("Dest key name=" + destKeyName)
+		err := storage.ObsCopyFile(setting.Bucket, srcKeyName, setting.Bucket, destKeyName)
 		if err != nil {
 			log.Info("copy failed.")
 		}
@@ -207,14 +212,13 @@ func ShowSingleModel(ctx *context.Context) {
 		ctx.ServerError("GetObsListObject:", err)
 		return
 	} else {
-		log.Info("get model file,size=" + fmt.Sprint(models))
+		log.Info("get model file,size=" + fmt.Sprint(len(models)))
 	}
 
 	ctx.Data["Dirs"] = models
 	ctx.Data["task"] = task
 	ctx.Data["ID"] = id
 	ctx.HTML(200, tplModelManageDownload)
-
 }
 
 func ShowModelPageInfo(ctx *context.Context) {

templates/repo/modelmanage/download.tmpl

@@ -18,7 +18,7 @@
           <div class="dir list">
              <a class="title" href="{{$.RepoLink}}/modelmanage/downloadall?id={{.ID}}">
                   下载所有
-            </a>
+             </a>
             {{if .Dirs}}
                 <table id="repo-files-table" class="ui single line table">
                   <tbody>