모델등록 pt 수정

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<Basic> 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();