From b388c22818ebad61e528df20dd9a276693616984 Mon Sep 17 00:00:00 2001
From: lewis <747342561@qq.com>
Date: Tue, 2 Nov 2021 19:57:01 +0800
Subject: [PATCH] fix 680 677

---
 routers/repo/modelarts.go | 26 +++++++++++++++-----------
 1 file changed, 15 insertions(+), 11 deletions(-)

diff --git a/routers/repo/modelarts.go b/routers/repo/modelarts.go
index 856826bd7..ea8ab0c06 100755
--- a/routers/repo/modelarts.go
+++ b/routers/repo/modelarts.go
@@ -4,6 +4,7 @@ import (
 	"encoding/json"
 	"errors"
 	"io"
+	"io/ioutil"
 	"net/http"
 	"os"
 	"path"
@@ -643,18 +644,26 @@ func TrainJobCreate(ctx *context.Context, form auth.CreateModelArtsTrainJobForm)
 		return
 	}
 
+	attach, err := models.GetAttachmentByUUID(uuid)
+	if err != nil {
+		log.Error("GetAttachmentByUUID(%s) failed:%v", uuid, err.Error())
+		return
+	}
+
+	//todo: del the codeLocalPath
+	_, err = ioutil.ReadDir(codeLocalPath)
+	if err == nil {
+		os.RemoveAll(codeLocalPath)
+	}
 	if err := git.Clone(repo.RepoPath(), codeLocalPath, git.CloneRepoOptions{}); err != nil {
 		log.Error("创建任务失败，任务名称已存在！: %s (%v)", repo.FullName(), err)
 		trainJobNewDataPrepare(ctx)
-		attach, err := models.GetAttachmentByUUID(uuid)
-		if err != nil {
-			log.Error("GetAttachmentByUUID(%s) failed:%v", uuid, err.Error())
-			return
-		}
+
 		ctx.Data["bootFile"] = form.BootFile
 		ctx.Data["uuid"] = form.Attachment
 		ctx.Data["datasetName"] = attach.Name
 		ctx.Data["params"] = form.Params
+		trainJobNewDataPrepare(ctx)
 		// ctx.RenderWithErr("Failed to clone repository", tplModelArtsTrainJobNew, &form)
 		ctx.RenderWithErr("创建任务失败，任务名称已存在！", tplModelArtsTrainJobNew, &form)
 		// ctx.RenderWithErr(err, tplModelArtsTrainJobNew, &form)
@@ -763,15 +772,10 @@ func TrainJobCreate(ctx *context.Context, form auth.CreateModelArtsTrainJobForm)
 		Parameters:       param,
 	}
 
-	err := modelarts.GenerateTrainJob(ctx, req)
+	err = modelarts.GenerateTrainJob(ctx, req)
 	if err != nil {
 		log.Error("GenerateTrainJob failed:%v", err.Error())
 		trainJobNewDataPrepare(ctx)
-		attach, err := models.GetAttachmentByUUID(uuid)
-		if err != nil {
-			log.Error("GetAttachmentByUUID(%s) failed:%v", uuid, err.Error())
-			return
-		}
 		ctx.Data["bootFile"] = form.BootFile
 		ctx.Data["uuid"] = form.Attachment
 		ctx.Data["datasetName"] = attach.Name