import sys import os import UtilPack as util from io import BytesIO from PyQt5.QtCore import Qt, QUrl, QSettings, QSize, QPoint, QByteArray from PyQt5.QtWidgets import QApplication, QWidget, QMainWindow, QPushButton, QVBoxLayout, QLineEdit, QHBoxLayout, QListWidget, QListWidgetItem, QLabel, QFileDialog from PyQt5.QtGui import QPixmap, QKeyEvent QApplication.setAttribute(Qt.AA_ShareOpenGLContexts) class MyApp(QMainWindow): def __init__(self): super().__init__() self.initUI() self.loadINI() def closeEvent(self, event): self.saveINI() event.accept() def resizeEvent(self, event): self.GetFitSize(self.label_Image.size()) print( self.label_Image.size() ) def initUI(self): layout = self.MakeUI() # 레이아웃을 윈도우에 적용 central_widget = QWidget() central_widget.setLayout(layout) self.setCentralWidget(central_widget) self.setWindowTitle('Manga Database') self.move(10, 10) self.show() def loadINI(self): settings = QSettings('MyApp', 'settings') window_size = settings.value('window/size', '800, 600') window_position = settings.value('window/position', '100, 100') def saveINI(self): settings = QSettings('MyApp', 'settings') # 키-값 형식으로 데이터 저장 settings.setValue('window/size', "1024,768") settings.setValue('window/position', "100,100") def MakeUI_Left(self): self.list_SrcPath = QListWidget() self.list_SrcPath.setMaximumHeight(400) self.list_SrcPath.setMaximumWidth(300) btn_Add = QPushButton("Add", self) btn_Add.clicked.connect(self.on_click_SrcAdd) btn_Del = QPushButton("Del", self) btn_Del.clicked.connect(self.on_click_SrcDel) btn_Read = QPushButton("Read!", self) btn_Read.clicked.connect(self.on_click_SrcRead) layout_Btns = QHBoxLayout() layout_Btns.addWidget(btn_Add) layout_Btns.addWidget(btn_Del) layout_Btns.addWidget(btn_Read) layout_Top = QVBoxLayout() layout_Top.addWidget(self.list_SrcPath) layout_Top.addLayout(layout_Btns) self.list_ArcList = QListWidget(self) self.list_ArcList.itemSelectionChanged.connect(self.on_Item_SelChanged_listArc) self.list_Infos = QListWidget(self) layout = QVBoxLayout() layout.addLayout(layout_Top) layout.addWidget(self.list_ArcList) layout.addWidget(self.list_Infos) return layout def MakeUI(self): layout_L = self.MakeUI_Left() # show Image in middle Layout self.label_Image = QLabel(self) # Load Default Image pixmap = QPixmap("layoutImg.jpeg") self.label_Image.setMaximumWidth(self.screen().size().width()-(self.list_ArcList.width()+400)) self.list_Items = QListWidget(self) self.list_Items.setFixedWidth(300) self.list_Items.itemSelectionChanged.connect(self.on_Item_SelChanged_Items) # 레이아웃 설정 layout = QHBoxLayout() layout.addLayout(layout_L) layout.addWidget(self.label_Image) layout.addWidget(self.list_Items) return layout def GetFitSize(self, szTarget): szDesktop = self.screen().size() szWindow = self.size() nTrgHeight = self.list_Items.height() nTrgWidth = szWindow.width() - ( self.list_ArcList.width() + self.list_Items.width() ) def on_click_SrcAdd(self): # 폴더 선택 창을 띄움 folder_path = QFileDialog.getExistingDirectory(self, '폴더 선택', '') if None == folder_path or True == util.IsEmptyStr(folder_path): return # 소스 폴더 목록에 추가 self.list_SrcPath.addItem(folder_path) # 폴더 내의 자식 폴더 목록을 가져온다. listFolders = util.ListSubDirectories(folder_path) # 선택한 폴더도 리스트에 추가한다. listFolders.append(folder_path) for folder in listFolders: # 폴더 내의 파일 목록을 가져온다. listFiles = util.ListContainFiles(folder) # 파일 목록을 훑어서 내용을 판단 # 1. 압축파일이 들어있나? # 2. 이미지 파일이 들어있나? isImgIn = False for pathFile in listFiles: filename = util.GetParentDirName(pathFile, 0) fileExt = util.GetExtStr(filename) # 압축파일이 들어있다면... 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 [".db"]: # 이름은 폴더/DB 파일이름 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, 0) item = QListWidgetItem(FolderName) # 폴더 경로를 따로 저장 item.setData(Qt.UserRole, folder) self.list_ArcList.addItem(item) def on_click_SrcDel(self): items = self.list_SrcPath.selectedItems() if not items: return selItemText = "" for item in items: selItemText = item.text() row = self.list_SrcPath.row(item) self.list_SrcPath.takeItem(row) def on_click_SrcRead(self): pass def on_Item_SelChanged_listArc(self): items = self.list_ArcList.selectedItems() self.list_Infos.clear() pathTarget = None for item in items: pathTarget = item.data(Qt.UserRole) if False == os.path.exists(pathTarget): return # 압축파일일 경우... listContents = [] if True == os.path.isfile(pathTarget): fileExt = util.GetExtStr(pathTarget) # 일단 zip 만... if fileExt.lower() in [".zip", ".cbz"]: listContents = util.GetZipContentList(pathTarget) elif fileExt.lower() == ".rar": listContents = [] elif True == os.path.isdir(pathTarget): listContents = util.ListFileExtRcr(pathTarget, [".jpg", ".jpeg", ".png", ".webp"]) self.list_Infos.addItem( QListWidgetItem(pathTarget) ) self.list_Infos.addItem( QListWidgetItem( f"{len(listContents)}" ) ) self.list_Items.clear() self.list_Items.addItems( sorted( listContents ) ) def on_Item_SelChanged_Items(self): items = self.list_Items.selectedItems() if 0 >= len(items): return selItemText = items[0].text() pixmap = QPixmap() pathTarget = self.list_Infos.item(0).text() if True == os.path.isfile(pathTarget): data = util.GetZippedFileByte(pathTarget, selItemText) imageIO = BytesIO(data) pixmap.loadFromData( QByteArray( imageIO.getvalue() ) ) elif True == os.path.isdir(pathTarget): if True == os.path.exists(selItemText): pixmap.load(selItemText) self.label_Image.setPixmap(pixmap) if __name__ == '__main__': app = QApplication(sys.argv) app.setQuitOnLastWindowClosed(True) main_window = MyApp() #main_window.show() sys.exit(app.exec_())