diff --git a/src/main/java/com/kamco/cd/kamcoback/model/service/ModelMngService.java b/src/main/java/com/kamco/cd/kamcoback/model/service/ModelMngService.java index 38929a9e..45bbd8db 100644 --- a/src/main/java/com/kamco/cd/kamcoback/model/service/ModelMngService.java +++ b/src/main/java/com/kamco/cd/kamcoback/model/service/ModelMngService.java @@ -15,6 +15,7 @@ import com.kamco.cd.kamcoback.upload.service.UploadService; import java.io.IOException; import java.nio.file.Files; import java.nio.file.Path; +import java.nio.file.Paths; import java.time.LocalDate; import java.util.List; import java.util.UUID; @@ -240,7 +241,8 @@ public class ModelMngService { int endPos = 20; List files = - FIleChecker.getFilesFromAllDepth(dirPath, "*", "pth,py,json", 10, "name", startPos, endPos); + FIleChecker.getFilesFromAllDepth( + dirPath, "*", "pth,pt,py,json", 10, "name", startPos, endPos); boolean hasPt = false; // pt 파일 존재 여부 @@ -257,6 +259,11 @@ public class ModelMngService { modelUploadResDto.setCdModelConfigPath(foldNm); modelUploadResDto.setCdModelConfigFileName(dto.getFileNm()); } + case "pt" -> { + modelUploadResDto.setClsModelPath(foldNm); + modelUploadResDto.setClsModelFileName(dto.getFileNm()); + hasPt = true; + } case "json" -> { modelUploadResDto.setJsonPath(foldNm); modelUploadResDto.setJsonFileName(dto.getFileNm()); @@ -264,9 +271,18 @@ public class ModelMngService { } } - // pt는 고정경로 등록 - modelUploadResDto.setClsModelPath(ptPath); - modelUploadResDto.setClsModelFileName(ptFileName); + if (!hasPt) { + + String defaultPath = ptPath; + String defaultFileName = ptFileName; + + Path ptPath = Paths.get(defaultPath, defaultFileName); + + if (Files.exists(ptPath)) { + modelUploadResDto.setClsModelPath(defaultPath); + modelUploadResDto.setClsModelFileName(defaultFileName); + } + } // int fileListPos = 0; // int fileTotCnt = files.size();