From 3d1c2a8c3dc7e56e40f22a8685e1d33999d5f518 Mon Sep 17 00:00:00 2001
From: Lee Young Hoon <nomalaid@gmail.com>
Date: Fri, 29 Nov 2024 16:16:59 +0900
Subject: [PATCH] Update UI.py
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

SubDirectory 처리됨. 자식의 자식 까지만.
---
 UI.py | 53 ++++++++++++++++++++++++++++++-----------------------
 1 file changed, 30 insertions(+), 23 deletions(-)

diff --git a/UI.py b/UI.py
index 7441ad1..a40d5e9 100644
--- a/UI.py
+++ b/UI.py
@@ -109,37 +109,44 @@ class MyApp(QMainWindow):
         if None == folder_path or True == util.IsEmptyStr(folder_path):
             return
         
-        # 폴더 목록에 추가
+        # 소스 폴더 목록에 추가
         self.list_SrcPath.addItem(folder_path)
 
-        # 폴더 내의 파일 목록을 가져온다.
-        listFiles = util.ListContainFiles(folder_path)
+        # 폴더 내의 자식 폴더 목록을 가져온다.
+        listFolders = util.ListSubDirectories(folder_path)
 
-        # 파일 목록을 훑어서 내용을 판단
-        # 1. 압축파일이 들어있나?
-        # 2. 이미지 파일이 들어있나?
-        isContainImg = False
-        for pathFile in listFiles:
-            filename = util.GetParentDirName(pathFile, 0)
-            fileExt = util.GetExtStr(filename)
+        for folder in listFolders:
+            # 폴더 내의 파일 목록을 가져온다.
+            listFiles = util.ListContainFiles(folder)
 
-            # 압축파일이 들어있다면...
-            if fileExt.lower() in [".zip", ".cbz", ".rar"]:
-                # 이름은 폴더/압축파일
-                FolderName = util.GetParentDirName(pathFile, 1)
-                ItemName = os.path.join(FolderName, filename)
-                item = QListWidgetItem(ItemName)
-                # 압축파일의 전체 경로를 따로 저장
-                item.setData(Qt.UserRole, pathFile)
-                self.list_ArcList.addItem(item)
+            # 파일 목록을 훑어서 내용을 판단
+            # 1. 압축파일이 들어있나?
+            # 2. 이미지 파일이 들어있나?
+            isImgIn = False
+            for pathFile in listFiles:
+                filename = util.GetParentDirName(pathFile, 0)
+                fileExt = util.GetExtStr(filename)
 
-            # 이미지 파일이 들어있다면...
-            if fileExt.lower() in [".jpg", ".webp", ".jpeg", ".png", ".gif"]:
+                # 압축파일이 들어있다면...
+                if fileExt.lower() in [".zip", ".cbz", ".rar"]:
+                    # 이름은 폴더/압축파일
+                    FolderName = util.GetParentDirName(pathFile, 1)
+                    ItemName = os.path.join(FolderName, filename)
+                    item = QListWidgetItem(ItemName)
+                    # 압축파일의 전체 경로를 따로 저장
+                    item.setData(Qt.UserRole, pathFile)
+                    self.list_ArcList.addItem(item)
+
+                # 이미지 파일이 들어있다면...
+                if fileExt.lower() in [".jpg", ".webp", ".jpeg", ".png", ".gif"]:
+                    isImgIn = True
+
+            if True == isImgIn:
                 # 이름은 폴더
-                FolderName = util.GetParentDirName(folder_path, 0)
+                FolderName = util.GetParentDirName(folder, 0)
                 item = QListWidgetItem(FolderName)
                 # 폴더 경로를 따로 저장
-                item.setData(Qt.UserRole, folder_path)
+                item.setData(Qt.UserRole, folder)
                 self.list_ArcList.addItem(item)